diff options
Diffstat (limited to 'prolog/problems/world_data/flows_2/common.py')
-rw-r--r-- | prolog/problems/world_data/flows_2/common.py | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/prolog/problems/world_data/flows_2/common.py b/prolog/problems/world_data/flows_2/common.py new file mode 100644 index 0000000..a3cf79b --- /dev/null +++ b/prolog/problems/world_data/flows_2/common.py @@ -0,0 +1,51 @@ +from operator import itemgetter +import socket +import prolog.engine +import prolog.util +from server.hints import Hint +import server.problems + +id = 10009 +number = 90 +visible = True +facts = 'mondial' + +solution = '''\ +flows(River, Sea) :- + river(River, _, _, Sea, _, _, _, _, _, _, _, _), + Sea \= null. +flows(River, Sea) :- + river(River, River2, _, _, _, _, _, _, _, _, _, _), + River2 \= null, + flows(River2, Sea). +''' + +hint_type = { +} + +test_cases = [ + ("flows('Rhone', X)", [{'X': "'Mediterranean Sea'"}]), + ("flows(X, 'Java Sea')", [{'X': "'Mahakam'"}, {'X': "'Barito'"}]), + ("flows('Drau', X)", [{'X': "'Black Sea'"}]), +] + +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=2.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 |