def test_sum_of_squares(self): from examples.sum_of_squares import test_calling_fn for i in range(5): self.assertEqual(test_calling_fn(i), sum([j * j for j in range(i)]))
def test_square(self): from examples.square import test_calling_fn for i in range(5): self.assertEqual(test_calling_fn(i), i * i)