RF: refactor to allow return of test results

Break up testing function to allow return of individual test results.

Author: matthew-brett

gofer/ok.py

@@ -207,9 +207,9 @@ def id_generator(size=6, chars=string.ascii_uppercase + string.digits):
     return ''.join(random.choice(chars) for _ in range(size))
 
 
-def grade_notebook(notebook_path, tests_glob=None):
+def run_nb_tests(notebook_path, tests_glob=None):
     """
-    Grade a notebook file & return grade
+    Run tests on a notebook and return score, test_results
     """
     try:
         # Lots of notebooks call grade_notebook in them. These notebooks are then
@@ -256,7 +256,14 @@ def grade_notebook(notebook_path, tests_glob=None):
 
     # avoid divide by zero error if there are no tests
     score = sum([r.grade for r in test_results])/max(len(test_results), 1)
+    return score, test_results
 
+
+def grade_notebook(notebook_path, tests_glob=None):
+    """
+    Grade a notebook file & return grade
+    """
+    score, test_results = run_nb_tests(notebook_path, tests_glob)
     # If within an IPython or Jupyter environment, display hints
     display_defined = False
     try: