def test_cur_state_when_opponent_should_move_to_start(self):
game = TicTacToe([['X', 'O', 'O'], ['X', 'O', ' '], [' ', 'X', 'X']])
ab = AlphaBeta()
mdp = FixedGameMDP(game, ab, 1)
env = Environment(mdp)
expected = TicTacToe([['X', 'O', 'O'], ['X', 'O', ' '],
['O', 'X', 'X']])
self.assertEqual(env.cur_state(), expected)
def test_do_action(self):
# X - O
# - - X
# - - O
game = TicTacToe().make_moves([1, 3, 6, 9])
mdp = FixedGameMDP(game.copy(), AlphaBeta(), 1)
env = Environment(mdp)
env.do_action(7)
expected = TicTacToe().make_moves([1, 3, 6, 9, 7, 4])
self.assertEqual(env.cur_state(), expected)
def test_cur_state(self):
game = TicTacToe()
mdp = FixedGameMDP(game, AlphaBeta(), 1)
env = Environment(mdp)
self.assertEqual(env.cur_state(), mdp.start_state())
self.assertEqual(env.cur_state(), game)