def test_convert_ek60():
"""Test converting """
# Unpacking data
# tmp = ConvertEK60(ek60_raw_path)
# tmp.load_ek60_raw()
# # Convert to .nc file
# tmp.raw2nc()
tmp = Convert(ek60_raw_path)
# Test saving zarr file
tmp.raw2zarr()
shutil.rmtree(tmp.zarr_path, ignore_errors=True) # delete non-empty folder
# consider alternative using os.walk() if have os-specific errors
# Test saving nc file and perform checks
tmp.raw2nc()
# Read .nc file into an xarray DataArray
ds_beam = xr.open_dataset(tmp.nc_path, group='Beam')
# Check if backscatter data from all channels are identical to those directly unpacked
for idx in tmp.config_datagram['transceivers'].keys():
# idx is channel index assigned by instrument, starting from 1
assert np.any(tmp.power_dict_split[0][idx - 1, :, :]
== # idx-1 because power_dict_split[0] has a numpy array
ds_beam.backscatter_r.sel(
frequency=tmp.config_datagram['transceivers'][idx]
['frequency']).data)
ds_beam.close()
os.remove(tmp.nc_path)
del tmp
def test_convert_AZFP():
# Read in the dataset that will be used to confirm working conversions. Generated from MATLAB code
ds_test = xr.open_dataset(azfp_test_path)
# Unpacking data
# tmp = ConvertAZFP(azfp_01a_path, azfp_xml_path)
# tmp.parse_raw()
tmp = Convert(azfp_01a_path, azfp_xml_path)
# Test saving zarr file
tmp.raw2zarr()
shutil.rmtree(tmp.zarr_path, ignore_errors=True)
# Test saving nc file and perform checks
tmp.raw2nc()
# Test beam group
with xr.open_dataset(tmp.nc_path, group='Beam') as ds_beam:
# Test frequency
assert np.array_equal(ds_test.frequency, ds_beam.frequency)
# Test sea absorption
# assert np.array_equal(ds_test.sea_abs, ds_beam.sea_abs)
# Test ping time
assert np.array_equal(ds_test.ping_time, ds_beam.ping_time)
# Test tilt x and y
assert np.array_equal(ds_test.tilt_x, ds_beam.tilt_x)
assert np.array_equal(ds_test.tilt_y, ds_beam.tilt_y)
# Test backscatter_r
assert np.array_equal(ds_test.backscatter, ds_beam.backscatter_r)
# Test environment group
with xr.open_dataset(tmp.nc_path, group='Environment') as ds_env:
# Test temperature
assert np.array_equal(ds_test.temperature, ds_env.temperature)
# Test sound speed. 1 value is used because sound speed is the same across frequencies
# assert ds_test.sound_speed == ds_env.sound_speed_indicative.values[0]
# with xr.open_dataset(tmp.nc_path, group="Vendor") as ds_vend:
# # Test battery values
# assert np.array_equal(ds_test.battery_main, ds_vend.battery_main)
# assert np.array_equal(ds_test.battery_tx, ds_vend.battery_tx)
ds_test.close()
os.remove(tmp.nc_path)
del tmp