Fix 'raise_input_error' inline with the bugfix in python/cpython#102978

Purdue-EBEC · Jan 13, 2025 · c11feab · c11feab
1 parent 2da0135
commit c11feab
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/generic_grader/utils/importer.py b/src/generic_grader/utils/importer.py
@@ -19,7 +19,7 @@ class InputError(Exception):
         """Custom Exception type."""
 
     @classmethod
-    def raise_input_error(cls):
+    def raise_input_error(cls, *args, **kwargs):
         """Raise our custom exception."""
         raise cls.InputError()
 

diff --git a/tests/utils/test_importer.py b/tests/utils/test_importer.py
@@ -68,6 +68,14 @@ def test_ignores_function_input(fix_syspath):
         "message": "Stuck at call to `input()` while importing `fake_func`",
         "object": "fake_func",
     },
+    {
+        # Tests the except block on line 51
+        "module": "fake_module",
+        "error": Importer.InputError,
+        "text": "input('foo', bar='spam')\nfake_func = lambda: None",
+        "message": "Stuck at call to `input()` while importing `fake_func`",
+        "object": "fake_func",
+    },
     {
         # Tests the except block on line 65
         "module": "fake_module",