def test_swapper():
mode = GameMode()
arrangement = [(Unit(SwapMover()), 2, 2)]
mode.set_arrangement(arrangement)
mode.set_size(8)
board = Board(mode)
board[5, 5] = None
x = 2
y = 2
for dx in range(-10, 10):
for dy in range(-10, 10):
if dx == 0 and dy == 0:
continue
to_x = x + dx
to_y = y + dy
if valid(to_x, to_y, 8):
assert board.do_pofig_move(x, y, to_x, to_y) is False
x = 1
y = 1
arrangement.clear()
arrangement.append((Unit(SwapMover()), 1, 1))
for dx in range(-1, 2):
for dy in range(-1, 2):
if dx == 0 and dy == 0:
continue
arrangement.append((Unit(KingMover()), x + dx, y + dy))
mode.set_arrangement(arrangement)
board = Board(mode)
for i, j in product(range(5, 8), range(5, 8)):
assert board.do_pofig_move(x, y, i, j) is True
assert board.do_pofig_move(i, j, x, y) is True
def test_bishop():
mode = GameMode()
mode.set_size(8)
mode.set_arrangement([(Unit(BishopMover()), 2, 2),
(Unit(BishopMover()), 3, 3)])
board = Board(mode)
x = 2
y = 2
assert board.do_pofig_move(2, 2, 4, 4) is False
for i in range(8):
for j in range(8):
if i != x and j != y:
board[i, j] = None
for dx in range(-10, 10):
for dy in range(-10, 10):
if dx == 0 and dy == 0:
continue
to_x = x + dx
to_y = y + dy
if valid(to_x, to_y, 8):
if abs(dx) == abs(dy):
assert board.do_pofig_move(x, y, to_x, to_y) is True
assert board.do_pofig_move(to_x, to_y, x, y) is True
else:
assert board.do_pofig_move(x, y, to_x, to_y) is False
def test_usual():
mode = GameMode()
mode.set_size(8)
mode.set_arrangement([(Unit(UsualMover()), 2, 2)])
x = 2
y = 2
board = Board(mode)
for dx in range(-10, 10):
for dy in range(-10, 10):
if dx == 0 and dy == 0:
continue
to_x = x + dx
to_y = y + dy
if valid(to_x, to_y, 8):
if abs(dx) <= 1 and abs(dy) <= 1 and dx * dy == 0:
assert board.do_pofig_move(x, y, to_x, to_y) is True
assert board.do_pofig_move(to_x, to_y, x, y) is True
else:
assert board.do_pofig_move(x, y, to_x, to_y) is False
arrangement = [(Unit(UsualMover()), 4, 4)]
x = 4
y = 4
for dx in range(-1, 2):
for dy in range(-1, 2):
if (dx == 0 and dy == 0) or dx * dy != 0:
continue
arrangement.append((Unit(PawnMover()), x + dx, y + dy))
mode.set_size(20)
mode.set_arrangement(arrangement)
board = Board(mode)
for dx in range(-2, 3, 2):
for dy in range(-2, 3, 2):
if (dx == 0 and dy == 0) or dx * dy != 0:
continue
to_x = x + dx
to_y = y + dy
assert board.do_pofig_move(x, y, to_x, to_y) is True
assert board.do_pofig_move(to_x, to_y, x, y) is True
def test_pawn():
mode = GameMode()
mode.set_size(8)
arrangement = [(Unit(PawnMover()), 2, 2)]
mode.set_arrangement(arrangement)
board = Board(mode)
x = 2
y = 2
for dx in range(-10, 10):
for dy in range(-10, 10):
if dx == 0 and dy == 0:
continue
to_x = x + dx
to_y = y + dy
if valid(to_x, to_y, 8):
if abs(dx) <= 1 and abs(dy) <= 1 and dx * dy == 0:
assert board.do_pofig_move(x, y, to_x, to_y) is True
assert board.do_pofig_move(to_x, to_y, x, y) is True
else:
assert board.do_pofig_move(x, y, to_x, to_y) is False
def test_rook():
mode = GameMode()
mode.set_size(8)
mode.set_arrangement([(Unit(RookMover()), 2, 2),
(Unit(RookMover()), 2, 3)])
board = Board(mode)
x = 2
y = 2
assert board.do_pofig_move(2, 2, 2, 6) is False
board[2, 3] = None
for dx in range(-10, 10):
for dy in range(-10, 10):
if dx == dy == 0:
continue
to_x = x + dx
to_y = y + dy
if valid(to_x, to_y, 8):
if dx == 0 or dy == 0:
assert board.do_pofig_move(x, y, to_x, to_y) is True
assert board.do_pofig_move(to_x, to_y, x, y) is True
else:
assert board.do_pofig_move(x, y, to_x, to_y) is False
def test_king():
mode = GameMode()
mode.set_size(8)
mode.set_arrangement([(Unit(KingMover()), 2, 2),
(Unit(KingMover()), 1, 1)])
board = Board(mode)
x = 2
y = 2
assert board.do_pofig_move(2, 2, 1, 1) is False
board[1, 1] = None
for dx in range(-10, 10):
for dy in range(-10, 10):
if dx == dy == 0:
continue
to_x = x + dx
to_y = y + dy
if valid(to_x, to_y, 8):
if abs(dx) <= 1 and abs(dy) <= 1:
assert board.do_pofig_move(x, y, to_x, to_y) is True
assert board.do_pofig_move(to_x, to_y, x, y) is True
else:
assert board.do_pofig_move(x, y, to_x, to_y) is False
def test_police():
mode = GameMode()
mode.set_size(15)
arrangement = [(Unit(PoliceManMover()), 4, 4)]
x = 4
y = 4
for dx in range(-1, 2):
for dy in range(-1, 2):
if dx != 0 or dy != 0 or dx != dy:
arrangement.append((Unit(KingMover()), x + dx, y + dy))
mode.set_arrangement(arrangement)
board = Board(mode)
for i in range(15):
for j in range(15):
if board[i, j] is not None and board[i,
j].player == board.player_B:
board[i, j] = None
for dx in range(-1, 2):
for dy in range(-1, 2):
p_x = x + dx
p_y = y + dy
for d2x in range(-1, 2):
for d2y in range(-1, 2):
if board[p_x + dx, p_y + dy] is None:
assert board.do_pofig_move(p_x, p_y, p_x + dx,
p_y + dy) is True
assert board.do_pofig_move(p_x + dx, p_y + dy, p_x,
p_y) is True
for i in range(15):
for j in range(15):
if board[i, j] is not None and (i != 4 or j != 4):
board[i, j].player = board.player_B
for dx in range(-1, 2):
for dy in range(-1, 2):
if dx == 0 and dy == 0:
continue
p_x = x + dx
p_y = y + dy
for d2x in range(-1, 2):
for d2y in range(-1, 2):
assert board.do_pofig_move(p_x, p_y, p_x + dx,
p_y + dy) is False
x = 2
y = 2
mode.set_size(5)
mode.set_arrangement([(Unit(PoliceManMover()), 2, 2)])
board = Board(mode)
for dx in range(-6, 6):
for dy in range(-6, 6):
if dx == dy and dx == 0:
continue
to_x = x + dx
to_y = y + dy
if valid(to_x, to_y, 6):
if abs(dx) <= 1 and abs(dy) <= 1:
assert board.do_pofig_move(x, y, to_x, to_y) is True
assert board.do_pofig_move(to_x, to_y, x, y) is True
else:
assert board.do_pofig_move(x, y, to_x, to_y) is False