def test_square_of_sum(self): self.assertEqual(my_module.square_of_sum(1, 3), 36)
import my_module sum_of_squares = my_module.sum_of_squares(1, 100) square_of_sum = my_module.square_of_sum(1, 100) print square_of_sum - sum_of_squares