python - 単体テストの問題

Question

関数をテストしたいのですが、これは間違いなく苦労しています。NO_PLayer が有効な場所である場合、関数はボードの最後の行をループします。

def validLocations(board):

    validLocationsArr = []

    column = 0
    row = 0
    while column < 6:

        if (board[5][column] == NO_PLAYER):
                validLocationsArr.append(column)

        column += 1

    return validLocationsArr

私はそれに取り組み、試してみましたが、

def test_validLocations(self):
    from agents.common import validLocations

    ret = validLocations(board)
    assert np.all(ret == NO_PLAYER)

誰でも私を助けることができますか？よろしくお願いします！

Answer 1

OPからの詳細な情報がなければ、問題は最後のassertステートメントにあると思います。

Numpyの代わりに Pythonall()を使用しnp.all()ます。

ret = validLocations(board)  # as before
assert all(loc == NO_PLAYER for loc in ret)