def _arg_func_keepdims(func, *args): base = GridData(np.arange(30).reshape(5, 6), axis_labels=('x', 'y'), axis_units=('a', 'b'), voxel_size=(1, 2), axis_offsets=(1, 0)) r = getattr(base, func)(*args, keepdims=True) r2 = getattr(base.view(ndarray), func)(*args, keepdims=True) assert_array_equal(r, r2)
def test_explicit_creation(): base = np.arange(30).reshape(5, 6) tt = GridData(base) assert_array_equal(base, tt.view(ndarray))