def test_bfs_large(self):
tree = Node.Node(
1, Node.Node(2, None, Node.Node(4, Node.Node(7, None, None),
None)),
Node.Node(
3, Node.Node(5, None, None),
Node.Node(6, None, Node.Node(8, Node.Node(9, None, None),
None))))
self.assertEqual(TreeOperations.breadth_first_search(tree),
[1, 2, 3, 4, 5, 6, 7, 8, 9])
def test_bfs_root(self):
tree = Node.Node(5, None, None)
self.assertEqual(TreeOperations.breadth_first_search(tree), [5])
def test_bfs_both(self):
tree = Node.Node(5, Node.Node(4, None, None), Node.Node(3, None, None))
self.assertEqual(TreeOperations.breadth_first_search(tree), [5, 4, 3])
def test_bfs_empty(self):
self.assertEqual(TreeOperations.breadth_first_search(None), [])