def test_preorder_single_node(self): bst = BinarySearchTree(data=5) stdout_org = sys.stdout stdout_mock = StdOutMock() try: sys.stdout = stdout_mock bst.preorder(bst.root) finally: sys.stdout = stdout_org self.assertEqual(str(stdout_mock), '5')
def test_preorder(self): bst = BinarySearchTree(data=5) bst.insert(3) bst.insert(7) bst.insert(1) bst.insert(2) bst.insert(9) bst.insert(6) stdout_org = sys.stdout stdout_mock = StdOutMock() try: sys.stdout = stdout_mock bst.preorder(bst.root) finally: sys.stdout = stdout_org self.assertEqual(str(stdout_mock), '5312769')