def test_get_all_boards_png_three_games(self):
chess_bot = Chess()
board1 = chess.Board()
board1.push_san("e4")
board1.push_san("e5")
game1 = Game()
game1.color_schema = "blue"
game1.board = board1
chess_bot.games.append(game1)
board2 = chess.Board()
board2.push_san("Nf3")
board2.push_san("d6")
game2 = Game()
game2.color_schema = "wood"
game2.board = board2
chess_bot.games.append(game2)
board3 = chess.Board()
board3.push_san("Nf3")
game3 = Game()
game3.color_schema = "green"
game3.board = board3
chess_bot.games.append(game3)
image_bytesio = asyncio.run(chess_bot.get_all_boards_png())
with open(
os.path.join('tests', 'support',
'get_all_boards_png_three_games.png'), 'rb') as f:
self.assertEqual(image_bytesio.read(), f.read())
def test_get_all_boards_png_one_game(self):
chess_bot = Chess()
board1 = chess.Board()
board1.push_san("e4")
board1.push_san("e5")
board1.push_san("Bc4")
game1 = Game()
game1.color_schema = "green"
game1.board = board1
chess_bot.games.append(game1)
image_bytesio = asyncio.run(chess_bot.get_all_boards_png())
with open(
os.path.join('tests', 'support',
'get_all_boards_png_one_game.png'), 'rb') as f:
self.assertEqual(image_bytesio.read(), f.read())
def test_get_all_boards_png_no_games_being_played(self):
chess_bot = Chess()
image_bytesio = asyncio.run(chess_bot.get_all_boards_png())
self.assertIsNone(image_bytesio)