def test_10(self): solution_input = 200 expected = 487067745 result = solution_grandest_staircase.solution(solution_input) self.assertEqual(result, expected)
def test_9(self): solution_input = 137 expected = 7755775 result = solution_grandest_staircase.solution(solution_input) self.assertEqual(result, expected)
def test_1(self): solution_input = 3 expected = 1 result = solution_grandest_staircase.solution(solution_input) self.assertEqual(result, expected)
def test_8(self): solution_input = 75 expected = 48445 result = solution_grandest_staircase.solution(solution_input) self.assertEqual(result, expected)