From e019b29657a69c19121c6a145d065b7e23baf093 Mon Sep 17 00:00:00 2001 From: Marianne Chevrot Date: Sun, 17 May 2020 11:09:15 +0200 Subject: add a test to grid generator --- tests_solver/test_solver.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) (limited to 'tests_solver') diff --git a/tests_solver/test_solver.py b/tests_solver/test_solver.py index 1f178cc..1d441ff 100644 --- a/tests_solver/test_solver.py +++ b/tests_solver/test_solver.py @@ -195,3 +195,36 @@ def test_kana_arrow_swap(): else: assert kanagrid_test is None + +def test_solver_load_and_type_1(): + # ./solver.py tests_solver/level_test_type.yaml + + kanagrid_orig = KanaGrid((6, 2), [ + Kana("ar_r", "sa"), Kana("norm" ), Kana("norm", "so"), Kana("rock", "se"), Kana("myst", "ka"), Kana("ar_d", "na"), + Kana("ar_u", "su"), Kana("ar_l", "nu"), Kana("void" ), Kana("void" ), Kana("froz" ), Kana("froz", "ke"), + ]) + + expected_grid = KanaGrid((6, 2), [ + Kana("ar_u", "su"), Kana("ar_r", "sa"), Kana("norm", "so"), Kana("rock", "se"), Kana("froz", "ke"), Kana("norm", "ka"), + Kana("ar_l", "nu"), Kana("norm" ), Kana("void" ), Kana("void" ), Kana("froz" ), Kana("ar_d", "na"), + ], action_count=6, score=8) + + type_yaml_level_file = 'tests_solver/level_test_type.yaml' + with open(type_yaml_level_file, encoding='utf8') as stream: + input_dict = yaml.safe_load(stream) + loaded_grid = KanaGrid.load(input_dict) + target_score = input_dict['target_score'] + max_actions = input_dict['max_actions'] + + mock_args = {'--help': False, + '--print': False, + '-p': False, + 'YAML_GRID': type_yaml_level_file, + } + + assert loaded_grid == kanagrid_orig + + kanagrid_new = search_all_solution(mock_args, loaded_grid, target_score, max_actions) + + assert kanagrid_new == expected_grid + -- cgit v1.2.3