def test_1_null_2_returns_1_2(self): a_root = TreeNode(1) a_root.right = TreeNode(2) observed = Solution().preorder_traversal(a_root) self.assertEqual(observed, [ 1, 2, ])
def test_only_left_children_size_5_returns(self): expected = list(range(0, 26, 5)) nodes = list(TreeNode(i) for i in expected) i = 0 while i < len(nodes) - 1: nodes[i].left = nodes[i + 1] i += 1 observed = Solution().preorder_traversal(nodes[0]) self.assertEqual(expected, observed)
def setUp(self): self.root = TreeNode(0) self.node1 = TreeNode(1) self.node2 = TreeNode(2) self.node3 = TreeNode(3) self.node4 = TreeNode(4) self.node5 = TreeNode(5) self.node5 = TreeNode(5) self.node6 = TreeNode(6) self.root.left = self.node1 self.root.right = self.node2 self.node1.left = self.node3 self.node1.right = self.node4 self.node2.left = self.node5 self.node2.right = self.node6 s = Solution() s.mylist = [] self.correct_answer = s.preorder(self.root)
def test_1null23_returns_123(self): expected = [ 1, 2, 3, ] nodes = list(map(lambda i: TreeNode(i), expected)) nodes[0].right = nodes[1] nodes[1].left = nodes[-1] observed = Solution().preorder_traversal(nodes[0]) self.assertEqual(observed, expected)
def test_3_1_2_returns_3_1_2(self): expected = [ 3, 1, 2, ] a_root = TreeNode(3) a_root.left = TreeNode(1) a_root.right = TreeNode(2) observed = Solution().preorder_traversal(a_root) self.assertEqual(observed, expected)
def test_1_2_returns_1_2(self): nodes = list(TreeNode(i) for i in range(1, 3)) a_root = TreeNode(1) a_root.left = TreeNode(2) observed = Solution().preorder_traversal(a_root) self.assertEqual(observed, [1, 2])
def test_1_returns_1(self): observed = Solution().preorder_traversal(TreeNode(1)) self.assertEqual(observed, [ 1, ])
def test_preorder_non_recusive3(self): #print "\n---------------------------------" #print "Preorder Non-recursive3" s = Solution() mylist = s.preorder_non_recursive3(self.root) self.assertEqual(mylist, self.correct_answer)
def test_preorder_non_recusive(self): s = Solution() mylist = s.preorder_non_recursive(self.root) self.assertEqual(mylist, self.correct_answer)