def testListSort(self):
listToSort = [2,3,5,1,6,7,89,8,4,12,34,99,77,32]
sortingTree = BST_rec()
boolResult = sortingTree.insertList(listToSort)
self.assertTrue(boolResult)
listToSort.sort()
self.assertEqual(sortingTree.traverseDFSinorder(), listToSort)
print("\ntestListSort PASSED")
def setUp(self):
self.perfect = BST_rec()
for key in [7,3,1,5,0,6,2,4,11,9,8,10,13,12,14]:
self.perfect.insert(key)
self.degenerate = BST_rec()
for key in [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14]:
self.degenerate.insert(key)
self.balanced = BST_rec()
for key in [7,5,3,1,2,0,4,6,12,10,9,11,13,14,8]:
self.balanced.insert(key)
self.rightHeavy = BST_rec()
for key in [5,3,10,1,4,0,2,7,6,9,8,13,11,14,12]:
self.rightHeavy.insert(key)
self.leftHeavy = BST_rec()
for key in [11,13,12,14,6,3,8,9,7,4,1,5,2,0,10]:
self.leftHeavy.insert(key)
self.treeList = [self.perfect, self.degenerate, self.balanced, self.rightHeavy, self.leftHeavy]
self.empty = BST_rec()
class TestBST(unittest.TestCase):
def setUp(self):
self.perfect = BST_rec()
for key in [7,3,1,5,0,6,2,4,11,9,8,10,13,12,14]:
self.perfect.insert(key)
self.degenerate = BST_rec()
for key in [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14]:
self.degenerate.insert(key)
self.balanced = BST_rec()
for key in [7,5,3,1,2,0,4,6,12,10,9,11,13,14,8]:
self.balanced.insert(key)
self.rightHeavy = BST_rec()
for key in [5,3,10,1,4,0,2,7,6,9,8,13,11,14,12]:
self.rightHeavy.insert(key)
self.leftHeavy = BST_rec()
for key in [11,13,12,14,6,3,8,9,7,4,1,5,2,0,10]:
self.leftHeavy.insert(key)
self.treeList = [self.perfect, self.degenerate, self.balanced, self.rightHeavy, self.leftHeavy]
self.empty = BST_rec()
def testEmpty(self):
expResult = str([])
self.assertEqual(expResult, self.empty.printBFS())
self.assertEqual(expResult, self.empty.printDFSinorder())
self.assertEqual(expResult, self.empty.printDFSpostorder())
self.assertEqual(expResult, self.empty.printDFSpreorder())
print("\ntestEmpty PASSED")
def testInsert(self):
boolResult = self.empty.insert(7)
expResult = str([7])
self.assertTrue(boolResult)
self.assertEqual(expResult, self.empty.printBFS())
self.assertEqual(expResult, self.empty.printDFSinorder())
self.assertEqual(expResult, self.empty.printDFSpostorder())
self.assertEqual(expResult, self.empty.printDFSpreorder())
print("\ntestInsert PASSED")
def testBFS(self):
expResult = str([7,3,11,1,5,9,13,0,2,4,6,8,10,12,14])
self.assertEqual(expResult, self.perfect.printBFS())
expResult = str([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14])
self.assertEqual(expResult, self.degenerate.printBFS())
expResult = str([7,5,12,3,6,10,13,1,4,9,11,14,0,2,8])
self.assertEqual(expResult, self.balanced.printBFS())
expResult = str([5,3,10,1,4,7,13,0,2,6,9,11,14,8,12])
self.assertEqual(expResult, self.rightHeavy.printBFS())
expResult = str([11,6,13,3,8,12,14,1,4,7,9,0,2,5,10])
self.assertEqual(expResult, self.leftHeavy.printBFS())
print("testBFS PASSED")
def testDFSinorder(self):
expResult = str([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14])
self.assertEqual(expResult, self.perfect.printDFSinorder())
self.assertEqual(expResult, self.degenerate.printDFSinorder())
self.assertEqual(expResult, self.balanced.printDFSinorder())
self.assertEqual(expResult, self.rightHeavy.printDFSinorder())
self.assertEqual(expResult, self.leftHeavy.printDFSinorder())
print("\ntestDFSinorder PASSED")
def testDFSpreorder(self):
expResult = str([7,3,1,0,2,5,4,6,11,9,8,10,13,12,14])
self.assertEqual(expResult, self.perfect.printDFSpreorder())
expResult = str([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14])
self.assertEqual(expResult, self.degenerate.printDFSpreorder())
expResult = str([7,5,3,1,0,2,4,6,12,10,9,8,11,13,14])
self.assertEqual(expResult, self.balanced.printDFSpreorder())
expResult = str([5,3,1,0,2,4,10,7,6,9,8,13,11,12,14])
self.assertEqual(expResult, self.rightHeavy.printDFSpreorder())
expResult = str([11,6,3,1,0,2,4,5,8,7,9,10,13,12,14])
self.assertEqual(expResult, self.leftHeavy.printDFSpreorder())
print("\ntestDFSpreorder PASSED")
def testDFSpostorder(self):
expResult = str([0,2,1,4,6,5,3,8,10,9,12,14,13,11,7])
self.assertEqual(expResult, self.perfect.printDFSpostorder())
expResult = str([14,13,12,11,10,9,8,7,6,5,4,3,2,1,0])
self.assertEqual(expResult, self.degenerate.printDFSpostorder())
expResult = str([0,2,1,4,3,6,5,8,9,11,10,14,13,12,7])
self.assertEqual(expResult, self.balanced.printDFSpostorder())
expResult = str([0,2,1,4,3,6,8,9,7,12,11,14,13,10,5])
self.assertEqual(expResult, self.rightHeavy.printDFSpostorder())
expResult = str([0,2,1,5,4,3,7,10,9,8,6,12,14,13,11])
self.assertEqual(expResult, self.leftHeavy.printDFSpostorder())
print("\ntestDFSpostorder PASSED")
def testFind(self):
for tree in self.treeList:
self.assertTrue(tree.find(0))
self.assertFalse(tree.find(15))
self.assertFalse(tree.find(-1))
self.assertTrue(tree.find(14))
self.assertFalse(self.empty.find(0))
print("\ntestFind PASSED")
def testDuplicateInsert(self):
boolResult = self.perfect.insert(1)
self.assertFalse(boolResult)
boolResult = self.balanced.insert(1)
self.assertFalse(boolResult)
boolResult = self.degenerate.insert(1)
self.assertFalse(boolResult)
boolResult = self.rightHeavy.insert(1)
self.assertFalse(boolResult)
boolResult = self.leftHeavy.insert(1)
self.assertFalse(boolResult)
print("\ntestDuplicateInsert PASSED")
def testCopy(self):
balancedCopy = self.balanced.copyTree()
balancedCopy.insert(15)
degenerateCopy = self.degenerate.copyTree()
degenerateCopy.insert(15)
perfectCopy = self.perfect.copyTree()
perfectCopy.insert(15)
rightHeavyCopy = self.rightHeavy.copyTree()
rightHeavyCopy.insert(15)
leftHeavyCopy = self.leftHeavy.copyTree()
leftHeavyCopy.insert(15)
#Test to make sure previous trees were not changed
expResult = str([7,3,11,1,5,9,13,0,2,4,6,8,10,12,14])
self.assertEqual(expResult, self.perfect.printBFS())
expResult = str([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14])
self.assertEqual(expResult, self.degenerate.printBFS())
expResult = str([7,5,12,3,6,10,13,1,4,9,11,14,0,2,8])
self.assertEqual(expResult, self.balanced.printBFS())
expResult = str([5,3,10,1,4,7,13,0,2,6,9,11,14,8,12])
self.assertEqual(expResult, self.rightHeavy.printBFS())
expResult = str([11,6,13,3,8,12,14,1,4,7,9,0,2,5,10])
self.assertEqual(expResult, self.leftHeavy.printBFS())
#Tests to see that the new trees were changed
expResult = str([7,3,11,1,5,9,13,0,2,4,6,8,10,12,14,15])
self.assertEqual(expResult, perfectCopy.printBFS())
expResult = str([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15])
self.assertEqual(expResult, degenerateCopy.printBFS())
expResult = str([7,5,12,3,6,10,13,1,4,9,11,14,0,2,8,15])
self.assertEqual(expResult, balancedCopy.printBFS())
expResult = str([5,3,10,1,4,7,13,0,2,6,9,11,14,8,12,15])
self.assertEqual(expResult, rightHeavyCopy.printBFS())
expResult = str([11,6,13,3,8,12,14,1,4,7,9,15,0,2,5,10])
self.assertEqual(expResult, leftHeavyCopy.printBFS())
print("\ntestCopy PASSED")
def testFindMin(self):
expResult = 0
for tree in self.treeList:
self.assertEqual(expResult,tree.findMin())
self.assertEqual(None, self.empty.findMin())
print("\ntestFindMin PASSED")
def testFindMax(self):
expResult = 14
for tree in self.treeList:
self.assertEqual(expResult,tree.findMax())
self.assertEqual(None, self.empty.findMax())
print("\ntestFindMax PASSED")
def testDeleteLeaf(self):
#Test for deleting a leaf
boolResults = []
for tree in self.treeList:
boolResults.append(tree.delete(14))
for result in boolResults:
self.assertTrue(result)
expResult = str([7,3,11,1,5,9,13,0,2,4,6,8,10,12])
self.assertEqual(expResult, self.perfect.printBFS())
expResult = str([0,1,2,3,4,5,6,7,8,9,10,11,12,13])
self.assertEqual(expResult, self.degenerate.printBFS())
expResult = str([7,5,12,3,6,10,13,1,4,9,11,0,2,8])
self.assertEqual(expResult, self.balanced.printBFS())
expResult = str([5,3,10,1,4,7,13,0,2,6,9,11,8,12])
self.assertEqual(expResult, self.rightHeavy.printBFS())
expResult = str([11,6,13,3,8,12,1,4,7,9,0,2,5,10])
self.assertEqual(expResult, self.leftHeavy.printBFS())
print("\ntestDeleteLeaf PASSED")
def testDeleteFail(self):
#Test for attempting to delete a nonexistent node
boolResults = []
for tree in self.treeList:
boolResults.append(tree.delete(15))
for result in boolResults:
self.assertFalse(result)
self.assertFalse(self.empty.delete(14))
#Test to make sure previous trees were not changed
expResult = str([7,3,11,1,5,9,13,0,2,4,6,8,10,12,14])
self.assertEqual(expResult, self.perfect.printBFS())
expResult = str([0,1,2,3,4,5,6,7,8,9,10,11,12,13,14])
self.assertEqual(expResult, self.degenerate.printBFS())
expResult = str([7,5,12,3,6,10,13,1,4,9,11,14,0,2,8])
self.assertEqual(expResult, self.balanced.printBFS())
expResult = str([5,3,10,1,4,7,13,0,2,6,9,11,14,8,12])
self.assertEqual(expResult, self.rightHeavy.printBFS())
expResult = str([11,6,13,3,8,12,14,1,4,7,9,0,2,5,10])
self.assertEqual(expResult, self.leftHeavy.printBFS())
print("\ntestDeleteFail PASSED")
def testDeleteInternal(self):
#Test for deleting an internal node
for tree in self.treeList:
tree.delete(3)
expResultLesser = "\[7, 2, 11, 1, 5, 9, 13, 0, 4, 6, 8, 10, 12, 14\]"
expResultGreater = "\[7, 4, 11, 1, 5, 9, 13, 0, 2, 6, 8, 10, 12, 14\]"
self.assertRegex(self.perfect.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
expResultLesser = "\[0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14\]"
expResultGreater = "\[0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14\]"
self.assertRegex(self.degenerate.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
expResultLesser = "\[7, 5, 12, 2, 6, 10, 13, 1, 4, 9, 11, 14, 0, 8\]"
expResultGreater = "\[7, 5, 12, 4, 6, 10, 13, 1, 9, 11, 14, 0, 2, 8\]"
self.assertRegex(self.balanced.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
expResultLesser = "\[5, 2, 10, 1, 4, 7, 13, 0, 6, 9, 11, 14, 8, 12\]"
expResultGreater = "\[5, 4, 10, 1, 7, 13, 0, 2, 6, 9, 11, 14, 8, 12\]"
self.assertRegex(self.rightHeavy.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
expResultLesser = "\[11, 6, 13, 2, 8, 12, 14, 1, 4, 7, 9, 0, 5, 10\]"
expResultGreater = "\[11, 6, 13, 4, 8, 12, 14, 1, 5, 7, 9, 0, 2, 10\]"
self.assertRegex(self.leftHeavy.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
print("\ntestDeleteInternal PASSED")
def testDeleteRoot(self):
#Test for deleting the root
self.balanced.delete(7)
self.perfect.delete(7)
self.degenerate.delete(0)
self.rightHeavy.delete(5)
self.leftHeavy.delete(11)
expResultLesser = "\[6, 3, 11, 1, 5, 9, 13, 0, 2, 4, 8, 10, 12, 14\]"
expResultGreater = "\[8, 3, 11, 1, 5, 9, 13, 0, 2, 4, 6, 10, 12, 14\]"
self.assertRegex(self.perfect.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
expResultLesser = "\[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14\]"
expResultGreater = "\[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14\]"
self.assertRegex(self.degenerate.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
expResultLesser = "\[6, 5, 12, 3, 10, 13, 1, 4, 9, 11, 14, 0, 2, 8\]"
expResultGreater = "\[8, 5, 12, 3, 6, 10, 13, 1, 4, 9, 11, 14, 0, 2\]"
self.assertRegex(self.balanced.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
expResultLesser = "\[4, 3, 10, 1, 7, 13, 0, 2, 6, 9, 11, 14, 8, 12\]"
expResultGreater = "\[6, 3, 10, 1, 4, 7, 13, 0, 2, 9, 11, 14, 8, 12\]"
self.assertRegex(self.rightHeavy.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
expResultLesser = "\[10, 6, 13, 3, 8, 12, 14, 1, 4, 7, 9, 0, 2, 5\]"
expResultGreater = "\[12, 6, 13, 3, 8, 14, 1, 4, 7, 9, 0, 2, 5, 10\]"
self.assertRegex(self.leftHeavy.printBFS(), ('%s|%s' %
(expResultLesser,expResultGreater)))
print("\ntestDeleteRoot PASSED")
def testListSort(self):
listToSort = [2,3,5,1,6,7,89,8,4,12,34,99,77,32]
sortingTree = BST_rec()
boolResult = sortingTree.insertList(listToSort)
self.assertTrue(boolResult)
listToSort.sort()
self.assertEqual(sortingTree.traverseDFSinorder(), listToSort)
print("\ntestListSort PASSED")