diff options
Diffstat (limited to 'prolog/problems/world_data/all_capitals_1/common.py')
-rw-r--r-- | prolog/problems/world_data/all_capitals_1/common.py | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/prolog/problems/world_data/all_capitals_1/common.py b/prolog/problems/world_data/all_capitals_1/common.py new file mode 100644 index 0000000..490c183 --- /dev/null +++ b/prolog/problems/world_data/all_capitals_1/common.py @@ -0,0 +1,44 @@ +from operator import itemgetter +import socket +import prolog.engine +import prolog.util +from server.hints import Hint +import server.problems + +id = 10002 +number = 20 +visible = True +facts = 'mondial' + +solution = '''\ +all_capitals(List) :- + findall(Capital, country(_, _, Capital, _, _, _), List). +''' + +hint_type = { +} + +test_cases = [ + ('all_capitals(X), length(X, L)', [{'L': '244'}]), +] + +def test(code, aux_code): + n_correct = 0 + engine_id = None + try: + engine_id, output = prolog.engine.create(code=code+aux_code, timeout=5.0) + if engine_id is not None and 'error' not in map(itemgetter(0), output): + # Engine successfully created, and no syntax error in program. + for query, answers in test_cases: + if prolog.engine.check_answers(engine_id, query=query, answers=answers, timeout=1.0, inference_limit=None): + n_correct += 1 + except socket.timeout: + pass + finally: + if engine_id: + prolog.engine.destroy(engine_id) + + hints = [{'id': 'test_results', 'args': {'passed': n_correct, 'total': len(test_cases)}}] + if n_correct == len(test_cases): + hints += [{'id': 'final_hint'}] + return n_correct, len(test_cases), hints |