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)

谁能帮我？提前谢谢了！

标签： pythontestingencodingmonte-carlo-tree-search

Answer 1

没有来自 OP 的进一步信息，我认为问题出在最后一条assert语句中。

使用Pythonall()而不是 Numpy 的np.all()：

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