def test_ptrPyArrayObjectReturnArg2(pymod_test_mod, random_1d_array_size,
input_type):
arg = array_utils.get_random_1d_array_of_size_and_type(
random_1d_array_size, input_type)
res = pymod_test_mod.ptrPyArrayObjectReturnArg(arg)
assert numpy.all(res == arg)
assert type(res) == type(arg)
assert res.dtype == arg.dtype
def test_returnNdAttr_1d(pymod_test_mod, input_type, random_1d_array_size):
arg = array_utils.get_random_1d_array_of_size_and_type(random_1d_array_size, input_type)
resNd = pymod_test_mod.returnNdAttr(arg)
assert resNd == len(arg.shape)
assert resNd == arg.ndim
resNdim = pymod_test_mod.returnNdimAttr(arg)
assert resNdim == len(arg.shape)
assert resNdim == arg.ndim
def test_returnStridesAsTuple1D2(pymod_test_mod, input_type, random_1d_array_size):
arg = array_utils.get_random_1d_array_of_size_and_type(random_1d_array_size, input_type)
expectedStrides = arg.strides
resStrides = pymod_test_mod.returnStridesAsTuple1D(arg)
# FIXME: Currently, Pymod incorrectly unwraps single-element-tuple return-types.
# Thus, `resStrides` should be a tuple-of-single-int, but instead it's an int.
# We will fix this soon...
resStrides = (resStrides,)
assert resStrides == expectedStrides
def test_returnNdAttr_1d(pymod_test_mod, input_type, random_1d_array_size):
arg = array_utils.get_random_1d_array_of_size_and_type(
random_1d_array_size, input_type)
resNd = pymod_test_mod.returnNdAttr(arg)
assert resNd == len(arg.shape)
assert resNd == arg.ndim
resNdim = pymod_test_mod.returnNdimAttr(arg)
assert resNdim == len(arg.shape)
assert resNdim == arg.ndim
def test_returnStridesAsTuple1D2(pymod_test_mod, input_type,
random_1d_array_size):
arg = array_utils.get_random_1d_array_of_size_and_type(
random_1d_array_size, input_type)
expectedStrides = arg.strides
resStrides = pymod_test_mod.returnStridesAsTuple1D(arg)
# FIXME: Currently, Pymod incorrectly unwraps single-element-tuple return-types.
# Thus, `resStrides` should be a tuple-of-single-int, but instead it's an int.
# We will fix this soon...
resStrides = (resStrides, )
assert resStrides == expectedStrides
def test_ptrPyArrayObjectReturnArg2(pymod_test_mod, random_1d_array_size, input_type):
arg = array_utils.get_random_1d_array_of_size_and_type(random_1d_array_size, input_type)
res = pymod_test_mod.ptrPyArrayObjectReturnArg(arg)
assert numpy.all(res == arg)
assert type(res) == type(arg)
assert res.dtype == arg.dtype
def test_returnPyArrayObjectPtrAsInt(pymod_test_mod, random_1d_array_size, input_type):
arg = array_utils.get_random_1d_array_of_size_and_type(random_1d_array_size, input_type)
res = pymod_test_mod.returnPyArrayObjectPtrAsInt(arg)
assert res == id(arg)
def test_returnFloat64DataPtrAsInt_1d(pymod_test_mod, random_1d_array_size):
arg = array_utils.get_random_1d_array_of_size_and_type(random_1d_array_size, numpy.float64)
res = pymod_test_mod.returnFloat64DataPtrAsInt(arg)
data_addr = _get_array_data_address(arg)
assert res == data_addr
def test_returnPyArrayObjectPtrAsInt(pymod_test_mod, random_1d_array_size,
input_type):
arg = array_utils.get_random_1d_array_of_size_and_type(
random_1d_array_size, input_type)
res = pymod_test_mod.returnPyArrayObjectPtrAsInt(arg)
assert res == id(arg)
def random_1d_array_of_integers(random_1d_array_size):
"""Return a randomly-sized 1-D array of random integers in the range [-100, 100]."""
return get_random_1d_array_of_size_and_type(random_1d_array_size)
def random_1d_array_of_bool(random_1d_array_size):
"""Return a randomly-sized array of random bool values."""
return get_random_1d_array_of_size_and_type(random_1d_array_size,
numpy.bool)
