def test_end_state_draw():
    from agents.common import check_end_state, GameState
    x = np.zeros((6, 7), dtype=int)
    x.fill(2)
    x[1::2, ::2] = 1
    x[::2, 1::2] = 1
    print(x)
    assert (check_end_state(x, PLAYER1, 1) == GameState.IS_DRAW)
def test_end_state_draw():
    from agents.common import check_end_state, GameState
    x = np.zeros((6, 7), dtype=int)
    x.fill(2)
    x[1::2, ::2] = 1
    x[::2, 1::2] = 1
    print(x)
    assert (check_end_state(x, PLAYER1, 1) == GameState.IS_DRAW)