def test_1(self):
a = TreeNode(1)
b = TreeNode(2)
c = TreeNode(3)
a.left = b
a.right = c
self.assertEqual(lot.levelOrder(a), [[1],[2,3]])
def test_something(self):
a = TreeNode(2)
b = TreeNode(1)
a.left = b
c = bi.Solution(a)
self.res = []
while c.hasNext():
self.res.append(c.next())
self.assertEqual(self.res, [1,2])
def test_1(self):
a = TreeNode(1)
b = TreeNode(2)
c = TreeNode(3)
d = TreeNode(4)
e = TreeNode(5)
a.left = b
a.right = c
c.left = d
c.right = e
self.assertEqual(lot.levelOrder(a), [[1],[2,3],[4,5]])
