def METHOD_NAME(cond):
    a_np = np.array(cond)
    x_np = np.array([[1, 2], [3, 4]])
    y_np = np.array([[9, 8], [7, 6]])
    a_num = num.array(a_np)
    x_num = num.array(x_np)
    y_num = num.array(y_np)
    assert np.array_equal(
        np.where(a_np, x_np, y_np), num.where(a_num, x_num, y_num)
    )