Resolves cs50#174 for all test cases.

Makaze · Dec 1, 2023 · 0943536 · 0943536
1 parent 12bbc64
commit 0943536
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/einstein/__init__.py b/einstein/__init__.py
@@ -36,10 +36,10 @@ def test14():
     output = check50.run("python3 einstein.py").stdin("14", 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"^1,?260(?:,?0{3}){5}(?:\.0+)?$", number) and not re.match(
@@ -58,10 +58,10 @@ def test50():
     output = check50.run("python3 einstein.py").stdin("50", 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"^4,?500(?:,?0{3}){5}(?:\.0+)?$", number) and not re.match(