def test_inserting_unsorted_values_should_alter_root(self):
"""Testing the rotate methods"""
rbt = RedBlackTree()
rbt.insert(9)
rbt.insert(11)
rbt.insert(10)
self.assertTrue(rbt.get_root().value == 10)
def test_bfs(func, size):
rbt = RedBlackTree()
for i in range(1, size):
rbt.insert(random_string(14))
this_func = eval(func)
this_func(rbt.get_root(), visit_node)
def test_node_inserted_should_be_red(self):
rbt = RedBlackTree()
rbt.insert(9)
rbt.insert(11)
self.assertTrue(rbt.get_root().get_right().is_red())
def test_inserting_greater_value_should_create_right_node(self):
rbt = RedBlackTree()
rbt.insert(9)
rbt.insert(11)
self.assertTrue(rbt.get_root().get_right().value == 11)
def test_inserting_lesser_value_should_create_left_node(self):
rbt = RedBlackTree()
rbt.insert(5)
rbt.insert(4)
self.assertTrue(rbt.get_root().get_left().value == 4)
def test_first_node_should_be_black(self):
rbt = RedBlackTree()
rbt.insert(5)
rbt.insert(4)
self.assertTrue(rbt.get_root().is_black())
