diff options
| -rw-r--r-- | monkey/edits.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/monkey/edits.py b/monkey/edits.py index f087074..829fcb6 100644 --- a/monkey/edits.py +++ b/monkey/edits.py @@ -207,6 +207,11 @@ if __name__ == '__main__': ] edits, submissions, queries = {}, {}, {} + try: + test_results = pickle.load(open('test_results.pickle', 'rb')) + except: + test_results = collections.defaultdict(dict) + for problem in Problem.list(): pid = problem.id solutions = [s for s in Solution.filter(problem_id=pid, done=True) @@ -226,8 +231,15 @@ if __name__ == '__main__': dependencies = sorted([p[2:] for p in other_problems if p.identifier in used_predicate_identifiers]) + # Check for cached results. + normal_code = stringify(rename_vars(tokenize(code))) + code_key = (normal_code, tuple(dependencies)) + if code_key in test_results[pid]: + return test_results[pid][code_key] + aux_code = '\n' + solutions_for_problems('prolog', dependencies) + '\n' + facts correct, hints = problem_module.test(code, aux_code) + test_results[pid][code_key] = correct return correct print('Analyzing traces for {}… '.format(problem.identifier), end='', flush=True) @@ -241,3 +253,4 @@ if __name__ == '__main__': traceback.print_exc() pickle.dump((edits, submissions, queries), open('edits.pickle', 'wb')) + pickle.dump(test_results, open('test_results.pickle', 'wb')) |