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("=============================")