def test_difference_of_squares_with_large_val(self): for n, exp in [ (5_000, 156270827082500), (4_000_000, 64000010666662666666000000), (900_000_000_000_000_000_000, 164_025_000_000_000_000_000_121_499_999_999_999_999_999_797_499_999_999_999_999_999_850_000_000_000_000_000_000 ) ]: self.assertEqual(difference_of_squares(n), exp)
def test_difference_of_squares_100(self): self.assertEqual(difference_of_squares(100), 25164150)
def test_difference_of_squares_5(self): self.assertEqual(difference_of_squares(5), 170)
def test_difference_of_squares_1(self): self.assertEqual(difference_of_squares(1), 0)
def test_difference_of_squares_100(self): assert difference_of_squares(100) == 25164150
def test_difference_of_squares_5(self): assert difference_of_squares(5) == 170
def test_difference_of_squares_1(self): assert difference_of_squares(1) == 0