Add type hints and default args to problem 15 and 34 of project euler

project_euler/problem_15/sol1.py

@@ -6,33 +6,22 @@
 from math import factorial
 
 
-def lattice_paths(n):
+def solution(n: int = 20) -> int:
     """
-        Returns the number of paths possible in a n x n grid starting at top left
-        corner going to bottom right corner and being able to move right and down
-        only.
-
-    bruno@bruno-laptop:~/git/Python/project_euler/problem_15$ python3 sol1.py 50
-    1.008913445455642e+29
-    bruno@bruno-laptop:~/git/Python/project_euler/problem_15$ python3 sol1.py 25
-    126410606437752.0
-    bruno@bruno-laptop:~/git/Python/project_euler/problem_15$ python3 sol1.py 23
-    8233430727600.0
-    bruno@bruno-laptop:~/git/Python/project_euler/problem_15$ python3 sol1.py 15
-    155117520.0
-    bruno@bruno-laptop:~/git/Python/project_euler/problem_15$ python3 sol1.py 1
-    2.0
-
-        >>> lattice_paths(25)
-        126410606437752
-        >>> lattice_paths(23)
-        8233430727600
-        >>> lattice_paths(20)
-        137846528820
-        >>> lattice_paths(15)
-        155117520
-        >>> lattice_paths(1)
-        2
+    Returns the number of paths possible in a n x n grid starting at top left
+    corner going to bottom right corner and being able to move right and down
+    only.
+
+    >>> solution(25)
+    126410606437752
+    >>> solution(23)
+    8233430727600
+    >>> solution(20)
+    137846528820
+    >>> solution(15)
+    155117520
+    >>> solution(1)
+    2
 
     """
     n = 2 * n  # middle entry of odd rows starting at row 3 is the solution for n = 1,
@@ -46,10 +35,10 @@ def lattice_paths(n):
     import sys
 
     if len(sys.argv) == 1:
-        print(lattice_paths(20))
+        print(solution(20))
     else:
         try:
             n = int(sys.argv[1])
-            print(lattice_paths(n))
+            print(solution(n))
         except ValueError:
             print("Invalid entry - please enter a number.")

project_euler/problem_34/sol1.py

@@ -18,7 +18,7 @@ def sum_of_digit_factorial(n: int) -> int:
     return sum(factorial(int(char)) for char in str(n))
 
 
-def compute() -> int:
+def solution() -> int:
     """
     Returns the sum of all numbers whose
     sum of the factorials of all digits
@@ -31,4 +31,4 @@ def compute() -> int:
 
 
 if __name__ == "__main__":
-    print(f"{compute()} = ")
+    print(f"{solution()} = ")