def test_init_tree(self):
"""
Checks the correctness of the intialization
"""
BinaryTree("a")
BinaryTree(1)
BinaryTree([2, 3])
def test_display_tree(self):
assert (str(self.tree) == "BT(1)")
self.tree.set_left(BinaryTree(2))
assert (str(self.tree) == "BT(1, BT(2) )")
self.tree.set_right(BinaryTree(3))
assert (str(self.tree) == "BT(1, BT(2) BT(3))")
def test_add_left(self):
assert len(self.tree) == 0
self.tree.set_left(None)
assert len(self.tree) == 0
self.tree.set_left(BinaryTree("là"))
def test_add_right(self):
assert len(self.tree) == 0
self.tree.set_right(BinaryTree(2))
assert len(self.tree) == 1
def test_init_tree_and_raises(self):
"""
Checks the correctness of the initialization
"""
with self.assertRaises(TypeError):
BinaryTree()
def setUp(self):
self.tree = BinaryTree(1)
class Test__BinaryTree(unittest.TestCase):
def setUp(self):
self.tree = BinaryTree(1)
def test_init_tree(self):
"""
Checks the correctness of the intialization
"""
BinaryTree("a")
BinaryTree(1)
BinaryTree([2, 3])
def test_init_tree_and_raises(self):
"""
Checks the correctness of the initialization
"""
with self.assertRaises(TypeError):
BinaryTree()
def test_add_right(self):
assert len(self.tree) == 0
self.tree.set_right(BinaryTree(2))
assert len(self.tree) == 1
def test_add_right_attribute_error(self):
with self.assertRaises(AssertionError):
self.tree.set_right(2)
def test_add_left(self):
assert len(self.tree) == 0
self.tree.set_left(None)
assert len(self.tree) == 0
self.tree.set_left(BinaryTree("là"))
def test_add_left_attribute_error(self):
with self.assertRaises(AssertionError):
self.tree.set_left("machin")
def test_display_tree(self):
assert (str(self.tree) == "BT(1)")
self.tree.set_left(BinaryTree(2))
assert (str(self.tree) == "BT(1, BT(2) )")
self.tree.set_right(BinaryTree(3))
assert (str(self.tree) == "BT(1, BT(2) BT(3))")