|
@@ -81,3 +81,27 @@ class TestLoudAudioFeatureExtractor(unittest.TestCase): |
|
|
test_extractor.teardown() |
|
|
test_extractor.teardown() |
|
|
self.assertEqual(test_extractor.features, []) |
|
|
self.assertEqual(test_extractor.features, []) |
|
|
|
|
|
|
|
|
|
|
|
class TestVideoActivityFeatureExtractor(unittest.TestCase): |
|
|
|
|
|
"""Test VideoActivityFeatureExtractor""" |
|
|
|
|
|
|
|
|
|
|
|
def test_init(self): |
|
|
|
|
|
video_source = TestSourceMedia().one_colour_silent_audio() |
|
|
|
|
|
test_extractor = extractors.VideoActivityFeatureExtractor(input_files=video_source) |
|
|
|
|
|
self.assertTrue(test_extractor) |
|
|
|
|
|
|
|
|
|
|
|
def test_init_noinput(self): |
|
|
|
|
|
"""test init - no input files""" |
|
|
|
|
|
with self.assertRaises(ValueError): |
|
|
|
|
|
test_extractor = extractors.VideoActivityFeatureExtractor() |
|
|
|
|
|
|
|
|
|
|
|
def test_extract(self): |
|
|
|
|
|
"""Test extract with basic input file runs with no errors""" |
|
|
|
|
|
video_source = TestSourceMedia().one_colour_silent_audio() |
|
|
|
|
|
test_extractor = extractors.VideoActivityFeatureExtractor(input_files=video_source) |
|
|
|
|
|
test_extractor.setup() |
|
|
|
|
|
test_extractor.run() |
|
|
|
|
|
test_extractor.teardown() |
|
|
|
|
|
self.assertTrue(test_extractor.features) |
|
|
|
|
|
|
|
|
|
|
|
# TODO: add sample video with activity to test _activitydetect() |
|
|
|
|
|
|