def test_require_flag_check(self, dtype):
possible_flags = [['C_CONTIGUOUS'], ['F_CONTIGUOUS']]
x = cupy.zeros((2, 3, 4), dtype)
for flags in possible_flags:
arr = cupy.require(x, dtype, flags)
for parameter in flags:
assert arr.flags[parameter]
assert arr.dtype == dtype
def test_require_incorrect_dtype(self, dtype):
x = cupy.zeros((2, 3, 4), dtype)
with pytest.raises(ValueError):
cupy.require(x, 'random', 'C')
def test_require_incorrect_requirments(self, dtype):
x = cupy.zeros((2, 3, 4), dtype)
with pytest.raises(ValueError):
cupy.require(x, dtype, ['W'])
def test_require_C_and_F_flags(self, dtype):
x = cupy.zeros((2, 3, 4), dtype)
with pytest.raises(ValueError):
cupy.require(x, dtype, ['C', 'F'])
def test_require_owndata(self, dtype):
x = cupy.zeros((2, 3, 4), dtype)
arr = x.view()
arr = cupy.require(arr, dtype, ['O'])
assert arr.flags['OWNDATA']
def test_require_empty_requirements(self, dtype):
x = cupy.zeros((2, 3, 4), dtype)
x = cupy.require(x, dtype, [])
assert x.flags['C_CONTIGUOUS']
