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