def test_simple(self): X = ["The light crashes"] m = MapToSynsets() Z = m.transform(X) self.assertEqual(len(Z), 1) self.assertTrue(isinstance(Z[0], str)) for word in ["light.a.01", "crash.v.01"]: self.assertIn(word, Z[0]) self.assertNotIn("crash.n.02", Z[0])
def test_empty(self): m = MapToSynsets() Z = m.transform([]) self.assertEqual(len(Z), 0)