|
@@ -416,10 +416,17 @@ class TestWordFeatureExtractor(unittest.TestCase): |
|
|
test_extractor._batched_model = MagicMock() |
|
|
test_extractor._batched_model = MagicMock() |
|
|
# set up and run the extractor |
|
|
# set up and run the extractor |
|
|
test_extractor.setup(words=self._MOCK_SENTENCE) |
|
|
test_extractor.setup(words=self._MOCK_SENTENCE) |
|
|
test_extractor.run() |
|
|
|
|
|
test_extractor.teardown() |
|
|
|
|
|
|
|
|
with self.subTest("batched"): |
|
|
|
|
|
test_extractor.run() |
|
|
|
|
|
test_extractor.teardown() |
|
|
|
|
|
self.assertEqual(len(test_extractor.features), 9) |
|
|
|
|
|
test_extractor.features = [] # reset features |
|
|
|
|
|
|
|
|
self.assertEqual(len(test_extractor.features), 9) |
|
|
|
|
|
|
|
|
with self.subTest("non-batched"): |
|
|
|
|
|
test_extractor.DEFAULT_PIPELINE_TYPE = "non-batched" |
|
|
|
|
|
test_extractor.run() |
|
|
|
|
|
test_extractor.teardown() |
|
|
|
|
|
self.assertEqual(len(test_extractor.features), 9) |
|
|
|
|
|
|
|
|
def test_extract_mocked_transcribe_no_matching_words(self): |
|
|
def test_extract_mocked_transcribe_no_matching_words(self): |
|
|
"""Mock out the actual call to transcribe but match no words in the sentence""" |
|
|
"""Mock out the actual call to transcribe but match no words in the sentence""" |
|
|