Python: add support for auxiliary code (like for Prolog)

1 files changed, 4 insertions, 4 deletions

diff --git a/python/problems/introduction/fast_fingers_2/common.py b/python/problems/introduction/fast_fingers_2/common.py
index 01ad941..fe50111 100644
--- a/python/problems/introduction/fast_fingers_2/common.py
+++ b/python/problems/introduction/fast_fingers_2/common.py
@@ -33,7 +33,7 @@ hint_type = {
     'final_hint_noif': Hint('final_hint_noif')
 }
 
-def test(python, code):
+def test(python, code, aux_code=''):
     # List of inputs: (expression to eval, stdin).
     test = [(5,'25\n'),
             (6, '29\n'),
@@ -58,7 +58,7 @@ def test(python, code):
         # hook randint
         tcode = random_code.format(t[0]) + code
 
-        answers = python(code=tcode, inputs=[(None, t[1])], timeout=1.0)
+        answers = python(code=aux_code+tcode, inputs=[(None, t[1])], timeout=1.0)
         output = answers[0][1]
 
         if str(test_out[ti]) in output and str(not test_out[ti]) not in output:
@@ -81,12 +81,12 @@ def test(python, code):
             
     return passed, hints
 
-def hint(python, code):
+def hint(python, code, aux_code=''):
     tokens = get_tokens(code)
 
     # run one test first to see if there are any exceptions
     test_in = [(None, '16\n')]
-    answer = python(code=code, inputs=test_in, timeout=1.0)
+    answer = python(code=aux_code+code, inputs=test_in, timeout=1.0)
     exc = answer[0][3]
     exc_hint = get_exception_desc(answer[0][3])
     # if have an exception!