Updated problem_06 in Project Euler (#2439)

* * rename variable
* fix type hint
* fix doctest

* added test function

* fixed import error

* fixup! Format Python code with psf/black push

Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
This commit is contained in:
Du Yuanchao
2020-09-17 17:37:53 +08:00
committed by GitHub
parent 86fb2991d5
commit 2de2267319
5 changed files with 42 additions and 14 deletions

View File

@ -16,7 +16,7 @@ numbers and the square of the sum.
import math
def solution(n):
def solution(n: int) -> int:
"""Returns the difference between the sum of the squares of the first n
natural numbers and the square of the sum.
@ -35,4 +35,7 @@ def solution(n):
if __name__ == "__main__":
import doctest
doctest.testmod()
print(solution(int(input().strip())))