def test_set_root(self): t1 = BinaryTree("cedar") t1.key = "oak" self.assertEquals(t1.key, "oak") t1.set_root("maple") self.assertEquals(t1.key, "maple") t1.set_root(100) self.assertEquals(t1.key, 100)
def test_get(self): t1 = BinaryTree("pine") t1.insert_right("right branch") t1.insert_left("left branch") self.assertEquals(t1.get_rc(), "right branch") self.assertEquals(t1.get_lc(), "left branch") self.assertEquals(t1.get_root(), "pine") t1.insert_right("right stick") t1.insert_left("left stick") t1.set_root("hemlock") self.assertEquals(t1.get_rc(), "right stick") self.assertEquals(t1.get_lc(), "left stick") self.assertEquals(t1.get_root(), "hemlock")