Преглед на файлове

test: [WordFE] cover non-batched

main
Rob Hallam преди 2 месеца
родител
ревизия
5ca6f9e9d0
променени са 1 файла, в които са добавени 10 реда и са изтрити 3 реда
  1. +10
    -3
      test/test_feature_extractors.py

+ 10
- 3
test/test_feature_extractors.py Целия файл

@@ -416,10 +416,17 @@ class TestWordFeatureExtractor(unittest.TestCase):
test_extractor._batched_model = MagicMock()
# set up and run the extractor
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):
"""Mock out the actual call to transcribe but match no words in the sentence"""


Зареждане…
Отказ
Запис