def test_not_dtype_object(self, ndim): nav_shape = tuple(np.random.randint(2, 5, size=ndim)) shape = nav_shape + (23, 2) peak_array = np.random.randint(99, size=shape) shape_output = ct._get_peak_array_shape(peak_array) assert nav_shape == shape_output
def test_dtype_object(self, ndim): shape = tuple(np.random.randint(2, 5, size=ndim)) peak_array = np.empty(shape, dtype=np.object) shape_output = ct._get_peak_array_shape(peak_array) assert shape == shape_output