def testdiff_squares_sums(self): self.assertEquals(0, problem006.diff_squares_sums(1)) self.assertEquals(4, problem006.diff_squares_sums(2)) self.assertEquals(22, problem006.diff_squares_sums(3)) self.assertEquals(170, problem006.diff_squares_sums(5)) self.assertEquals(2640, problem006.diff_squares_sums(10))
#!/usr/bin/env python import sys import problem006 print(problem006.diff_squares_sums(int(sys.argv[1])))