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 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()