瀏覽代碼

test: add TestVAFE

main
Rob Hallam 2 月之前
父節點
當前提交
4eaddf9db2
共有 1 個文件被更改,包括 24 次插入0 次删除
  1. +24
    -0
      test/test_feature_extractors.py

+ 24
- 0
test/test_feature_extractors.py 查看文件

@@ -81,3 +81,27 @@ class TestLoudAudioFeatureExtractor(unittest.TestCase):
test_extractor.teardown()
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()


Loading…
取消
儲存