Update test_helpers.py

1d0f6100 · Adam Blank · 4a7bcfcd · 1d0f6100
Commit 1d0f6100 authored 3 months ago by Adam Blank
Hide whitespace changes
Inline Side-by-side

Showing

with 10 additions and 3 deletions
+10 -3
--- a/tests/helpers/test_helpers.py
+++ b/tests/helpers/test_helpers.py
@@ -27,10 +27,17 @@ def myprint(*args, **kwargs):
    output += " ".join(str(arg) for arg in args) + "\n"
+class OutOfInputError(Exception):
+    pass
 def myinput(prompt):
-    global input_index
+    try:
-    response = input_list[input_index]
+        global input_index
-    input_index += 1
+        response = input_list[input_index]
+        input_index += 1
+    except IndexError:
+        raise OutOfInputError
    return response