def test_eval_last_move_no_blunder_no_mate(self):
board = chess.Board()
board.push_san("g4")
board.push_san("e5")
game = Game()
game.board = board
game.player1 = FakeDiscordUser(id=1)
game.player2 = FakeDiscordUser(id=2)
game.current_player = game.player1
game.last_eval = Cp(0)
chess_bot = Chess()
chess_bot.games.append(game)
chess_bot.stockfish_limit["time"] = 2
result = asyncio.run(chess_bot.eval_last_move(game))
self.assertFalse(result["blunder"])
self.assertIsNone(result["mate_in"])
def test_eval_last_move_no_blunder_mate_in_two(self):
board = chess.Board(
"Q2r4/1p1k4/1P3ppp/1Kp1r3/4p2b/1B3P2/2P2q2/8 w - - 6 44")
game = Game()
game.board = board
game.player1 = FakeDiscordUser(id=1)
game.player2 = FakeDiscordUser(id=2)
game.current_player = game.player1
game.last_eval = Mate(2)
chess_bot = Chess()
chess_bot.games.append(game)
result = asyncio.run(chess_bot.eval_last_move(game))
self.assertFalse(result["blunder"])
if chess_bot.is_stockfish_enabled():
self.assertEqual(result["mate_in"], 2)
else:
self.assertIsNone(result["mate_in"])
def test_eval_last_move_lost_position_blunders_mate(self):
board = chess.Board(
"Q1kr4/1p6/1P3ppp/1Kp1r3/4p2b/1B3P2/2P2q2/8 b - - 5 43")
game = Game()
game.board = board
game.player1 = FakeDiscordUser(id=1)
game.player2 = FakeDiscordUser(id=2)
game.current_player = game.player1
game.last_eval = Cp(1000)
chess_bot = Chess()
chess_bot.games.append(game)
result = asyncio.run(chess_bot.eval_last_move(game))
if chess_bot.is_stockfish_enabled():
self.assertTrue(result["blunder"])
self.assertEqual(result["mate_in"], -2)
else:
self.assertFalse(result["blunder"])
self.assertIsNone(result["mate_in"])
def test_eval_last_move_last_move_blunder_mate_in_one(self):
board = chess.Board()
board.push_san("g4")
board.push_san("e5")
board.push_san("f4")
game = Game()
game.board = board
game.player1 = FakeDiscordUser(id=1)
game.player2 = FakeDiscordUser(id=2)
game.current_player = game.player1
game.last_eval = Cp(0)
chess_bot = Chess()
chess_bot.games.append(game)
result = asyncio.run(chess_bot.eval_last_move(game))
if chess_bot.is_stockfish_enabled():
self.assertTrue(result["blunder"])
self.assertEqual(result["mate_in"], 1)
else:
self.assertFalse(result["blunder"])
self.assertIsNone(result["mate_in"])