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))
Example #2
0
#!/usr/bin/env python

import sys

import problem006

print(problem006.diff_squares_sums(int(sys.argv[1])))