def test_3(self): text_preprocessing_result = PicklableMock() text_preprocessing_result.org_token_values = [] result = self.filler.extract(text_preprocessing_result, None) self.assertIsNone(result)
def test_2(self): expected = "тинькофф" text_preprocessing_result = PicklableMock() text_preprocessing_result.org_token_values = ["тинькофф", "втб", "мегафон"] result = self.filler.extract(text_preprocessing_result, None) self.assertEqual(expected, result)