Add prolog.engine.test

Depends on run_tests/3.

1 files changed, 13 insertions, 0 deletions

diff --git a/prolog/engine.py b/prolog/engine.py
index a25a07d..dff577c 100644
--- a/prolog/engine.py
+++ b/prolog/engine.py
@@ -114,6 +114,19 @@ class PrologEngine(object):
         message = re.sub(r'_G[0-9]*', '_', message)
         return message
 
+def test(name, code):
+    engine = PrologEngine(code=code)
+    reply = engine.ask("run_tests({}, '{}', Results)".format(name, engine.id))
+    engine.destroy()
+
+    if reply['event'] != 'success':
+        raise Exception('testing procedure failed')
+
+    results = re.findall(r'(?:success|failure)\([^)]*\)', reply['data'][0]['Results'])
+    n_total = len(results)
+    n_passed = len([r for r in results if r.startswith('success')])
+    return (n_passed, n_total)
+
 # Basic sanity check.
 if __name__ == '__main__':
     engine = PrologEngine(code='dup([],[]). dup([H|T],[H,H|TT]) :- dup(T,TT).')