def test_minimax_1(self): t = TTT(3) state = [[1, -1, 0], [-1, 1, 0], [0, 0, 0]] state = np.array(state, dtype='int') state = state.reshape(-1) t._state = state [score, move] = minimax(t.get_state(), 1, t) self.assertListEqual(list(state), list(t.get_state())) self.assertEqual(8, move) self.assertEqual(5, score)
def test_alpha_beta_1(self): t = TTT(3) player = ABPruning(3) state = [[1, -1, 0], [-1, 1, 0], [0, 0, 0]] state = np.array(state, dtype='int') state = state.reshape(-1) t._state = state [score, move] = player.get(t.get_state(), 1) self.assertListEqual(list(state), list(t.get_state())) self.assertEqual(8, move) self.assertEqual(5, score)