assertions

cornell-dti · Oct 23, 2023 · 75acb6a · 75acb6a
1 parent 4d13bda
commit 75acb6a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/test_prereq_parse.py b/python/test_prereq_parse.py
@@ -92,6 +92,7 @@ def test_raw_prereqs_coreqs(verbose=False):
         elif verbose:
             print(f"WRONG Answer: {answer} Got: {response}")
     print(f"{correct / total * 100}%")
+    assert correct / total * 100 == 100
 
 
 @pytest.mark.skipif(os.environ.get("RUN_LANGCHAIN_TESTS") == "false", reason="skip")
@@ -174,3 +175,4 @@ def test_prereqs_coreqs(index=None, shorten=False, verbose=False, hard=False):
         elif verbose:
             print(f"WRONG Answer: {answer} Got: {response}")
     print(f"{correct / total * 100}%")
+    assert correct / total * 100 == 100