def setUp(self):
self.bst = BinaryTree()
r = BinaryNode(5)
rl = BinaryNode(3)
r.right = BinaryNode(6)
r.left = rl
rl.left = BinaryNode(2)
rl.right = BinaryNode(4)
self.r = r
def constructAVL(aList):
bt = BinaryTree()
for val in aList:
bt.add(val)
return bt
def setUp(self):
self.bst = BinaryTree()
with open(path) as f:
for line in f:
row = re.split('\t', line.splitlines()[0])
data = MoviesData(int(row[0]), row[1], row[2])
array.append(data)
end = time.time()
print("Loaded file: " + file + " -> contains " + str(len(array)) +
" rows of data")
print("Elapsed time: " + str(end - start) + " sec\n")
print("Inserting data into BinaryTree")
start = time.time()
tree = BinaryTree()
for data in array:
tree.add(data[1], data)
end = time.time()
print("Elapsed time: " + str(end - start) + " sec\n")
arrayOfObjectsToFind = []
for number in range(1000):
arrayOfObjectsToFind.append(array[random.randint(0, len(array) - 1)])
print("Find by shortName starting for 1000 random objects")
elapsedTime = 0
for data in arrayOfObjectsToFind:
start = time.time()
tree.findByShortName(data[1])
