From 813a7ccf701a51f49a1482165daeb0998bc40865 Mon Sep 17 00:00:00 2001 From: wuliaozhiji Date: Sat, 19 Feb 2022 14:59:05 -0500 Subject: [PATCH] [+] A2 P2.1 Test edge case --- assignments/A2/a2_part2.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/assignments/A2/a2_part2.py b/assignments/A2/a2_part2.py index b25fa6b..494f8b0 100644 --- a/assignments/A2/a2_part2.py +++ b/assignments/A2/a2_part2.py @@ -62,11 +62,24 @@ def generate_complete_game_tree(root_move: str, game_state: a2_minichess.Miniche >>> t = generate_complete_game_tree('*', g, 2) >>> {sub.move for sub in t.get_subtrees()} == {'a2b3', 'b2c3', 'b2a3', 'c2d3', 'c2b3', 'd2c3'} True - >>> t: a2_game_tree.GameTree = t.find_subtree_by_move('c2d3') + >>> t = t.find_subtree_by_move('c2d3') >>> len(t.get_subtrees()) 1 >>> t.get_subtrees()[0].move 'd4d3' + >>> g = a2_minichess.MinichessGame() + >>> g.make_move('a2b3') + >>> g.make_move('b4b3') + >>> g.make_move('c2b3') + >>> g.make_move('c4b3') + >>> g.make_move('b2c3') + >>> g.make_move('b3c4') + >>> t = generate_complete_game_tree('b3c4', g, 3) + >>> len(t.get_subtrees()) + 9 + >>> t = t.find_subtree_by_move('b1a2') + >>> [sub.move for sub in t.get_subtrees()] + [] """ tree = a2_game_tree.GameTree(root_move, game_state.is_white_move())