From 2fca429796202bfc3bc6ff9891589fe69c62e6dc Mon Sep 17 00:00:00 2001 From: MaximSmolskiy Date: Sun, 31 Oct 2021 03:27:34 +0300 Subject: [PATCH 1/2] Add solution doctests --- project_euler/problem_078/sol1.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/project_euler/problem_078/sol1.py b/project_euler/problem_078/sol1.py index f92cf0f4020c..dc7fcc4ceca6 100644 --- a/project_euler/problem_078/sol1.py +++ b/project_euler/problem_078/sol1.py @@ -21,6 +21,12 @@ def solution(number: int = 1000000) -> int: """ + >>> solution(1) + 1 + + >>> solution(9) + 14 + >>> solution() 55374 """ From d7131bb5cddf1c3a00aec30df57f19d737d8e87c Mon Sep 17 00:00:00 2001 From: MaximSmolskiy Date: Sun, 31 Oct 2021 03:29:00 +0300 Subject: [PATCH 2/2] Improve solution function --- project_euler/problem_078/sol1.py | 1 + 1 file changed, 1 insertion(+) diff --git a/project_euler/problem_078/sol1.py b/project_euler/problem_078/sol1.py index dc7fcc4ceca6..7e5938c4c466 100644 --- a/project_euler/problem_078/sol1.py +++ b/project_euler/problem_078/sol1.py @@ -40,6 +40,7 @@ def solution(number: int = 1000000) -> int: if index > i: break item += partitions[i - index] * sign + item %= number index += j if index > i: break