Merge pull request #16 from gradescope/ibrahim/ch24859/instructors-ca…

…n-set-late-penalties-in-an

Instructors can override results (e.g. set late penalties) in a post-processor function

docs/conf.py

@@ -184,3 +184,6 @@
 }
 
 source_suffix = ['.rst', '.md']
+
+# Document constructors as well as class doc strings
+autoclass_content = 'both'

gradescope_utils/autograder_utils/json_test_runner.py

@@ -127,15 +127,22 @@ class JSONTestRunner(object):
 
     def __init__(self, stream=sys.stdout, descriptions=True, verbosity=1,
                  failfast=False, buffer=True, visibility=None,
-                 stdout_visibility=None):
+                 stdout_visibility=None, post_processor=None):
         """
         Set buffer to True to include test output in JSON
+
+
+        post_processor: if supplied, will be called with the final JSON
+        data before it is written, allowing the caller to overwrite the
+        test results (e.g. add a late penalty) by editing the results
+        dict in the first argument.
         """
         self.stream = stream
         self.descriptions = descriptions
         self.verbosity = verbosity
         self.failfast = failfast
         self.buffer = buffer
+        self.post_processor = post_processor
         self.json_data = {}
         self.json_data["tests"] = []
         self.json_data["leaderboard"] = []
@@ -174,6 +181,9 @@ def run(self, test):
             total_score += test["score"]
         self.json_data["score"] = total_score
 
+        if self.post_processor is not None:
+            self.post_processor(self.json_data)
+
         json.dump(self.json_data, self.stream, indent=4)
         self.stream.write('\n')
         return result