Pārlūkot izejas kodu

test: add TestVAFE

main
Rob Hallam pirms 3 mēnešiem
vecāks
revīzija
4eaddf9db2
1 mainītis faili ar 24 papildinājumiem un 0 dzēšanām
  1. +24
    -0
      test/test_feature_extractors.py

+ 24
- 0
test/test_feature_extractors.py Parādīt failu

@@ -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()


Notiek ielāde…
Atcelt
Saglabāt