def test_maximumTotal_return_true_result(self): triangle = [[6],[3,5],[9,7,1],[4,6,8,4]] T = Triangle(triangle) (val, path) = T.maximumTotal() self.assertEqual(val,26) self.assertEqual(path, [0, 1, 1, 2] )
def test_maximumTotal_return_true_result2(self):
triangle = [[6],[3,5]]
T = Triangle(triangle)
(val, path) = T.maximumTotal()
self.assertEqual(val,11)
self.assertEqual(path, [0, 1] )
def test_if_list_is_empty(self): triangle = [] T = Triangle(triangle) (res) = T.maximumTotal() self.assertEqual(res,False)
#!/usr/bin/env python
import sys
import time
from triangle import Triangle
arr = [[6],[3,5],[9,7,1],[4,6,8,4]]
t = time.time()
Triangle = Triangle(arr)
(sum,path) = Triangle.maximumTotal()
t = time.time() - t
print("========== results ==========")
print(" triangle: %s " % (arr))
print(" maximum total: %d" % (sum))
print(" path: %s " % (path))
print(" execution time: %fs" % (t))
print("=============================")
