|
@@ -451,3 +451,13 @@ class TestWordFeatureExtractor(unittest.TestCase): |
|
|
test_extractor.teardown() |
|
|
test_extractor.teardown() |
|
|
|
|
|
|
|
|
self.assertEqual(len(test_extractor.features), 3) |
|
|
self.assertEqual(len(test_extractor.features), 3) |
|
|
|
|
|
|
|
|
|
|
|
def test_transcribe(self): |
|
|
|
|
|
"""Test _transcribe -- it calls model.transcribe""" |
|
|
|
|
|
video_source = TestSourceMedia().one_colour_silent_audio() |
|
|
|
|
|
test_extractor = extractors.WordFeatureExtractor(input_files=video_source) |
|
|
|
|
|
mock_model = MagicMock() |
|
|
|
|
|
test_extractor._transcribe(mock_model, "test.mp4") |
|
|
|
|
|
mock_model.transcribe.assert_called_once() |
|
|
|
|
|
|
|
|
|
|
|
|