Pownie/backgammon
1
0
Fork 0
Code Issues 6 Pull Requests Packages Projects Releases Wiki Activity
2312c9cb2a
backgammon/test.py

975 lines
30 KiB
Python
Raw Blame History

import unittest
from board import Board
class TestIsMoveValid(unittest.TestCase):
def setUp(self):
self.initial = Board.initial_state
def test_simple_movement(self):
self.assertEqual(Board.is_move_valid(self.initial, 1, 2, (1, 3)), True) # White
self.assertEqual(Board.is_move_valid(self.initial, -1, 3, (24, 21)), True) # Black
def test_backwards_movement_invalid(self):
self.assertEqual(Board.is_move_valid(self.initial, 1, 2, (12, 10)), False)
self.assertEqual(Board.is_move_valid(self.initial, -1, 3, (8, 11)), False)
def test_face_value_match_move_length(self):
self.assertEqual(Board.is_move_valid(self.initial, 1, 2, (1, 3)), True)
self.assertEqual(Board.is_move_valid(self.initial, 1, 2, (1, 4)), False)
def test_bear_in(self):
board = ( 1,
1, 0, 0, 0, 0, -5,
0, -3, 0, 0, 0, 5,
-5, 0, 0, 0, 3, 0,
5, 0, 0, 0, 0, -2,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 2, (0, 2)), True)
board = ( 0,
2, 0, 0, 0, 0, -5,
0, -3, 0, 0, 0, 5,
-5, 0, 0, 0, 3, 0,
5, 0, 0, 0, 0, -1,
-1 )
self.assertEqual(Board.is_move_valid(board, -1, 2, (25, 23)), True)
def test_force_bear_in(self):
board = ( 1,
1, 0, 0, 0, 0, -5,
0, -3, 0, 0, 0, 5,
-5, 0, 0, 0, 3, 0,
5, 0, 0, 0, 0, -2,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 2, (1, 3)), False)
self.assertEqual(Board.is_move_valid(board, -1, 3, (24, 21)), True)
board = ( 0,
2, 0, 0, 0, 0, -5,
0, -3, 0, 0, 0, 5,
-5, 0, 0, 0, 3, 0,
5, 0, 0, 0, 0, -1,
-1 )
self.assertEqual(Board.is_move_valid(board, -1, 2, (24, 22)), False)
self.assertEqual(Board.is_move_valid(board, 1, 2, (1, 3)), True)
def test_owned_checker_at_from(self):
self.assertEqual(Board.is_move_valid(self.initial, 1, 2, (3, 5)), False) # No checkers
self.assertEqual(Board.is_move_valid(self.initial, -1, 2, (23, 21)), False) # No checkers
self.assertEqual(Board.is_move_valid(self.initial, 1, 2, (8, 10)), False) # Opponent checkers
self.assertEqual(Board.is_move_valid(self.initial, -1, 2, (1, 3)), False) # Opponent checkers
def test_no_block_at_to(self):
self.assertEqual(Board.is_move_valid(self.initial, 1, 5, (1, 6)), False)
self.assertEqual(Board.is_move_valid(self.initial, -1, 5, (24, 19)), False)
def test_able_to_hit_opponent_checkers(self):
board = ( 0,
2, 0, 0, -1, 0, -4,
0, -3, 0, 0, 0, 5,
-5, 0, 0, 0, 3, 0,
4, 0, 1, 0, 0, -2,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 3, (1, 4)), True)
self.assertEqual(Board.is_move_valid(board, -1, 3, (24, 21)), True)
def test_bear_off_simple_white(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 1,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 2, (24, 26)), True)
self.assertEqual(Board.is_move_valid(board, 1, 1, (24, 25)), True)
def test_bear_off_simple_black(self):
board = ( 0,
-1, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0 )
self.assertEqual(Board.is_move_valid(board, -1, 2, (1, -1)), True)
self.assertEqual(Board.is_move_valid(board, -1, 1, (1, 0)), True)
def test_bear_off_with_higher_face_value(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 1, 1, 0,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 5, (23, 28)), False)
self.assertEqual(Board.is_move_valid(board, 1, 5, (22, 27)), True)
def test_bear_off_all_in_last_quadrant(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 1, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 1,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 1, (24, 25)), False)
board = ( 1,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 1,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 1, (24, 25)), False)
board = ( 0,
-1, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
-1 )
self.assertEqual(Board.is_move_valid(board, -1, 1, (1, 0)), False)
board = ( 0,
-1, 0, 0, 0, 0, 0,
0, 0, 0, -1, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0 )
self.assertEqual(Board.is_move_valid(board, -1, 1, (1, 0)), False)
# TODO: More tests for bearing off are needed
def test_bear_off_non_backmost(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 1, 1,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 2, (23, 25)), True)
self.assertEqual(Board.is_move_valid(board, 1, 1, (24, 25)), True)
self.assertEqual(Board.is_move_valid(board, 1, 2, (24, 26)), False)
def test_bear_off_quadrant_limits_white(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 1,
1, 1, 1, 1, 1, 1,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 2, (23, 25)), False)
self.assertEqual(Board.is_move_valid(board, 1, 1, (24, 25)), False)
def test_bear_off_quadrant_limits_black(self):
board = ( 0,
-1, -1, -1, -1, -1, -1,
-1, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0 )
self.assertEqual(Board.is_move_valid(board, -1, 2, (2, 0)), False)
self.assertEqual(Board.is_move_valid(board, -1, 1, (1, 0)), False)
def test_bear_off_quadrant_limits_white_2(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
1, 0, 0, 0, 0, 1,
0 )
self.assertEqual(Board.is_move_valid(board, 1, 1, (24, 25)), True)
def test_bear_off_quadrant_limits_black_2(self):
board = ( 0,
-1, 0, 0, 0, 0, -1,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0 )
self.assertEqual(Board.is_move_valid(board, -1, 1, (1, 0)), True)
class TestNumOfChecker(unittest.TestCase):
def test_simple_1(self):
board = ( 0,
2, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 5,
0, 0, 0, 0, 3, 0,
4, 0, 1, 0, 0, 0,
0 )
self.assertEqual(Board.num_of_checkers_for_player(board,1), 2+5+3+4+1)
self.assertEqual(Board.num_of_checkers_for_player(board,-1), 0)
board = ( 0,
-2, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, -5,
0, 0, 0, 0, -3, 0,
-4, 0, -1, 0, 0, 0,
0 )
self.assertEqual(Board.num_of_checkers_for_player(board,-1), 2+5+3+4+1)
self.assertEqual(Board.num_of_checkers_for_player(board,1), 0)
class TestIsWinner(unittest.TestCase):
def test_is_winner(self):
board = ( 0,
2, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 5,
0, 0, 0, 0, 3, 0,
4, 0, 1, 0, 0, 0,
0 )
self.assertEqual(Board.outcome(board), {1: -2, -1: 2})
board = ( 0,
0, 0, 0, -1, 0, -4,
0, -3, 0, 0, 0, 0,
-5, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, -2,
0 )
self.assertEqual(Board.outcome(board), {1: 2, -1: -2})
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, -2,
0 )
self.assertEqual(Board.outcome(board), {1: 1, -1: -1})
board = ( 0,
2, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 5,
0, 0, 0, 0, 3, 0,
5, 0, 0, 0, 0, 0,
0 )
self.assertEqual(Board.outcome(board), {1: -2, -1: 2})
class TestDoMove(unittest.TestCase):
def test_simple_move(self):
board = ( 0,
2, 0, -1, 0, 0, 0,
0, 0, 0, 0, 0, 5,
0, 0, 0, 0, 3, 0,
4, 0, 1, 0, 0, 0,
0 )
expected_board = ( 0,
2, -1, 0, 0, 0, 0,
0, 0, 0, 0, 0, 5,
0, 0, 0, 0, 3, 0,
4, 0, 1, 0, 0, 0,
0 )
self.assertEqual(Board.do_move(board, -1, (3, 2)), expected_board)
def test_bear_in(self):
board = ( 1,
1, 0, 0, 0, 0, -5,
0, -3, 0, 0, 0, 5,
-5, 0, 0, 0, 3, 0,
5, 0, 0, 0, 0, -2,
0 )
expected_board = ( 0,
2, 0, 0, 0, 0, -5,
0, -3, 0, 0, 0, 5,
-5, 0, 0, 0, 3, 0,
5, 0, 0, 0, 0, -2,
0 )
self.assertEqual(Board.do_move(board, 1, (0, 1)), expected_board)
board = ( 0,
2, 0, 0, 0, 0, -5,
0, -3, 0, 0, 0, 5,
-5, 0, 0, 0, 3, 0,
5, 0, 0, 0, 0, -1,
-1 )
expected_board = ( 0,
2, 0, 0, 0, 0, -5,
0, -3, 0, 0, 0, 5,
-5, 0, 0, 0, 3, 0,
5, 0, 0, 0, 0, -2,
0 )
self.assertEqual(Board.do_move(board, -1, (25, 24)), expected_board)
def test_bear_off(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 1,
0 )
expected_board = tuple([0] * 26)
self.assertEqual(Board.do_move(board, 1, (24, 30)), expected_board)
self.assertEqual(Board.do_move(board, 1, (24, 26)), expected_board)
self.assertEqual(Board.do_move(board, 1, (24, 25)), expected_board)
board = ( 0,
-1, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0 )
expected_board = tuple([0] * 26)
self.assertEqual(Board.do_move(board, -1, (1, 0)), expected_board)
self.assertEqual(Board.do_move(board, -1, (1, -1)), expected_board)
self.assertEqual(Board.do_move(board, -1, (1, -4)), expected_board)
class TestAnyMoveValid(unittest.TestCase):
def test_bear_in_but_blocked(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
2, 2, 0, 2, 2, 2,
-1 )
self.assertEqual(Board.any_move_valid(board, -1, (3,6)), False)
self.assertEqual(Board.any_move_valid(board, -1, (6,3)), False)
self.assertEqual(Board.any_move_valid(board, -1, (3,4)), True)
self.assertEqual(Board.any_move_valid(board, -1, (4,1)), True)
def test_simple_block(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
2, 2, -1, 0, 0, 0,
0 )
self.assertEqual(Board.any_move_valid(board, -1, (1,1)), False)
self.assertEqual(Board.any_move_valid(board, -1, (2,2)), False)
self.assertEqual(Board.any_move_valid(board, -1, (2,1)), False)
self.assertEqual(Board.any_move_valid(board, -1, (1,2)), False)
self.assertEqual(Board.any_move_valid(board, -1, (3,1)), True)
self.assertEqual(Board.any_move_valid(board, -1, (1,3)), True)
self.assertEqual(Board.any_move_valid(board, -1, (4,4)), True)
def test_white_bear_off_odd(self):
board = (0,
-14, -1, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 1,
0)
self.assertEqual(Board.any_move_valid(board, 1, (1,2)), True)
self.assertEqual(Board.any_move_valid(board, 1, (2,4)), True)
self.assertEqual(Board.any_move_valid(board, 1, (4,6)), True)
def test_black_bear_off_odd(self):
board = (0,
-1, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 1, 14,
0)
self.assertEqual(Board.any_move_valid(board, -1, (1,2)), True)
self.assertEqual(Board.any_move_valid(board, -1, (2,4)), True)
self.assertEqual(Board.any_move_valid(board, -1, (4,6)), True)
class TestLegalMoves(unittest.TestCase):
def test_white_bear_off(self):
board = (0,
-14, -1, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 1,
0)
expected_board_set = {( 0,
-14, -1, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0 )}
self.assertEqual(Board.calculate_legal_states(board, 1, (1,2)), expected_board_set)
self.assertEqual(Board.calculate_legal_states(board, 1, (2,4)), expected_board_set)
self.assertEqual(Board.calculate_legal_states(board, 1, (4,2)), expected_board_set)
self.assertEqual(Board.calculate_legal_states(board, 1, (4,6)), expected_board_set)
def test_black_bear_off(self):
board = (0,
-1, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 1, 14,
0)
expected_board_set = {( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 1, 14,
0 )}
self.assertEqual(Board.calculate_legal_states(board, -1, (1,2)), expected_board_set)
self.assertEqual(Board.calculate_legal_states(board, -1, (2,4)), expected_board_set)
self.assertEqual(Board.calculate_legal_states(board, -1, (4,2)), expected_board_set)
self.assertEqual(Board.calculate_legal_states(board, -1, (4,6)), expected_board_set)
def test_blocked(self):
board = ( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
2, 2, -1, 0, 0, 0,
0 )
expected_board_set_3_1 = {( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -1, 0,
2, 2, 0, 0, 0, 0,
0 )}
expected_board_set_3_4 = {( 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, -1, 0, 0, 0, 0,
2, 2, 0, 0, 0, 0,
0 )}
self.assertEqual(Board.calculate_legal_states(board, -1, (1,2)), { board })
self.assertEqual(Board.calculate_legal_states(board, -1, (1,1)), { board })
self.assertEqual(Board.calculate_legal_states(board, -1, (3,1)), expected_board_set_3_1)
self.assertEqual(Board.calculate_legal_states(board, -1, (3,4)), expected_board_set_3_4)
def test_multiple_checkers(self):
board = ( 0,
0, 0, 1, 0, 0, 1,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 )
expected_board_set_3_5 = {
( 0,
0, 0, 0, 0, 0, 1,
0, 0, 0, 0, 1, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
0, 0, 1, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 1, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
0, 0, 0, 0, 0, 0,
0, 1, 1, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 )
}
self.assertEqual(Board.calculate_legal_states(board, 1, (3,5)), expected_board_set_3_5)
def test_double_roll(self):
board = ( 0,
2, 0, 0, 2, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 )
expected_board_set_1_1 = {
( 0,
1, 0, 0, 2, 1, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
2, 0, 0, 1, 0, 0,
0, 1, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
0, 0, 2, 2, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
2, 0, 0, 0, 0, 2,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
0, 2, 0, 0, 2, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
1, 1, 0, 1, 0, 0,
1, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
1, 0, 1, 1, 0, 1,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
0, 1, 1, 1, 1, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
2, 0, 0, 0, 1, 0,
1, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
0, 2, 0, 1, 0, 1,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
1, 1, 0, 0, 1, 1,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
0, 1, 0, 3, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 ),
( 0,
1, 0, 1, 0, 2, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, -2, 0,
0 )
}
self.assertEqual(Board.calculate_legal_states(board, 1, (1,1)), expected_board_set_1_1)
def test_hit_on_bear_in(self):
board = (0,
-2, 0, -2, -1, -3, -2,
0, -2, 0, 0, 0, 0,
-1, 0, 0, 0, 0, 0,
6, 0, 3, 1, 1, -4,
-1)
expected_board_set = { (1,
-2, 0, -2, -1, -3, -2,
0, -2, 0, 0, 0, 0,
-1, 0, 0, 0, 0, -1,
6, 0, 3, 0, 1, -4,
0),
(1,
-2, 0, -2, -1, -3, -2,
0, -2, -1, 0, 0, 0,
0, 0, 0, 0, 0, 0,
6, 0, 3, -1, 1, -4,
0),
(1,
-2, 0, -2, -2, -3, -2,
0, -1, 0, 0, 0, 0,
-1, 0, 0, 0, 0, 0,
6, 0, 3, -1, 1, -4,
0),
(1,
-2, -1, -2, -1, -3, -1,
0, -2, 0, 0, 0, 0,
-1, 0, 0, 0, 0, 0,
6, 0, 3, -1, 1, -4,
0),
(1,
-3, 0, -2, -1, -2, -2,
0, -2, 0, 0, 0, 0,
-1, 0, 0, 0, 0, 0,
6, 0, 3, -1, 1, -4,
0),
(1,
-2, 0, -2, -1, -3, -2,
0, -2, 0, 0, 0, 0,
-1, 0, 0, 0, 0, 0,
6, -1, 3, -1, 1, -3,
0)
}
self.assertEqual(Board.calculate_legal_states(board, -1, (4,3)), expected_board_set)
class TestBoardFlip(unittest.TestCase):
def test_flip_board(self):
board = (0,
-14, -1, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 1,
0)
expected_board = ( 0,
-1, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 1, 14,
0 )
self.assertEqual(Board.flip(board), expected_board)
def test_flip_board_bar(self):
board = (2,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
-7)
expected_board = (7,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0,
-2)
self.assertEqual(Board.flip(board), expected_board)
def test_flip_board_extensive(self):
board = (4,
-5, -1, 0, 4, 3, 0,
0, -1, 0, -5, 0, 0,
0, 3, 0, 0, 0, 0,
0, 0, 0, -1, 0, 1,
-2)
expected_board = (2,
-1, 0, 1, 0, 0, 0,
0, 0, 0, 0, -3, 0,
0, 0, 5, 0, 1, 0,
0, -3, -4, 0, 1, 5,
-4)
self.assertEqual(Board.flip(board), expected_board)
def test_inverse(self):
board = (4,
-5, -1, 0, 4, 3, 0,
0, -1, 0, -5, 0, 0,
0, 3, 0, 0, 0, 0,
0, 0, 0, -1, 0, 1,
-2)
self.assertEqual(Board.flip(Board.flip(board)), board)
def test_tesauro_initial(self):
board = Board.initial_state
expected = (1,1,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0.0,
0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
1,1,1,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,0,0,
0.0,
0,
1,
0
)
import numpy as np
self.assertTrue((Board.board_features_tesauro(board, 1) ==
np.array(expected).reshape(1, 198)).all())
def test_tesauro_bars(self):
board = list(Board.initial_state)
board[1] = 0
board[0] = 2
board[24] = 0
board[25] = -2
board = tuple(board)
expected = (0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1.0,
0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
1,1,1,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1.0,
0,
1,
0
)
import numpy as np
self.assertTrue((Board.board_features_tesauro(board, 1) ==
np.array(expected).reshape(1, 198)).all())
def test_tesauro_home(self):
board = list(Board.initial_state)
board[1] = 0
board[24] = 0
board = tuple(board)
expected = (0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0.0,
2,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
1,1,1,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0.0,
2,
1,
0
)
import numpy as np
self.assertTrue((Board.board_features_tesauro(board, 1) ==
np.array(expected).reshape(1, 198)).all())
def test_tesauro_black_player(self):
board = Board.initial_state
expected = (1,1,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0.0,
0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
1,1,1,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,1,1,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
0,0,0,0,
1,1,0,0,
0.0,
0,
0,
1
)
import numpy as np
self.assertTrue((Board.board_features_tesauro(board, -1) ==
np.array(expected).reshape(1, 198)).all())
if __name__ == '__main__':
unittest.main()
Reference in New Issue View Git Blame Copy Permalink