def test_one_argument(self):
    for xp in (numpy, cupy):
        cond = testing.shaped_random((3, 4), xp, dtype=xp.bool_)
        x = testing.shaped_random((2, 3, 4), xp, xp.int32)
        with pytest.raises(ValueError):
            xp.where(cond, x)
def test_one_argument(self):
    for xp in (numpy, cupy):
        cond = testing.shaped_random((3, 4), xp, dtype=xp.bool_)
        x = testing.shaped_random((2, 3, 4), xp, xp.int32)
        with pytest.raises(ValueError):
            xp.where(cond, x)