def dataset_to_fieldset(val, **kwarg):
# we try to import xarray as locally as possible to reduce startup time
# try to write the xarray as a GRIB file, then read into a fieldset
import xarray as xr
from cfgrib.xarray_to_grib import to_grib
if not isinstance(val, xr.core.dataset.Dataset):
raise TypeError(
"dataset_to_fieldset requires a variable of type xr.core.dataset.Dataset;"
" was supplied with ",
builtins.type(val),
)
f, tmp = tempfile.mkstemp(".grib")
os.close(f)
try:
# could add keys, e.g. grib_keys={'centre': 'ecmf'})
to_grib(val, tmp, **kwarg)
except:
print(
"Error trying to write xarray dataset to GRIB for conversion to Metview Fieldset"
)
raise
# TODO: tell Metview that this is a temporary file that should be deleted when no longer needed
fs = read(tmp)
fs.set_temporary(1)
return fs
def test_to_grib(canonic_da, tmpdir):
out_path = tmpdir.join("res.grib")
canonic_ds = canonic_da.to_dataset(name="t")
with pytest.warns(FutureWarning):
xarray_to_grib.to_grib(canonic_ds, str(out_path))
def test_to_grib(canonic_da: xr.DataArray, tmpdir: py.path.local) -> None:
out_path = tmpdir.join("res.grib")
canonic_ds = canonic_da.to_dataset(name="t")
with pytest.warns(FutureWarning):
xarray_to_grib.to_grib(canonic_ds, str(out_path))