def test_rotate_inst2beam(make_data=False):
td = load('RDI_test01_rotate_beam2inst.nc')
rotate2(td, 'beam', inplace=True)
td_awac = load('AWAC_test01_earth2inst.nc')
rotate2(td_awac, 'beam', inplace=True)
td_sig = load('BenchFile01_rotate_beam2inst.nc')
rotate2(td_sig, 'beam', inplace=True)
td_sig_i = load('Sig1000_IMU_rotate_beam2inst.nc')
rotate2(td_sig_i, 'beam', inplace=True)
td_sig_ie = load('Sig500_Echo_earth2inst.nc')
rotate2(td_sig_ie, 'beam', inplace=True)
if make_data:
save(td_awac, 'AWAC_test01_inst2beam.nc')
save(td_sig_ie, 'Sig500_Echo_inst2beam.nc')
return
cd_td = tr.dat_rdi.copy(deep=True)
cd_awac = load('AWAC_test01_inst2beam.nc')
cd_sig = tr.dat_sig.copy(deep=True)
cd_sig_i = tr.dat_sig_i.copy(deep=True)
cd_sig_ie = load('Sig500_Echo_inst2beam.nc')
# # The reverse RDI rotation doesn't work b/c of NaN's in one beam
# # that propagate to others, so we impose that here.
cd_td['vel'].values[:, np.isnan(cd_td['vel'].values).any(0)] = np.NaN
assert_allclose(td, cd_td, atol=1e-5)
assert_allclose(td_awac, cd_awac, atol=1e-5)
assert_allclose(td_sig, cd_sig, atol=1e-5)
assert_allclose(td_sig_i, cd_sig_i, atol=1e-5)
assert_allclose(td_sig_ie, cd_sig_ie, atol=1e-5)
def test_rotate_earth2inst():
td_rdi = load('RDI_test01_rotate_inst2earth.nc')
rotate2(td_rdi, 'inst', inplace=True)
tdwr2 = load('winriver02_rotate_ship2earth.nc')
rotate2(tdwr2, 'inst', inplace=True)
td_awac = tr.dat_awac.copy(deep=True)
rotate2(td_awac, 'inst', inplace=True) # AWAC is in earth coords
td_sig = load('BenchFile01_rotate_inst2earth.nc')
rotate2(td_sig, 'inst', inplace=True)
td_sig_i = load('Sig1000_IMU_rotate_inst2earth.nc')
rotate2(td_sig_i, 'inst', inplace=True)
cd_rdi = load('RDI_test01_rotate_beam2inst.nc')
cd_wr2 = tr.dat_wr2
# ship and inst are considered equivalent in dolfy
cd_wr2.attrs['coord_sys'] = 'inst'
cd_awac = load('AWAC_test01_earth2inst.nc')
cd_sig = load('BenchFile01_rotate_beam2inst.nc')
cd_sig_i = load('Sig1000_IMU_rotate_beam2inst.nc')
assert_allclose(td_rdi, cd_rdi, atol=1e-5)
assert_allclose(tdwr2, cd_wr2, atol=1e-5)
assert_allclose(td_awac, cd_awac, atol=1e-5)
assert_allclose(td_sig, cd_sig, atol=1e-5)
# known failure due to orientmat, see test_vs_nortek
#assert_allclose(td_sig_i, cd_sig_i, atol=1e-3)
npt.assert_allclose(td_sig_i.accel.values,
cd_sig_i.accel.values,
atol=1e-3)
def rotate(axis):
# BenchFile01.ad2cp
td_sig = rotate2(tr.dat_sig, axis, inplace=False)
# Sig1000_IMU.ad2cp no userdata
td_sig_i = rotate2(tr.dat_sig_i, axis, inplace=False)
# VelEchoBT01.ad2cp
td_sig_ieb = rotate2(tr.dat_sig_ieb, axis, inplace=False)
# Sig500_Echo.ad2cp
td_sig_ie = rotate2(tr.dat_sig_ie, axis, inplace=False)
td_sig_vel = load_nortek_matfile(base.rfnm('BenchFile01.mat'))
td_sig_i_vel = load_nortek_matfile(base.rfnm('Sig1000_IMU.mat'))
td_sig_ieb_vel, vel_bt = load_nortek_matfile(base.rfnm('VelEchoBT01.mat'))
td_sig_ie_vel = load_nortek_matfile(base.rfnm('Sig500_Echo.mat'))
nens = 100
# ARHS inst2earth orientation matrix check
# Checks the 1,1 element because the nortek orientmat's shape is [9,:] as
# opposed to [3,3,:]
if axis == 'inst':
assert_allclose(td_sig_i.orientmat[0][0].values,
td_sig_i_vel['omat'][0, :nens],
atol=1e-7)
assert_allclose(td_sig_ieb.orientmat[0][0].values,
td_sig_ieb_vel['omat'][0, :][..., :nens],
atol=1e-7)
# 4-beam velocity
assert_allclose(td_sig.vel.values, td_sig_vel[axis][..., :nens], atol=1e-5)
assert_allclose(td_sig_i.vel.values,
td_sig_i_vel[axis][..., :nens],
atol=5e-3)
assert_allclose(td_sig_ieb.vel.values,
td_sig_ieb_vel[axis][..., :nens],
atol=5e-3)
assert_allclose(td_sig_ie.vel.values,
td_sig_ie_vel[axis][..., :nens],
atol=1e-5)
# 5th-beam velocity
if axis == 'beam':
assert_allclose(td_sig_i.vel_b5.values,
td_sig_i_vel['b5'][..., :nens],
atol=1e-5)
assert_allclose(td_sig_ieb.vel_b5.values,
td_sig_ieb_vel['b5'][..., :nens],
atol=1e-5)
assert_allclose(td_sig_ie.vel_b5.values,
td_sig_ie_vel['b5'][..., :nens],
atol=1e-5)
# bottom-track
assert_allclose(td_sig_ieb.vel_bt.values,
vel_bt[axis][..., :nens],
atol=5e-3)
def test_rotate_beam2inst():
td = load('vector_data01_rotate_inst2beam.nc')
rotate2(td, 'inst', inplace=True)
tdm = load('vector_data_imu01_rotate_inst2beam.nc')
rotate2(tdm, 'inst', inplace=True)
cd = tr.dat.copy(deep=True)
cdm = tr.dat_imu.copy(deep=True)
assert_allclose(td, cd, atol=1e-6)
assert_allclose(tdm, cdm, atol=1e-6)
def test_rotate_earth2inst():
td = load('vector_data01_rotate_inst2earth.nc')
rotate2(td, 'inst', inplace=True)
tdm = load('vector_data_imu01_rotate_inst2earth.nc')
rotate2(tdm, 'inst', inplace=True)
cd = tr.dat.copy(deep=True)
cdm = tr.dat_imu.copy(deep=True)
# The heading/pitch/roll data gets modified during rotation, so it
# doesn't go back to what it was.
cdm = cdm.drop_vars(['heading', 'pitch', 'roll'])
tdm = tdm.drop_vars(['heading', 'pitch', 'roll'])
assert_allclose(td, cd, atol=1e-6)
assert_allclose(tdm, cdm, atol=1e-6)
def test_rotate_inst2beam(make_data=False):
td = tr.dat.copy(deep=True)
rotate2(td, 'beam', inplace=True)
tdm = tr.dat_imu.copy(deep=True)
rotate2(tdm, 'beam', inplace=True)
if make_data:
save(td, 'vector_data01_rotate_inst2beam.nc')
save(tdm, 'vector_data_imu01_rotate_inst2beam.nc')
return
cd = load('vector_data01_rotate_inst2beam.nc')
cdm = load('vector_data_imu01_rotate_inst2beam.nc')
assert_allclose(td, cd, atol=1e-6)
assert_allclose(tdm, cdm, atol=1e-6)
def test_rotate_warnings(self):
warn1 = tr.dat.copy(deep=True)
warn2 = tr.dat.copy(deep=True)
warn2.attrs['coord_sys'] = 'flow'
warn3 = tr.dat.copy(deep=True)
warn3.attrs['inst_model'] = 'ADV'
warn4 = tr.dat.copy(deep=True)
warn4.attrs['inst_model'] = 'adv'
with self.assertRaises(Exception):
rotate2(warn1, 'ship')
with self.assertRaises(Exception):
rotate2(warn2, 'earth')
with self.assertRaises(Exception):
set_inst2head_rotmat(warn3, np.eye(3))
set_inst2head_rotmat(warn3, np.eye(3))
with self.assertRaises(Exception):
set_inst2head_rotmat(warn4, np.eye(3))
def test_rotate_earth2principal(make_data=False):
td = load('vector_data01_rotate_inst2earth.nc')
td.attrs['principal_heading'] = calc_principal_heading(td['vel'])
rotate2(td, 'principal', inplace=True)
tdm = load('vector_data_imu01_rotate_inst2earth.nc')
tdm.attrs['principal_heading'] = calc_principal_heading(tdm['vel'])
rotate2(tdm, 'principal', inplace=True)
if make_data:
save(td, 'vector_data01_rotate_earth2principal.nc')
save(tdm, 'vector_data_imu01_rotate_earth2principal.nc')
return
cd = load('vector_data01_rotate_earth2principal.nc')
cdm = load('vector_data_imu01_rotate_earth2principal.nc')
assert_allclose(td, cd, atol=1e-6)
assert_allclose(tdm, cdm, atol=1e-6)
def test_rotate_inst2earth(make_data=False):
td = tr.dat.copy(deep=True)
rotate2(td, 'earth', inplace=True)
tdm = tr.dat_imu.copy(deep=True)
rotate2(tdm, 'earth', inplace=True)
tdo = tr.dat.copy(deep=True)
omat = tdo['orientmat']
tdo = rotate2(tdo.drop_vars('orientmat'), 'earth', inplace=False)
tdo['orientmat'] = omat
if make_data:
save(td, 'vector_data01_rotate_inst2earth.nc')
save(tdm, 'vector_data_imu01_rotate_inst2earth.nc')
return
cd = load('vector_data01_rotate_inst2earth.nc')
cdm = load('vector_data_imu01_rotate_inst2earth.nc')
assert_allclose(td, cd, atol=1e-6)
assert_allclose(tdm, cdm, atol=1e-6)
assert_allclose(tdo, cd, atol=1e-6)
def test_rotate_earth2principal(make_data=False):
td_rdi = load('RDI_test01_rotate_inst2earth.nc')
td_sig = load('BenchFile01_rotate_inst2earth.nc')
td_awac = tr.dat_awac.copy(deep=True)
td_rdi.attrs['principal_heading'] = calc_principal_heading(
td_rdi.vel.mean('range'))
td_sig.attrs['principal_heading'] = calc_principal_heading(
td_sig.vel.mean('range'))
td_awac.attrs['principal_heading'] = calc_principal_heading(
td_awac.vel.mean('range'), tidal_mode=False)
rotate2(td_rdi, 'principal', inplace=True)
rotate2(td_sig, 'principal', inplace=True)
rotate2(td_awac, 'principal', inplace=True)
if make_data:
save(td_rdi, 'RDI_test01_rotate_earth2principal.nc')
save(td_sig, 'BenchFile01_rotate_earth2principal.nc')
save(td_awac, 'AWAC_test01_earth2principal.nc')
return
cd_rdi = load('RDI_test01_rotate_earth2principal.nc')
cd_sig = load('BenchFile01_rotate_earth2principal.nc')
cd_awac = load('AWAC_test01_earth2principal.nc')
assert_allclose(td_rdi, cd_rdi, atol=1e-5)
assert_allclose(td_awac, cd_awac, atol=1e-5)
assert_allclose(td_sig, cd_sig, atol=1e-5)
def test_rotate_beam2inst(make_data=False):
td_rdi = rotate2(tr.dat_rdi, 'inst', inplace=False)
td_sig = rotate2(tr.dat_sig, 'inst', inplace=False)
td_sig_i = rotate2(tr.dat_sig_i, 'inst', inplace=False)
td_sig_ieb = rotate2(tr.dat_sig_ieb, 'inst', inplace=False)
if make_data:
save(td_rdi, 'RDI_test01_rotate_beam2inst.nc')
save(td_sig, 'BenchFile01_rotate_beam2inst.nc')
save(td_sig_i, 'Sig1000_IMU_rotate_beam2inst.nc')
save(td_sig_ieb, 'VelEchoBT01_rotate_beam2inst.nc')
return
cd_rdi = load('RDI_test01_rotate_beam2inst.nc')
cd_sig = load('BenchFile01_rotate_beam2inst.nc')
cd_sig_i = load('Sig1000_IMU_rotate_beam2inst.nc')
cd_sig_ieb = load('VelEchoBT01_rotate_beam2inst.nc')
assert_allclose(td_rdi, cd_rdi, atol=1e-5)
assert_allclose(td_sig, cd_sig, atol=1e-5)
assert_allclose(td_sig_i, cd_sig_i, atol=1e-5)
assert_allclose(td_sig_ieb, cd_sig_ieb, atol=1e-5)
def test_rotate_earth2principal_set_declination():
declin = 3.875
td = load('vector_data01_rotate_inst2earth.nc')
td0 = td.copy(deep=True)
td.attrs['principal_heading'] = calc_principal_heading(td['vel'])
rotate2(td, 'principal', inplace=True)
set_declination(td, declin, inplace=True)
rotate2(td, 'earth', inplace=True)
set_declination(td0, -1, inplace=True)
set_declination(td0, declin, inplace=True)
td0.attrs['principal_heading'] = calc_principal_heading(td0['vel'])
rotate2(td0, 'earth', inplace=True)
assert_allclose(td0, td, atol=1e-6)
def test_rotate_inst2earth(make_data=False):
# AWAC & Sig500 are loaded in earth
td_awac = tr.dat_awac.copy(deep=True)
rotate2(td_awac, 'inst', inplace=True)
td_sig_ie = tr.dat_sig_ie.copy(deep=True)
rotate2(td_sig_ie, 'inst', inplace=True)
td_sig_o = td_sig_ie.copy(deep=True)
td = rotate2(tr.dat_rdi, 'earth', inplace=False)
tdwr2 = rotate2(tr.dat_wr2, 'earth', inplace=False)
td_sig = load('BenchFile01_rotate_beam2inst.nc')
rotate2(td_sig, 'earth', inplace=True)
td_sig_i = load('Sig1000_IMU_rotate_beam2inst.nc')
rotate2(td_sig_i, 'earth', inplace=True)
if make_data:
save(td_awac, 'AWAC_test01_earth2inst.nc')
save(td, 'RDI_test01_rotate_inst2earth.nc')
save(tdwr2, 'winriver02_rotate_ship2earth.nc')
save(td_sig, 'BenchFile01_rotate_inst2earth.nc')
save(td_sig_i, 'Sig1000_IMU_rotate_inst2earth.nc')
save(td_sig_ie, 'Sig500_Echo_earth2inst.nc')
return
td_awac = rotate2(load('AWAC_test01_earth2inst.nc'),
'earth',
inplace=False)
td_sig_ie = rotate2(load('Sig500_Echo_earth2inst.nc'),
'earth',
inplace=False)
td_sig_o = rotate2(td_sig_o.drop_vars('orientmat'), 'earth', inplace=False)
cd = load('RDI_test01_rotate_inst2earth.nc')
cdwr2 = load('winriver02_rotate_ship2earth.nc')
cd_sig = load('BenchFile01_rotate_inst2earth.nc')
cd_sig_i = load('Sig1000_IMU_rotate_inst2earth.nc')
assert_allclose(td, cd, atol=1e-5)
assert_allclose(tdwr2, cdwr2, atol=1e-5)
assert_allclose(td_awac, tr.dat_awac, atol=1e-5)
assert_allclose(td_sig, cd_sig, atol=1e-5)
assert_allclose(td_sig_i, cd_sig_i, atol=1e-5)
assert_allclose(td_sig_ie, tr.dat_sig_ie, atol=1e-5)
npt.assert_allclose(td_sig_o.vel, tr.dat_sig_ie.vel, atol=1e-5)