单元测试问题

问题描述

我想测试一个功能，但我肯定在这个功能上很挣扎。如果 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)

谁能帮帮我？ 非常感谢！

解决方法

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

使用 Python's all() 而不是 Numpy 的 np.all()：

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