class TagsTest(TestCase): def setUp(self): self.t1 = Tag(name='test1').save() self.t2 = Tag(name='test2').save() self.t3 = Tag(name='test3').save() def tearDown(self): Tag.drop_collection() def test_add_parent(self): self.t2.add_parent(self.t1) self.t3.add_parent(self.t1) self.assertEquals(self.t1 in self.t2.parents, True) self.assertEquals(self.t1 in self.t3.parents, True) self.assertEquals(self.t2 in self.t1.children, True) self.assertEquals(self.t3 in self.t1.children, True) def test_add_child(self): self.t1.add_child(self.t2) self.t1.add_child(self.t3) self.assertEquals(self.t1 in self.t2.parents, True) self.assertEquals(self.t1 in self.t3.parents, True) self.assertEquals(self.t2 in self.t1.children, True) self.assertEquals(self.t3 in self.t1.children, True)