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