def test_case(color, points, bar=None, off=None): board = Board(points, bar, off) frozen_board = FrozenBoard(points, bar, off) self.assertEqual(board, frozen_board) # Pick a random few possible moves and test for dice in (Dice(roll=(1, 2)), Dice(roll=(4, 5))): for moves, after_board in sorted( board.possible_moves(color, dice))[:10]: new_frozen_board = frozen_board.copy_apply_moves(moves) new_board = board.mutable_copy() for move in moves: new_board.apply_move(move) self.assertEqual(new_board, new_frozen_board) self.assertEqual(new_board, after_board)
def test_white_can_bear_off_from_24(self): board = Board(repeat_point(23) + repeat_point(1, 1, Color.White)) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(1, 6)))), set([ (Move(Color.White, 24, 1), ), (Move(Color.White, 24, 6), ), ]))
def test_single_valid_move(self): board = Board( repeat_point(10) + [Point(1, Color.White)] + [Point(2, Color.White)] + repeat_point(12, 2, Color.Black)) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(1, 3)))), set([ (Move(Color.White, 11, 1), ), ]))
def test_doubles(self): board = Board( repeat_point(10) + [Point(1, Color.White)] + repeat_point(13)) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(2, 2)))), set([ (Move(Color.White, 11, 2), Move(Color.White, 13, 2), Move(Color.White, 15, 2), Move(Color.White, 17, 2)), ]))
def test_single_pip_anywhere(self): board = Board( repeat_point(10) + [Point(1, Color.White)] + repeat_point(13)) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(1, 2)))), set([ (Move(Color.White, 11, 1), Move(Color.White, 12, 2)), (Move(Color.White, 11, 2), Move(Color.White, 13, 1)), ]))
def test_one_blocked_pip_bear_off(self): """Sames as test_one_pip_bear_off but 1 is blocked""" board = Board( repeat_point(4) + [Point(2, Color.White)] + [Point(1, Color.Black)] + repeat_point(18)) self.assertEqual( extract_moves(board.possible_moves(Color.Black, Dice(roll=(1, 6)))), set([ (Move(Color.Black, 6, 6), ), ]))
def test_hit_opportunity(self): board = Board( repeat_point(10) + [Point(1, Color.White)] + [Point(1, Color.Black)] + repeat_point(12)) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(1, 2)))), set([ (Move(Color.White, 11, 1), Move(Color.White, 12, 2)), (Move(Color.White, 11, 2), Move(Color.White, 13, 1)), ]))
def test_can_get_off_bar_with_parly_full_home(self): board = Board(repeat_point(5, 2, Color.Black) + repeat_point(19)) board.add_bar(Color.White) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(1, 6)))), set([ ( Move(Color.White, Move.Bar, 6), Move(Color.White, 6, 1), ), ]))
def test_can_get_off_bar_with_empty_home(self): board = Board(repeat_point(24)) board.add_bar(Color.White) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(1, 6)))), set([ ( Move(Color.White, Move.Bar, 1), Move(Color.White, 1, 6), ), ( Move(Color.White, Move.Bar, 6), Move(Color.White, 6, 1), ), ]))
def test_one_pip_bear_off(self): board = Board( repeat_point(5) + [Point(1, Color.Black)] + repeat_point(18)) self.assertEqual( extract_moves(board.possible_moves(Color.Black, Dice(roll=(1, 6)))), set([ ( Move(Color.Black, 6, 1), Move(Color.Black, 5, 6), ), # Practically, this move is idenical since the above two # moves result in bearing off. However the above move uses # both dice, and so is considered correct # (Move(Color.Black, 6, 6),), ]))
def test_two_pip_bear_off(self): board = Board( repeat_point(19) + [Point(2, Color.White)] + repeat_point(4)) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(5, 6)))), set([ # These movesets are identical to each other because they both bear # off and could be deduped, but don't seem harmful, so leaving for now ( Move(Color.White, 20, 6), Move(Color.White, 20, 5), ), ( Move(Color.White, 20, 5), Move(Color.White, 20, 6), ), ]))
def test_double_roll_filled_board_performance(self): board = Board( repeat_point(2, 1, Color.White) + repeat_point(2, 1, Color.Black) + repeat_point(20)) with assertRuntime(self, 0.1): board.possible_moves(Color.White, Dice(roll=(2, 2)))
def test_cant_get_off_bar_with_full_home(self): board = Board(repeat_point(6, 2, Color.Black) + repeat_point(18)) board.add_bar(Color.White) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(1, 6)))), set())
def test_perfect_prime_no_moves(self): board = Board([Point(2, Color.White)] + repeat_point(6, 2, Color.Black) + repeat_point(17)) self.assertEqual( extract_moves(board.possible_moves(Color.White, Dice(roll=(6, 6)))), set())