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']