Fix fail due to number in prompt bug

Makaze · Dec 1, 2023 · 12bbc64 · 12bbc64
1 parent b2bec75
commit 12bbc64
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/einstein/__init__.py b/einstein/__init__.py
@@ -14,10 +14,10 @@ def test1():
     output = check50.run("python3 einstein.py").stdin("1", prompt=False).stdout()
 
     # Extract number from stdout
-    match = re.search(r"([.,]?(?:\d[.,]?)+)", output)
+    match = re.findall(r"([.,]?(?:\d[.,]?)+)", output)
     if match is None:
         raise check50.Failure("Looks like your program didn't output a number!")
-    number = match.group(0)
+    number = match[-1]
 
     # Match correct number
     if not re.match(r"^90(?:,?0{3}){5}(?:\.0+)?$", number) and not re.match(