Python ConnectFourGame.get_valid_moves Examples

Example #1

File: test_connect_four.py Project: weirichd/connect_four

def test_get_valid_moves():
    under_test = ConnectFourGame()

    expected = [0, 1, 2, 3, 4, 5, 6]

    actual = under_test.get_valid_moves()

    assert actual == expected

Example #2

File: test_connect_four.py Project: weirichd/connect_four

def test_get_valid_moves_with_some_columns_full():
    under_test = ConnectFourGame()
    under_test.board = np.array(
        [
            [1, 0, 2, 0, 2, 0, 0],
            [1, 0, 2, 0, 1, 0, 0],
            [2, 0, 1, 0, 2, 0, 0],
            [1, 0, 2, 0, 1, 0, 0],
            [1, 0, 2, 0, 2, 0, 1],
            [1, 0, 2, 0, 1, 1, 2],
        ]
    )

    expected = [1, 3, 5, 6]

    actual = under_test.get_valid_moves()

    assert actual == expected