def test_sum_of_squares_huge(self):
# Model supports numbers > 100. Frontend does not
record = NumberRecord(number=101)
self.assertEqual(
record.sum_of_squares(),
348551
)
def test_sum_of_squares_out_of_range(self):
record = NumberRecord(number=0)
self.assertIsNone(record.sum_of_squares())
def test_sum_of_squares_bigger(self):
record = NumberRecord(number=10)
self.assertEqual(
record.sum_of_squares(),
385
)
def test_sum_of_squares_small(self):
record = NumberRecord(number=3)
self.assertEqual(
record.sum_of_squares(),
14
)
