hidefix 0.12.0

Concurrent HDF5 and NetCDF4 reader (experimental)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

import xarray as xr
import numpy as np
import netCDF4 as nc4


def test_open_subgroup(tmp_path) -> None:
    # Create a netCDF file with a dataset stored within a group within a
    # group
    tmp_file = tmp_path / 'test.nc'
    rootgrp = nc4.Dataset(tmp_file, "w")
    foogrp = rootgrp.createGroup("foo")
    bargrp = foogrp.createGroup("bar")
    ds = bargrp
    ds.createDimension("time", size=10)
    x = np.arange(10)
    ds.createVariable("x", np.int32, dimensions=("time", ))
    ds.variables["x"][:] = x
    rootgrp.close()

    expected = xr.Dataset()
    expected["x"] = ("time", x)

    # check equivalent ways to specify group
    for group in "foo/bar", "/foo/bar", "foo/bar/", "/foo/bar/":
        actual = xr.open_dataset(tmp_file, engine='netcdf4', group=group)
        assert np.all(actual["x"] == expected["x"])

        hfx = xr.open_dataset(tmp_file, engine='hidefix', group=group)
        assert np.all(hfx["x"] == expected["x"])