def test_basic(self): ann = Annotations() a = Symbol("a") next_a = Symbol("next(a)") init_a = Symbol("init(a)") ann.add(a, "next", next_a) ann.add(a, "init", init_a) ann.add(a, "related", next_a) ann.add(a, "related", init_a) self.assertIn(a, ann) self.assertEqual(set([next_a]), ann.annotations(a)["next"]) self.assertEqual(set([init_a]), ann.annotations(a)["init"]) self.assertEqual(set([init_a, next_a]), ann.annotations(a)["related"]) self.assertEqual(set([a]), ann.all_annotated_formulae("next")) self.assertEqual(set([a]), ann.all_annotated_formulae("init")) self.assertEqual(set([a]), ann.all_annotated_formulae("related")) self.assertEqual(set(), ann.all_annotated_formulae("non-existent"))
def test_remove(self): ann = Annotations() a = Symbol("a") next_a = Symbol("next(a)") init_a = Symbol("init(a)") ann.add(a, "next", next_a) ann.add(a, "init", init_a) ann.add(a, "related", next_a) ann.add(a, "related", init_a) self.assertIn(a, ann) ann.remove(a) self.assertNotIn(a, ann) self.assertEqual(None, ann.annotations(a)) self.assertEqual(set([]), ann.all_annotated_formulae("next")) self.assertEqual(set([]), ann.all_annotated_formulae("init")) self.assertEqual(set([]), ann.all_annotated_formulae("related")) self.assertEqual(set(), ann.all_annotated_formulae("non-existent"))