def test_remove_child(self):
a = HOG(id="a")
b = HOG(id="b")
c = HOG(id="b")
self.assertListEqual(a.children, [])
a.add_child(b)
a.add_child(c)
self.assertListEqual(a.children, [b,c])
a.remove_child(b)
self.assertListEqual(a.children, [c])
def test_cannot_add_any_type_as_child(self):
a = HOG()
with self.assertRaises(TypeError):
a.add_child("test")
def test_cannot_be_child_of_self(self):
a = HOG()
with self.assertRaises(ECE):
a.add_child(a)
def test_hog_can_be_nested_and_traversed_up_and_down(self):
a = HOG(id="a")
b = HOG(id="b")
a.add_child(b)
self.assertIn(b, b.parent.children)