From 76802c10b14c056601623bdce606a63d9df9aaff Mon Sep 17 00:00:00 2001
From: Martin Thoma <info@martin-thoma.de>
Date: Sat, 2 Apr 2022 08:26:40 +0200
Subject: [PATCH] Mugwump (Python): Add types

---
 62_Mugwump/python/mugwump.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/62_Mugwump/python/mugwump.py b/62_Mugwump/python/mugwump.py
index 6eb63ccc..d1e3f7c3 100644
--- a/62_Mugwump/python/mugwump.py
+++ b/62_Mugwump/python/mugwump.py
@@ -1,5 +1,6 @@
 from math import sqrt
 from random import randint
+from typing import List, Tuple
 
 
 def introduction() -> None:
@@ -18,7 +19,7 @@ you how far you are from each mugwump."""
     )
 
 
-def generate_mugwumps(n=4):
+def generate_mugwumps(n: int = 4) -> List[List[int]]:
     mugwumps = []
     for _ in range(n):
         current = [randint(0, 9), randint(0, 9)]
@@ -26,14 +27,14 @@ def generate_mugwumps(n=4):
     return mugwumps
 
 
-def reveal_mugwumps(mugwumps):
+def reveal_mugwumps(mugwumps: List[List[int]]) -> None:
     print("Sorry, that's 10 tries.  Here's where they're hiding.")
     for idx, mugwump in enumerate(mugwumps, 1):
         if mugwump[0] != -1:
             print(f"Mugwump {idx} is at {mugwump[0]},{mugwump[1]}")
 
 
-def calculate_distance(guess, mugwump):
+def calculate_distance(guess: Tuple[int, int], mugwump: List[int]) -> float:
     d = sqrt(((mugwump[0] - guess[0]) ** 2) + ((mugwump[1] - guess[1]) ** 2))
     return d
 
@@ -47,7 +48,7 @@ def play_again() -> None:
         exit()
 
 
-def play_round():
+def play_round() -> None:
     mugwumps = generate_mugwumps()
     turns = 1
     score = 0