def testMin(self):
tree = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
tree.insert(d)
expected = min(l)
self.assertEqual(expected, tree.min())
def testbstClone(self):
bst = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bst.insert(d)
clone = bst.bstClone()
self.assertTrue(bst.sameTree(clone))
def testPreOrder(self):
tree = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
tree.insert(d)
for v in tree.preOrder():
pass
def printTreeToList(self):
bst = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bst.insert(d)
blist = bst.treeToList()
bst.printList(blist)
def testInorderSuccessor(self):
bst = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bst.insert(d)
self.assertEqual(bst.right.inOrderSuccessor().data, bst.right.right.left.data)
self.assertEqual(bst.left.inOrderSuccessor().data, bst.left.right.left.data)
self.assertEqual(bst.right.right.inOrderSuccessor().data, bst.right.right.right.data)
def testWellOrderedNess(self):
bst = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bst.insert(d)
self.assertTrue(bst.wellOrdered())
bst.left.data = 300
self.assertFalse(bst.wellOrdered())
def testInOrder(self):
tree = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
tree.insert(d)
gen = tree.inOrder()
l.sort() # inOrder gives the elements in sorted Order always in BinaryTree
for i in l:
self.assertEqual(i, gen.next())
def testCommonAncestor(self):
bst = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bst.insert(d)
self.assertEqual(bst.commonAncestor(4, 12), 8)
self.assertEqual(bst.commonAncestor(4, 10), 8)
self.assertEqual(bst.commonAncestor(4, 14), 8)
self.assertEqual(bst.commonAncestor(1, 15), 8)
def testBreadthFirstTraversal(self):
bst = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bst.insert(d)
expectedBFSOrder = [8, 4, 12, 2, 6, 10, 14, 1, 3, 5, 7, 9, 11, 13, 15]
gen = bst.bfsTraversal()
for expectedValue in expectedBFSOrder:
self.assertEqual(expectedValue, gen.next())
def testSearchForData(self):
tree = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
tree.insert(d)
for i in l:
self.assertTrue(tree.search(i))
self.assertFalse(tree.search(-1))
def testCreateBSTFromTwoTraversals(self):
inOrder = [1, 2, 3, 4, 5, 6 ,7 ,8 ,9, 10, 11, 12, 13, 14, 15]
preOrder = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
bstFromTraversals = BinarySearchTree.createBSTFromTwoTraversals(inOrder, preOrder, 0, len(inOrder)-1)
bst = BinarySearchTree()
l = [8, 12, 4, 14, 2, 6, 10, 9, 7, 5, 11, 13, 3, 1, 15]
for v in l:
bst.insert(v)
self.assertTrue(bst.sameTree(bstFromTraversals))
def testMirror(self):
tree = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
tree.insert(d)
tree.mirror()
# when you do a mirror of a BST, the inOrder gives you elements in descending order
l.sort(reverse=True)
inOrderGen = tree.inOrder()
for v in l:
self.assertEqual(v, inOrderGen.next())
def testDepth(self):
tree = BinarySearchTree()
l = range(100)
for d in l:
tree.insert(d)
self.assertEqual(tree.getDepth(), len(l))
tree1 = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
tree1.insert(d)
self.assertEqual(tree1.getDepth(), 4)
def testIsBST(self):
bstOne = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bstOne.insert(d)
self.assertTrue(bstOne.isBST())
bstTwo = BinarySearchTree()
l = range(100)
for d in l:
bstTwo.insert(d)
self.assertTrue(bstTwo.isBST())
def testSum(self):
l = [-15, 5, 6, -8, 1, 3, 9, 2, 6, None, None, None, None, 4, 0, None, None, None, None, None, None, None, -1, 10, None]
s = 0
for v in l:
if v: s += v
bt = BinaryTree()
bt.create(l)
self.assertEqual(bt.sum(), s)
bstTwo = BinarySearchTree()
l = [8, 2, 1, 4, 5, 3, 9, 10, 15, 13, 12, 14, 18]
for v in l:
bstTwo.insert(v)
self.assertEqual(bstTwo.sum(), sum(l))
def testHasPathSum(self):
tree = BinarySearchTree()
l = range(100)
s = 0
for d in l:
tree.insert(d)
s += d
self.assertTrue(tree.hasPathSum(s))
tree1 = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
tree1.insert(d)
expected_sums = [8, 12, 14, 15, 17, 18, 23, 25, 20, 30, 34, 39, 41, 47, 49]
for e in expected_sums:
self.assertTrue(tree1.hasPathSum(e))
def testDelete(self):
tree = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
tree.insert(d)
# delete elements one by one and verify that it doesn't exist
deleted = []
for x in l:
deleted.append(x)
tree = tree.delete(x)
#verfiy all other elements exist
checks = [d for d in l if d not in deleted]
for v in checks:
self.assertTrue(tree and tree.search(v))
#verify x doesn't exist
self.assertFalse(tree and tree.search(x))
def testSpaceEfficientTraversal(self):
bstOne = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bstOne.insert(d)
bstTwo = bstOne
clone = bstOne
genOne = bstTwo.spaceEfficientTraversal()
genTwo = bstOne.inOrder()
l.sort() # inOrder gives the elements in sorted Order always in BinaryTree
for i in l:
self.assertEqual(i, genOne.next())
self.assertEqual(i, genTwo.next())
#create a copyTree method in the API and use it to test the similarity[Deep Copy]
self.assertTrue(bstOne.sameTree(clone))
def testSameTree(self):
treeOne = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
treeOne.insert(d)
treeTwo = BinarySearchTree()
for d in l:
treeTwo.insert(d)
self.assertTrue(treeOne.sameTree(treeTwo))
self.assertTrue(treeTwo.sameTree(treeOne))
treeTwo.insert(100)
self.assertFalse(treeTwo.sameTree(treeOne))
treeOne.insert(100)
self.assertTrue(treeTwo.sameTree(treeOne))
def testGetLeafCount(self):
bst = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bst.insert(d)
self.assertEqual(bst.getLeafCount(), 8)
bst = BinarySearchTree()
bst.insert(2)
bst.insert(1)
self.assertEqual(bst.getLeafCount(), 1)
bst.insert(3)
self.assertEqual(bst.getLeafCount(), 2)
def testDiameter(self):
bstOne = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bstOne.insert(d)
self.assertEqual(bstOne.diameter(), 7)
# single node
b = BinarySearchTree()
b.insert(8)
self.assertEqual(b.diameter(), 1)
b.insert(4)
self.assertEqual(b.diameter(), 2)
bstTwo = BinarySearchTree()
l = [8, 2, 1, 4, 5, 3, 9, 10, 15, 13, 12, 14, 18]
for v in l:
bstTwo.insert(v)
self.assertEqual(bstTwo.diameter(), 9)
def testGetInternalNodesCount(self):
bst = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bst.insert(d)
def testCountNoOfNodes(self):
tree = BinarySearchTree()
l = range(100)
for d in l:
tree.insert(d)
self.assertEqual(tree.size(), len(l))
def testPostOrder(self):
tree = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
tree.insert(d)
def testRadius(self):
"""
Given a binary tree find all the nodes at k distance from a given node.
"""
bst = BinarySearchTree()
l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15]
for d in l:
bst.insert(d)
k2list = [2, 6, 10, 14]
k3list = [1, 3, 5, 7, 9, 11, 13, 15]
k2gen = bst.getRadiusList(2)
for v in k2gen:
self.assertTrue(v in k2list)
k3gen = bst.getRadiusList(3)
for v in k3gen:
self.assertTrue(v in k3list)
k1gen = bst.getRadiusList(1)
for v in k1gen:
self.assertTrue(v in [4, 12])
k0gen = bst.getRadiusList(0)
for v in k0gen:
self.assertTrue(v in [8])
# try with different tree structure
"""
8 -----------------> radius 0 =>[8]
/ \
2 9 -----------------> radius 1 =>[2, 9]
/ \ \
1 4 10 -----------------> radius 2 =>[1, 4, 10]
/ \ \
3 5 15 -----------------> radius 3 =>[3, 5, 15]
/ \
13 18 -----------------> radius 1 =>[13, 18]
/ \
12 14 -----------------> radius 1 =>[12, 14]
"""
bstTwo = BinarySearchTree()
l = [8, 2, 1, 4, 5, 3, 9, 10, 15, 13, 12, 14, 18]
for v in l:
bstTwo.insert(v)
k3gen = bstTwo.getRadiusList(3)
k3list = [3, 5, 15]
for v in k3gen:
self.assertTrue(v in k3list)
k4gen = bstTwo.getRadiusList(4)
k4list = [13, 18]
for v in k4gen:
self.assertTrue(v in k4list)
k5gen = bstTwo.getRadiusList(5)
k5list = [12, 14]
for v in k5gen:
self.assertTrue(v in k5list)
