def test_node_remove_child(self): par = Parameter(3, name='par') child_1 = Parameter(1, name='child_1') child_2 = Parameter(2, name='child_2') par.set_children([child_1, child_2]) par.remove_child(child_1) self.assertEqual(par.get_children(), [child_2])
def test_node_remove_child(self): par = Parameter(3, name='par') child_1 = Parameter(1, name='child_1') child_2 = Parameter(2, name='child_2') par.set_children([child_1, child_2]) par.remove_child(child_1) self.assertEqual(par.get_children(), [child_2]) with self.assertRaises(TypeError): par.remove_child("notanode")
def test_node_remove_parent(self): par = Parameter(3, name='par') parent_1 = Parameter(1, name='parent_1') parent_2 = Parameter(2, name='parent_2') parent_1.add_child(par) parent_2.add_child(par) with self.assertRaises(NodeException): par.remove_parent(parent_1) self.assertEqual(par.get_parents(), [parent_1, parent_2]) parent_1.remove_child(par) self.assertEqual(par.get_parents(), [parent_2]) with self.assertRaises(TypeError): par.remove_parent("notanode")