def test_king_face_king(self):
fen = '9/4k4/9/9/9/9/4R4/9/9/5K3 b - - 0 1'
fen = replace_chess(fen)
board = Board(fen)
correct_set = {"e8e9", "e8e7", "e8d8"}
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)
def test_pawn4(self):
fen = '9/9/9/9/7Pp/8c/9/9/9/9 b - - 0 33'
board = Board(fen)
correct_set = {'i4a4', 'i4b4', 'i4c4', 'i4d4', 'i4e4', 'i4f4', 'i4g4',
'i4h4', 'i4i3', 'i4i2', 'i4i1', 'i4i0', }
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)
def test_elephant3(self):
fen = '9/9/9/p1p6/2b6/9/9/9/9/9 b - - 0 1'
fen = replace_chess(fen)
board = Board(fen)
correct_set = {'c5a7', 'c5e7', 'a6a5'}
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)
def test_elephant2(self):
fen = '9/9/4b4/9/9/9/9/9/9/9 b - - 0 1'
fen = replace_chess(fen)
board = Board(fen)
correct_set = {'e7c9', 'e7c5', 'e7g9', 'e7g5'}
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)
def test_elephant(self):
fen = '9/9/9/9/9/9/9/4B4/3K1p3/9 w - - 0 1'
fen = replace_chess(fen)
board = Board(fen)
correct_set = {'d1e1', 'd1d2', 'd1d0', 'e2c4', 'e2g4'}
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)
def test_king_win(self):
fen = '9/4k4/9/9/9/9/4R4/9/9/5K3 w - - 0 1'
fen = replace_chess(fen)
board = Board(fen)
correct_set = {"e3e8", "e3e0", "e3e1", "e3e2", "e3e4", "e3e5", "e3e6", "e3e7", "e3a3", "e3b3", "e3c3", "e3d3",
"e3f3", "e3g3", "e3h3", "e3i3", "f0e0", "f0f1"}
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)
def test_horse(self):
fen = '9/9/9/9/5p3/9/3RN4/4p1P2/9/9 w - - 0 1'
fen = replace_chess(fen)
board = Board(fen)
correct_set = {'e3d5', 'e3f5', 'e3g4', 'd3d0', 'd3d1', 'd3d2', 'd3d4',
'd3d5', 'd3d6', 'd3d7', 'd3d8', 'd3d9', 'd3c3', 'd3b3',
'd3a3', 'g2g3', }
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)
def test_pawn3(self):
"""
测试黑方过河小兵的行为。
:return:
"""
fen = '9/9/9/p1p3p1p/9/4pR3/9/9/9/9 b - - 0 1'
board = Board(fen)
correct_set = {'a6a5', 'c6c5', 'e4d4', 'e4f4', 'e4e3', 'g6g5', 'i6i5'}
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)
def test_pawn2(self):
"""
测试红方过河小兵的行为。
:return:
"""
fen = '9/9/9/9/3rP4/9/P1P3P1P/9/9/9 w - - 0 1'
board = Board(fen)
correct_set = {'a3a4', 'c3c4', 'e5d5', 'e5f5', 'e5e6', 'g3g4', 'i3i4'}
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)
def test_cannon2(self):
fen = '9/9/9/9/9/9/p1R1C4/9/4N4/9 w - - 0 1'
fen = replace_chess(fen)
board = Board(fen)
correct_set = {'e3e2', 'e3e4', 'e3e5', 'e3e6', 'e3e7', 'e3e8', 'e3e9',
'e3f3', 'e3g3', 'e3h3', 'e3i3', 'e3d3', 'e3a3', 'e1c2',
'e1c0', 'e1d3', 'e1f3', 'e1g2', 'e1g0', 'c3c0', 'c3c1',
'c3c2', 'c3c4', 'c3c5', 'c3c6', 'c3c7', 'c3c8', 'c3c9',
'c3a3', 'c3b3', 'c3d3' }
result_set = set()
for step in board.generate_legal_moves():
result_set.add(step.ucci())
self.assertEqual(correct_set, result_set)