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_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"))
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"))
