<filename>tests/test_agents_common_initilize_game_state.py
import numpy as np

from agents.common import BoardPiece, NO_PLAYER, initialize_game_state


def test_initialize_game_state():
    ret = initialize_game_state()

    assert isinstance(ret, np.ndarray)
    assert ret.dtype == BoardPiece
    assert ret.shape == (6, 7)
    assert np.all(ret == NO_PLAYER)

<filename>tests/test_agents_common_initilize_game_state.py
import numpy as np

from agents.common import BoardPiece, NO_PLAYER, initialize_game_state


def test_initialize_game_state():
    ret = initialize_game_state()

    assert isinstance(ret, np.ndarray)
    assert ret.dtype == BoardPiece
    assert ret.shape == (6, 7)
    assert np.all(ret == NO_PLAYER)
