test_modelgrids.py 文件源码

def __call__(self,filename=None,varname=None):
    if self.array_type == 'numpy':
            out = Dataset(filename).variables[varname][:].squeeze()
        elif self.array_type == 'xarray':
            ds = xr.open_dataset(filename,chunks=self.chunks,lock=False)
            out = ds[varname]
    elif self.array_type == 'dask_from_numpy':
        d = Dataset(filename).variables[varname][:].squeeze()
        out = da.from_array(np.array(d), chunks=self.chunks)
        elif self.array_type == 'dask_from_netcdf':
            d = Dataset(filename).variables[varname]
            out = da.from_array(d, chunks=self.chunks)
        return out