def test_square_of_sum_out_of_range(self): record = NumberRecord(number=0) self.assertIsNone(record.square_of_sum())
def test_square_of_sum_bigger(self): record = NumberRecord(number=10) self.assertEqual( record.square_of_sum(), 3025 )
def test_square_of_sum_huge(self): record = NumberRecord(number=101) self.assertEqual( record.square_of_sum(), 26532801 )
def test_square_of_sum_small(self): record = NumberRecord(number=3) self.assertEqual( record.square_of_sum(), 36 )