Пример #1
0
	def test_single_right_rotation_1(self):
		data = [3, 2, 1]
		avl_tree = AVLTree(data)
		assert avl_tree.root.data == 2
		assert avl_tree.root.left_child.data == 1
		assert avl_tree.root.right_child.data == 3
		assert avl_tree.items_level_order() == [2, 1, 3]
Пример #2
0
	def test_right_left_rotation_2(self):
		data = [20, 10, 40, 30, 50, 35]
		avl_tree = AVLTree(data)
		assert avl_tree.root.data == 30
		assert avl_tree.items_level_order() == [30, 20, 40, 10, 35, 50]
Пример #3
0
	def test_right_left_rotation_1(self):
		data = [1, 8, 4]
		avl_tree = AVLTree(data)
		assert avl_tree.root.data == 4
		assert avl_tree.items_level_order() == [4, 1, 8]
Пример #4
0
	def test_left_right_rotation_2(self):
		data = [1, 4, 8, 12, 16, 20, 25, 30, 35, 28]
		avl_tree = AVLTree(data)
		assert avl_tree.root.data == 12
		assert avl_tree.items_level_order() == [12, 4, 25, 1, 8, 20, 30, 16, 28, 35]
Пример #5
0
	def test_single_right_rotation_2(self):
		data = [40, 50, 30, 10, 5]
		avl_tree = AVLTree(data)
		assert avl_tree.root.data == 40
		assert avl_tree.items_level_order() == [40, 10, 50, 5, 30]
Пример #6
0
	def test_tree_level_order_traversal(self):
		data = [20, 10, 40, 30, 50]
		avl_tree = AVLTree(data)
		assert avl_tree.items_level_order() == [20, 10, 40, 30, 50]