def test_game_file_system_works():
""" Game file system working correct """
if os.path.exists('game.tchess'):
os.remove('game.tchess')
if os.path.exists('other.tchess'):
os.remove('other.tchess')
proc = subprocess.Popen(
PY_EXE + ' tchess', shell=True,
stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE
)
proc.communicate(input='mv 2.1 to 3.1\nexit'.encode())
assert os.path.isfile('game.tchess')
saved_game = load_game_from_file('game.tchess')
assert isinstance(saved_game, Game)
assert saved_game.version == Game().version
assert saved_game.logs == ['mv 2.1 to 3.1']
proc = subprocess.Popen(
PY_EXE + ' tchess other.tchess', shell=True,
stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE
)
proc.communicate(input='mv 2.2 to 3.2\nexit'.encode())
assert os.path.isfile('other.tchess')
saved_game2 = load_game_from_file('other.tchess')
assert isinstance(saved_game2, Game)
assert saved_game2.version == Game().version
assert saved_game2.logs == ['mv 2.2 to 3.2']
proc = subprocess.Popen(
PY_EXE + ' tchess other.tchess', shell=True,
stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE
)
proc.communicate(input='mv 7.3 to 6.3\nexit'.encode())
assert os.path.isfile('other.tchess')
saved_game2 = load_game_from_file('other.tchess')
assert isinstance(saved_game2, Game)
assert saved_game2.version == Game().version
assert saved_game2.logs == ['mv 2.2 to 3.2', 'mv 7.3 to 6.3']
proc = subprocess.Popen(
PY_EXE + ' tchess --replay --replay-speed=0.1 other.tchess', shell=True,
stdout=subprocess.PIPE
)
proc.communicate()
os.remove('game.tchess')
os.remove('other.tchess')
def test_default_state_is_valid():
""" Default state of chess board is valid """
game = Game()
assert game.board[2:6] == ([[None] * 8] * 4)
for item in game.board[1]:
assert item.name == Piece.PAWN
assert item.color == 'white'
for item in game.board[6]:
assert item.name == Piece.PAWN
assert item.color == 'black'
colors = [
['white', 0],
['black', 7],
]
for color in colors:
index = color[1]
color = color[0]
assert game.board[index][0].name == Piece.ROOK
assert game.board[index][7].color == color
assert game.board[index][1].name == Piece.KNIGHT
assert game.board[index][6].color == color
assert game.board[index][2].name == Piece.BISHOP
assert game.board[index][5].color == color
assert game.board[index][3].name == Piece.KING
assert game.board[index][4].color == color
def test_command_s_works():
""" Command `s` for show allowed cells to go working correct """
game = Game()
game.run_command('s 2.1')
assert game.highlight_cells == [[2, 0], [3, 0]]
game.run_command('mv 2.1 3.1')
game.run_command('s 3.1')
assert game.highlight_cells == [[3, 0]]
def test_checkmate_and_example():
""" Checkmate works with a example """
commands = [
'mv 2.6 3.6',
'mv 7.1 6.1',
'mv 1.5 3.7',
'mv 6.1 5.1',
'mv 1.2 3.3',
'mv 5.1 4.1',
'mv 3.3 5.4',
'mv 4.1 3.1',
'mv 3.7 7.3',
]
game = Game()
for command in commands:
assert game.current_check is None
game.run_command(command)
assert game.current_check == 'black'
game.run_command('mv 8.4 7.3',)
assert game.is_end
assert game.winner == 'white'
def test_turn_changer_works():
""" Game turn can be changed correctly """
game = Game()
assert game.turn == 'white'
game.change_turn()
assert game.turn == 'black'
game.change_turn()
assert game.turn == 'white'
def test_pawn_promotion():
""" Pawn promotion system works """
commands = [
'mv 2.1 4.1',
'mv 7.2 5.2',
'mv 1.1 3.1',
'mv 5.2 4.1',
'mv 3.1 3.2',
'mv 4.1 3.1',
'mv 3.2 8.2',
'mv 3.1 2.1',
'mv 8.2 8.1',
'mv 2.1 1.1',
'mv 2.1 1.1 > ' + Piece.ICONS[Piece.ROOK],
]
game = Game()
for cmd in commands:
game.run_command(cmd)
assert game.board[0][0].name == Piece.ROOK
assert game.board[0][0].color == 'black'
def test_command_back_works():
""" Command `back` works """
game = Game()
assert game.logs == []
game.run_command('mv 2.1 3.1')
game.run_command('mv 7.3 5.3')
assert game.logs == ['mv 2.1 3.1', 'mv 7.3 5.3']
assert game.board[1][0] is None
assert game.board[2][0] is not None
assert game.board[6][2] is None
assert game.board[4][2] is not None
game.run_command('back')
assert game.logs == ['mv 2.1 3.1']
assert game.board[1][0] is None
assert game.board[2][0] is not None
assert game.board[6][2] is not None
assert game.board[4][2] is None
game = Game()
assert str_contains_all(game.run_command('back'), ['first', 'move'])
def test_command_runner_works():
""" Game command runner works """
game = Game()
assert 'invalid' in game.run_command('fdfgdd').lower()
assert str_contains_all(game.run_command('mv 4.3 to 6-3').lower(), ['error', 'source', 'empty'])
assert game.run_command('mv hi to 43543').lower().startswith('invalid')
assert str_contains_all(game.run_command('mv 7.2 to 5.3').lower(), ['error', 'turn'])
assert str_contains_all(game.run_command('mv 1.4 to 2.4').lower(), ['not', 'allowed'])
assert game.board[1][0] is not None
assert game.board[3][0] is None
assert str_contains_all(game.run_command('mv 2.1 to 4.1').lower(), ['moved', '2.1', '4.1'])
assert game.board[1][0] is None
assert game.board[3][0] is not None
assert game.board[3][0].name == Piece.PAWN
def test_log_list_is_working():
""" Logs will be saved """
game = Game()
commands = [
'move 2.1 to 3.1',
'move 7.1 5.1',
]
game.run_command('gfdgfd')
for command in commands:
game.run_command(command)
assert game.logs == commands
def test_bishop_move_validation_works():
""" Bishop move validation working correct """
game = Game()
assert str_contains_all(game.run_command('s 1.3'), ['cannot', 'move'])
assert str_contains_all(game.run_command('s 8.6'), ['cannot', 'move'])
game.run_command('mv 2.4 3.4')
game.run_command('s 1.3')
assert game.highlight_cells == [[1, 3], [2, 4], [3, 5], [4, 6], [5, 7]]
game.run_command('mv 7.1 6.1')
game.run_command('mv 1.3 4.6')
game.run_command('s 4.6')
assert game.highlight_cells == [
[4, 6], [5, 7], [2, 4], [1, 3], [0, 2], [2, 6], [4, 4], [5, 3], [6, 2]
]
game.run_command('mv 7.4 6.4')
game.run_command('mv 4.6 6.4')
assert game.board[3][5] is None
assert game.board[5][3].name == Piece.BISHOP
game.run_command('s 8.3')
assert game.highlight_cells == [[6, 3], [5, 4], [4, 5], [3, 6], [2, 7]]
game.run_command('mv 8.3 5.6')
game.run_command('s 5.6')
assert game.highlight_cells == [[3, 4], [2, 3], [5, 6], [5, 4], [6, 3], [7, 2], [3, 6], [2, 7]]
game.run_command('mv 2.1 3.1')
game.run_command('mv 5.6 3.4')
assert game.board[4][5] is None
assert game.board[2][3].name == Piece.BISHOP
def test_knight_move_validation_works():
""" Knight move validation working correct """
game = Game()
game.run_command('s 1.2')
assert game.highlight_cells == [[2, 2], [2, 0]]
game.run_command('s 8.7')
assert game.highlight_cells == [[5, 7], [5, 5]]
game.run_command('mv 1.2 3.3')
game.run_command('s 3.3')
assert game.highlight_cells == [[3, 4], [3, 0], [4, 3], [4, 1], [0, 1]]
game.run_command('mv 7.2 5.2')
assert str_contains_all(game.run_command('mv 3.3 5.2').lower(), ['moved', 'to'])
def test_king_move_validation_works():
""" King move validation working correct """
game = Game()
assert str_contains_all(game.run_command('s 1.4').lower(), ['piece', 'move', 'cannot'])
assert str_contains_all(game.run_command('s 8.4').lower(), ['piece', 'move', 'cannot'])
game.run_command('mv 2.4 4.4')
game.run_command('s 1.4')
assert game.highlight_cells == [[1, 3]]
game.run_command('mv 7.4 5.4')
game.run_command('s 8.4')
assert game.highlight_cells == [[6, 3]]
game.run_command('mv 1.4 2.4')
game.run_command('s 2.4')
assert game.highlight_cells == [[2, 4], [2, 3], [2, 2], [0, 3]]
game.run_command('mv 8.4 7.4')
game.run_command('s 7.4')
assert game.highlight_cells == [[7, 3], [5, 4], [5, 3], [5, 2]]
game.run_command('mv 2.3 4.3')
game.run_command('mv 5.4 4.3')
game.run_command('mv 4.4 5.4')
game.run_command('mv 7.4 6.4')
game.run_command('s 6.4')
assert game.highlight_cells == [[6, 3], [5, 4], [5, 2], [4, 4], [4, 3], [4, 2]]
game.run_command('mv 2.2 3.2')
assert str_contains_all(game.run_command('mv 6.4 5.4').lower(), ['moved', 'to'])
def test_rook_move_validation_works():
""" Rock move validation working correct """
game = Game()
assert str_contains_all(game.run_command('s 1.1').lower(), ['piece', 'move', 'cannot'])
assert str_contains_all(game.run_command('s 8.1').lower(), ['piece', 'move', 'cannot'])
game.run_command('mv 2.1 4.1')
game.run_command('s 1.1')
assert game.highlight_cells == [[1, 0], [2, 0]]
game.run_command('mv 7.1 5.1')
game.run_command('s 8.1')
assert game.highlight_cells == [[6, 0], [5, 0]]
game.run_command('mv 1.1 3.1')
game.run_command('s 3.1')
assert game.highlight_cells == [
[1, 0], [0, 0], [2, 1], [2, 2], [2, 3], [2, 4], [2, 5], [2, 6], [2, 7]
]
game.run_command('mv 8.1 6.1')
game.run_command('s 6.1')
assert game.highlight_cells == [
[6, 0], [7, 0], [5, 1], [5, 2], [5, 3], [5, 4], [5, 5], [5, 6], [5, 7]
]
game.run_command('mv 2.2 4.2')
game.run_command('mv 5.1 4.2')
game.run_command('s 6.1')
assert game.highlight_cells == [
[4, 0], [3, 0], [6, 0], [7, 0], [5, 1], [5, 2], [5, 3], [5, 4], [5, 5], [5, 6], [5, 7]
]
game.run_command('mv 3.1 3.2')
game.run_command('s 3.2')
assert game.highlight_cells == [
[3, 1], [1, 1], [2, 2], [2, 3], [2, 4], [2, 5], [2, 6], [2, 7], [2, 0]
]
def test_pawn_move_validation_works():
""" Pawn move validation working correct """
game = Game()
game.run_command('mv 2.1 3.1')
assert game.board[1][0] is None
assert game.board[2][0].name == Piece.PAWN
game.run_command('mv 7.1 5.1')
assert game.board[6][0] is None
assert game.board[4][0].name == Piece.PAWN
assert str_contains_all(game.run_command('mv 2.2 3.5').lower(), ['error'])
assert game.board[1][1] is not None
assert game.board[2][4] is None
game = Game()
game.run_command('mv 2.1 4.1')
game.run_command('mv 7.1 5.1')
assert str_contains_all(
game.run_command('mv 4.1 5.1').lower(),
['error', 'location', 'allowed', 'not']
)
assert str_contains_all(game.run_command('s 4.1').lower(), ['cannot', 'move'])
game = Game()
game.run_command('mv 2.1 4.1')
game.run_command('mv 7.8 6.8')
game.run_command('mv 4.1 5.1')
game.run_command('mv 6.8 5.8')
game.run_command('mv 5.1 6.1')
game.run_command('mv 7.2 6.1')
assert str_contains_all(game.run_command('s 7.1').lower(), ['piece', 'move', 'cannot'])