From 806ad621f30800295302b95ebe73912b17e2188d Mon Sep 17 00:00:00 2001 From: wuliaozhiji Date: Sat, 19 Feb 2022 15:19:30 -0500 Subject: [PATCH] [+] A2 P2.2 Add more tests --- assignments/A2/a2_part2.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/assignments/A2/a2_part2.py b/assignments/A2/a2_part2.py index a7e028e..9de3b75 100644 --- a/assignments/A2/a2_part2.py +++ b/assignments/A2/a2_part2.py @@ -75,14 +75,21 @@ def generate_complete_game_tree(root_move: str, game_state: a2_minichess.Miniche >>> g.make_move('b2c3') >>> g.make_move('b3c4') >>> t = generate_complete_game_tree('b3c4', g, 3) + >>> t.white_win_probability + 1.0 >>> len(t.get_subtrees()) 9 >>> t = t.find_subtree_by_move('b1a2') >>> len(t.get_subtrees()) 0 + >>> t.white_win_probability + 1.0 + >>> t = generate_complete_game_tree('*', g, 4) + >>> t.white_win_probability + 1.0 """ tree = a2_game_tree.GameTree(root_move, game_state.is_white_move(), - int(game_state.get_winner() == 'White')) + float(game_state.get_winner() == 'White')) # Base case if d == 0: