fix override on root state, add check_func test

Author: machow

pythonwhat/check_funcs.py

@@ -216,11 +216,12 @@ def override(solution, state=None):
                 new_ast = node
                 break
 
+    kwargs  = state.messages[-1] if state.messages else {}
     child = state.to_child_state(
             solution_subtree = new_ast,
             student_subtree = state.student_tree,
             highlight = state.highlight,
-            append_message = {'msg': "", 'kwargs': state.messages[-1]['kwargs']}
+            append_message = {'msg': "", 'kwargs': kwargs}
             )
 
     return child

tests/test_test_function.py

@@ -757,17 +757,26 @@ def test_pass_sig_false(self):
         sct_payload = helper.run(self.data)
         self.assertTrue(sct_payload['correct'])
 
-    @unittest.skip("TODO: implement override")
     def test_pass_sig_false_override(self):
         self.setup_color()
-        self.data["DC_SCT"].replace('color', 'c')
+        self.data["DC_CODE"] = self.data["DC_CODE"].replace('color', 'c')
         self.data['DC_SCT'] =  """
-Ex().check_function('f', 0, signature=False).override("f(c = 'blue')").check_args('c').has_equal_ast()
+Ex().override("f(c = 'blue')").check_function('f', 0, signature=False).check_args('c').has_equal_ast()
 """
 
         sct_payload = helper.run(self.data)
         self.assertTrue(sct_payload['correct'])
 
+    @unittest.skip("TODO: override code isn't parsed, so can't get args part")
+    def test_pass_sig_false_override_after_check(self):
+        self.setup_color()
+        self.data["DC_CODE"] = self.data["DC_CODE"].replace('color', 'c')
+        self.data['DC_SCT'] =  """
+Ex().check_function('f', 0, signature=False).override("f(c = 'blue')").check_args('c').has_equal_ast()
+"""
+        sct_payload = helper.run(self.data)
+        self.assertTrue(sct_payload['correct'])
+
 
 class TestFunctionComplexArgs(unittest.TestCase):
     def setUp(self):