Multiple test cases passed as dicts, check_code() is iterated based on no. of test cases

1 files changed, 2 insertions, 2 deletions

diff --git a/yaksh/python_stdout_evaluator.py b/yaksh/python_stdout_evaluator.py
index 815752f..9443e37 100644
--- a/yaksh/python_stdout_evaluator.py
+++ b/yaksh/python_stdout_evaluator.py
@@ -25,11 +25,11 @@ def redirect_stdout():
 class PythonStdoutEvaluator(CodeEvaluator):
     """Tests the Python code obtained from Code Server"""
 
-    def check_code(self, user_answer, test_case_data):
+    def check_code(self, user_answer, expected_output):
         success = False
 
         tb = None
-        expected_output = test_case_data[0]
+        # expected_output = test_case_data[0]
         submitted = compile(user_answer, '<string>', mode='exec')
         with redirect_stdout() as output_buffer:
             g = {}