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)