def test_remove_annotation(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)
ann.remove_annotation(a, "next")
self.assertNotIn("next", ann.annotations(a))
self.assertEqual(set([init_a]), ann.annotations(a)["init"])
self.assertEqual(set([init_a, next_a]), ann.annotations(a)["related"])
self.assertEqual(set([]), 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_annotation(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)
ann.remove_annotation(a, "next")
self.assertNotIn("next", ann.annotations(a))
self.assertEqual(set([init_a]), ann.annotations(a)["init"])
self.assertEqual(set([init_a, next_a]), ann.annotations(a)["related"])
self.assertEqual(set([]), 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"))
