[feat] add testing function

hopre/core.py

@@ -1,6 +1,6 @@
 """User interface for HoPRe"""
 
-import os
+import os, sys, getopt, json
 from swiplserver import PrologMQI, PrologThread
 from hopre.utils import assert_all_json, tokenize
 
@@ -155,5 +155,45 @@ def hopre() -> None:
                 print()
 
 
+def test(filename: str | None = None, encoding: str = "UTF-8") -> None:
+    """Perform all tests given in the JSON file.
+
+    :param filename: Path to JSON file, if provided
+    :param encoding: Encoding of JSON file
+    """
+    if not filename:
+        filename = os.path.join(BASE_DIR, "punTestSuite.json")
+
+    print(f"Running tests in {filename}...\n")
+
+    with open(filename, "r", encoding=encoding) as f:
+        testcases = json.load(f)
+
+    assert isinstance(testcases, list)
+
+    with PrologMQI(output_file_name=os.path.join(BASE_DIR, "output.log")) as mqi:
+        with mqi.create_thread() as pthread:
+            init_kb(pthread)
+
+            for tc in testcases:
+                assert "input" in tc.keys()
+                assert "joke" in tc.keys()
+
+                sentences = [tokenize(s) for s in tc["input"]]
+                result, hopre_ans = check_pun(sentences, pthread)
+
+                print(f'Provided input:\n{tc["input"]}')
+                print(f'Answer: {"Pun" if tc["joke"] else "Not a pun"}')
+                print(f'HoPRe: {"Pun" if hopre_ans else "Not a pun"}')
+                if hopre_ans:
+                    print(f"Analysis: {result}")
+                print(f'Status: {"SUCCESS" if hopre_ans == tc["joke"] else "FAIL"}')
+                print()
+
+
 if __name__ == "__main__":
-    hopre()
+    opts, _ = getopt.getopt(sys.argv[1:], "t", ["test"])
+    if opts:
+        test()
+    else:
+        hopre()