def test_feature_extractor_object(self): rtepair = rte_corpus.pairs(['rte3_dev.xml'])[33] extractor = RTEFeatureExtractor(rtepair) self.assertEqual(extractor.hyp_words, {'member', 'China', 'SCO.'}) self.assertEqual(extractor.overlap('word'), set()) self.assertEqual(extractor.overlap('ne'), {'China'}) self.assertEqual(extractor.hyp_extra('word'), {'member'})
def test_feature_extractor_object(self): rtepair = rte_corpus.pairs(["rte3_dev.xml"])[33] extractor = RTEFeatureExtractor(rtepair) assert extractor.hyp_words == {"member", "China", "SCO."} assert extractor.overlap("word") == set() assert extractor.overlap("ne") == {"China"} assert extractor.hyp_extra("word") == {"member"}