def test_end_game(self): grid_shape = GRID_SHAPE game_state = GameState(grid_shape=grid_shape, currentPlayer=PLAYER_1, board=np.full(grid_shape, PLAYER_1, dtype=np.int8)) is_end_game = game_state.checkForEndGame() self.assertEqual(is_end_game, True) game_state = GameState(grid_shape=grid_shape, currentPlayer=PLAYER_1, board=np.full(grid_shape, NONE, dtype=np.int8)) is_end_game = game_state.checkForEndGame() self.assertEqual(is_end_game, False)