class AlphaNumericFeatureExtractorTests(unittest.TestCase): def setUp(self): self.extractor = NEFeatureExtractor() def test_get_features(self): obj = { 'target': ['a', 'boy', 'hits', '3', 'dogs', 'a11i', 'o8', 'www1'], 'source': ['un', 'garcon', 'bate', '3', 'ou', '4', 'chiens', 'b2b'], 'token': ['3', 'dogs', 'Tinky', 'and', 'Winky'], 'index': (3, 8), 'source_token': ['3', 'ou', 'Tinky', 'et', 'Winky'], 'source_index': (3, 8) } (src_ne, tg_ne) = self.extractor.get_features(obj) self.assertEqual(src_ne, 1) self.assertEqual(tg_ne, 1) def test_get_features_no_src(self): obj_no_src = { 'target': ['a', 'boy', 'hits', '3', 'dogs', 'a11i', 'o8', 'www1'], 'source': ['un', 'garcon', 'bate', '3', 'ou', '4', 'chiens', 'b2b'], 'token': ['3', 'dogs', 'Tinky', 'and', 'Winky'], 'index': (3, 8), 'source_token': [], 'source_index': [] } (src_ne, tg_ne) = self.extractor.get_features(obj_no_src) self.assertEqual(src_ne, 0) self.assertEqual(tg_ne, 1) def test_get_features_no_NE(self): obj_no_src = { 'target': ['a', 'boy', 'hits', '3', 'dogs', 'a11i', 'o8', 'www1'], 'source': ['un', 'garcon', 'bate', '3', 'ou', '4', 'chiens', 'b2b'], 'token': ['3', 'dogs', 'aDa', 'and', 'ooO'], 'index': (3, 8), 'source_token': [], 'source_index': [] } (src_ne, tg_ne) = self.extractor.get_features(obj_no_src) self.assertEqual(src_ne, 0) self.assertEqual(tg_ne, 0)
class AlphaNumericFeatureExtractorTests(unittest.TestCase): def setUp(self): self.extractor = NEFeatureExtractor() def test_get_features(self): obj = {'target': ['a', 'boy', 'hits', '3', 'dogs', 'a11i', 'o8', 'www1'], 'source': ['un', 'garcon', 'bate', '3', 'ou', '4', 'chiens', 'b2b'], 'token': ['3', 'dogs', 'Tinky', 'and', 'Winky'], 'index': (3, 8), 'source_token': ['3', 'ou', 'Tinky', 'et', 'Winky'], 'source_index': (3, 8)} (src_ne, tg_ne) = self.extractor.get_features(obj) self.assertEqual(src_ne, 1) self.assertEqual(tg_ne, 1) def test_get_features_no_src(self): obj_no_src = {'target': ['a', 'boy', 'hits', '3', 'dogs', 'a11i', 'o8', 'www1'], 'source': ['un', 'garcon', 'bate', '3', 'ou', '4', 'chiens', 'b2b'], 'token': ['3', 'dogs', 'Tinky', 'and', 'Winky'], 'index': (3, 8), 'source_token': [], 'source_index': []} (src_ne, tg_ne) = self.extractor.get_features(obj_no_src) self.assertEqual(src_ne, 0) self.assertEqual(tg_ne, 1) def test_get_features_no_NE(self): obj_no_src = {'target': ['a', 'boy', 'hits', '3', 'dogs', 'a11i', 'o8', 'www1'], 'source': ['un', 'garcon', 'bate', '3', 'ou', '4', 'chiens', 'b2b'], 'token': ['3', 'dogs', 'aDa', 'and', 'ooO'], 'index': (3, 8), 'source_token': [], 'source_index': []} (src_ne, tg_ne) = self.extractor.get_features(obj_no_src) self.assertEqual(src_ne, 0) self.assertEqual(tg_ne, 0)
def setUp(self): self.extractor = NEFeatureExtractor()