2 files changed, 8 insertions, 2 deletions
diff --git a/python/problems/while_and_if/buy_five/common.py b/python/problems/while_and_if/buy_five/common.py
index c7081c2..84be034 100644
--- a/python/problems/while_and_if/buy_five/common.py
+++ b/python/problems/while_and_if/buy_five/common.py
@@ -53,7 +53,10 @@ def test(python, code):
     for output, correct in zip(outputs, test_out):
         if string_almost_equal(output, correct):
             n_correct += 1
-    return n_correct, len(test_in)
+
+    passed = n_correct == len(test_in)
+    hints = [{'id': 'test_results', 'args': {'passed': n_correct, 'total': len(test_in)}}]
+    return passed, hints
 
 def hint(python, code):
     # run one test first to see if there are any exceptions
diff --git a/python/problems/while_and_if/competition/common.py b/python/problems/while_and_if/competition/common.py
index 79f7447..4be8b12 100644
--- a/python/problems/while_and_if/competition/common.py
+++ b/python/problems/while_and_if/competition/common.py
@@ -48,7 +48,10 @@ def test(python, code):
     for output, correct in zip(outputs, test_out):
         if string_almost_equal(output, correct):
             n_correct += 1
-    return n_correct, len(test_in)
+
+    passed = n_correct == len(test_in)
+    hints = [{'id': 'test_results', 'args': {'passed': n_correct, 'total': len(test_in)}}]
+    return passed, hints
 
 def hint(python, code):
     # run one test first to see if there are any exceptions