better error messages

Author: stgm

tests/_static_analysis.py

@@ -1,8 +1,29 @@
+from checkpy import *
 from typing import Any
 import ast
 
 from checkpy import static
 
+def defines_function(name: str) -> bool:
+    check = name in static.getFunctionDefinitions()
+    if not check:
+        raise AssertionError(f"`{name}` is niet aanwezig")
+    return check
+
+def not_in_code(construct: type):
+    check = construct in static.AbstractSyntaxTree()
+    name = str(construct).split(".")[1].split("'")[0].lower()
+    if check:
+        raise AssertionError(f"`{name}` mag niet gebruikt worden in deze opdracht")
+    return check
+
+def in_code(construct: type):
+    check = construct in static.AbstractSyntaxTree()
+    name = str(construct).split(".")[1].split("'")[0].lower()
+    if not check:
+        raise AssertionError(f"`{name}` moet gebruikt worden in deze opdracht")
+    return check
+
 def has_syntax_error():
     try:
         compile(static.getSource(), "<your program>", "exec")

tests/acidTest.py

@@ -1,10 +1,11 @@
 from checkpy import *
 from _basics_no_listcomp import *
+from _static_analysis import *
 
 @t.passed(doctest_ok)
 def has_functions():
     """functie `is_acidic` is aanwezig"""
-    assert "is_acidic" in static.getFunctionDefinitions(), "`is_acidic` is niet aanwezig"
+    assert defines_function("is_acidic")
 
 @t.passed(has_functions)
 def test_weeks_elapsed(test):

tests/colaTest.py

@@ -3,14 +3,15 @@
 import checkpy.assertlib as asserts
 
 from _basics_no_listcomp import *
+from _static_analysis import *
 
 @t.passed(doctest_ok)
 def has_functions():
     """alle gevraagde functies zijn aanwezig"""
-    assert "check_coin" in static.getFunctionDefinitions(), "`check_coin` is niet aanwezig"
-    assert "determine_due" in static.getFunctionDefinitions(), "`determine_due` is niet aanwezig"
-    assert "prompt_coin" not in static.getFunctionDefinitions(), "`prompt_coin` is aanwezig, maar dat staat niet in de opdracht"
-    assert ast.While in static.AbstractSyntaxTree(), "er wordt geen gebruik gemaakt van een `while`-loop"
+    assert defines_function("check_coin")
+    assert defines_function("determine_due")
+    assert defines_function("prompt_coin")
+    assert in_code(ast.While)
 
 @t.passed(doctest_ok)
 @t.test(10)

tests/fdrTest.py

@@ -1,15 +1,16 @@
 from checkpy import *
 from _basics_no_listcomp import *
+from _static_analysis import *
 
 @t.passed(doctest_ok)
 def has_functions():
     """alle gevraagde functies zijn aanwezig"""
-    assert "pie_percent" in static.getFunctionDefinitions(), "`pie_percent` is niet aanwezig"
-    assert "triple" in static.getFunctionDefinitions(), "`triple` is niet aanwezig"
-    assert "absdiff" in static.getFunctionDefinitions(), "`absdiff` is niet aanwezig"
-    assert "kmmiles" in static.getFunctionDefinitions(), "`kmmiles` is niet aanwezig"
-    assert "avg3" in static.getFunctionDefinitions(), "`avg3` is niet aanwezig"
-    assert "avg3of4" in static.getFunctionDefinitions(), "`avg3of4` is niet aanwezig"
+    assert defines_function("pie_percent")
+    assert defines_function("triple")
+    assert defines_function("absdiff")
+    assert defines_function("kmmiles")
+    assert defines_function("avg3")
+    assert defines_function("avg3of4")
 
 @t.passed(has_functions)
 def test_pie_percent(test):

tests/string_functionsTest.py

@@ -7,12 +7,11 @@
 @t.passed(doctest_ok)
 def has_functions():
     """alle gevraagde functies zijn aanwezig"""
-    assert "repeat" in static.getFunctionDefinitions(), "`repeat` is niet aanwezig"
-    assert "total_length" in static.getFunctionDefinitions(), "`total_length` is niet aanwezig"
-    
-    assert ast.If not in static.AbstractSyntaxTree()
-    assert ast.While not in static.AbstractSyntaxTree()
-    assert ast.For not in static.AbstractSyntaxTree()
+    assert defines_function("repeat")
+    assert defines_function("total_length")
+    assert not_in_code(ast.If)
+    assert not_in_code(ast.While)
+    assert not_in_code(ast.For)
 
 @t.passed(has_functions)
 def test_repeat(test):

tests/wekenTest.py

@@ -1,10 +1,11 @@
 from checkpy import *
 from _basics_no_listcomp import *
+from _static_analysis import *
 
 @t.passed(doctest_ok)
 def has_functions():
     """functie `weeks_elapsed` is aanwezig"""
-    assert "weeks_elapsed" in static.getFunctionDefinitions(), "`weeks_elapsed` is niet aanwezig"
+    assert defines_function("weeks_elapsed")
 
 @t.passed(has_functions)
 def test_weeks_elapsed(test):