def test_index(): a_nc_subset = EddiesObservations.load_file( a_filename, indexs=dict(obs=slice(500, 1000)) ) a_zarr_subset = EddiesObservations.load_from_zarr( memory_store, indexs=dict(obs=slice(500, 1000)), buffer_size=50 ) assert a_nc_subset == a_zarr_subset
import zarr from py_eddy_tracker.data import get_path from py_eddy_tracker.observations.observation import EddiesObservations a_filename, c_filename = ( get_path("Anticyclonic_20190223.nc"), get_path("Cyclonic_20190223.nc"), ) a = EddiesObservations.load_file(a_filename) a_raw = EddiesObservations.load_file(a_filename, raw_data=True) memory_store = zarr.group() # Dataset was raw loaded from netcdf and save in zarr a_raw.to_zarr(memory_store, chunck_size=100000) # We load zarr data without raw option a_zarr = EddiesObservations.load_from_zarr(memory_store) c = EddiesObservations.load_file(c_filename) def test_merge(): new = a.merge(c) assert len(new) == len(a) + len(c) def test_zarr_raw(): assert a == a_zarr def test_index(): a_nc_subset = EddiesObservations.load_file( a_filename, indexs=dict(obs=slice(500, 1000)))