Merge pull request #154 from datacamp/fix-test-or

fix probe arg partialing for nested test_or

Author: Filip Schouwenaars

pythonwhat/probe.py

@@ -113,7 +113,8 @@ def __iter__(self):
 
     def partial(self):
         """Return partial of original function call"""
-        return partial(self.data['func'], **self.data['bound_args'].arguments)
+        ba = self.data['bound_args']
+        return partial(self.data['func'], *ba.args, **ba.kwargs)
 
     def update_child_calls(self):
         """Replace child nodes on original function call with their partials"""

tests/test_test_or.py

@@ -48,5 +48,17 @@ def test_Fail1_no_lam(self):
         self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"])
         self.test_Fail1()
 
+class TestNestedOr(unittest.TestCase):
+    def setUp(self):
+        self.data = {
+                "DC_SOLUTION": """for ii in range(10): print('yes') if ii < 2 else print('no')""",
+                "DC_SCT": """test_for_loop(1, body=test_if_exp(body=test_or(test_student_typed('print'))))"""
+                }
+
+    def test_pass(self):
+        self.data["DC_CODE"] = self.data["DC_SOLUTION"]
+        sct_payload = helper.run(self.data)
+        self.assertTrue(sct_payload['correct'])
+
 if __name__ == "__main__":
     unittest.main()