From 70de1a963610bae8d8a1ff6a09d1d17ce71837cd Mon Sep 17 00:00:00 2001
From: Timotej Lazar <timotej.lazar@fri.uni-lj.si>
Date: Tue, 9 Feb 2016 15:55:07 +0100
Subject: Fix test function call in monkey.edits

---
 monkey/edits.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'monkey')

diff --git a/monkey/edits.py b/monkey/edits.py
index 82e5066..dfdf164 100644
--- a/monkey/edits.py
+++ b/monkey/edits.py
@@ -57,7 +57,8 @@ def get_edits_from_trace(trace, test, id):
             if action.type == 'prolog_solve':
                 queries.add(action.query.rstrip(' .'))
             elif action.type == 'test':
-                correct = test(code)
+                passed, total = test(code)
+                correct = passed == total
                 submissions.add((code, correct))
                 if correct:
                     # Ignore actions after the first correct version.
-- 
cgit v1.2.1