1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
import xarray as xr
import numpy as np
from kerchunk import netCDF3
arr = np.random.rand(1, 10, 10)
data = xr.DataArray(
data=arr.squeeze(),
dims=["x", "y"],
name="data",
)
bdata = xr.Dataset({"data": data}, attrs={"attr0": 3}).to_netcdf(
format="NETCDF3_CLASSIC"
)
# Add parameterize test for storage formats (.json, .parquet)
def test_reference_netcdf(m):
m.pipe("data.nc3", bdata)
h = netCDF3.netcdf_recording_file("memory://data.nc3")
out = h.translate()
ds = xr.open_dataset(
out, engine="kerchunk", storage_options={"remote_protocol": "memory"}
)
assert (ds.data == data).all()
|
