Skip to content

Commit

Permalink
test multi-dimensional vars
Browse files Browse the repository at this point in the history
  • Loading branch information
mark-ep committed Sep 8, 2016
1 parent a6aa472 commit 6d4e4c5
Show file tree
Hide file tree
Showing 6 changed files with 32 additions and 48 deletions.
Binary file modified test_data/data/test_l1b/temp/output.nc
Binary file not shown.
7 changes: 2 additions & 5 deletions test_data/data/test_l1bs/expected/expected.txt
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,10 @@ z_vel_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
meas_x_pos_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
meas_y_pos_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
meas_z_pos_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
roll_sat_pointing_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
pitch_sat_pointing_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
yaw_sat_pointing_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
range_ku_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
i2q2_meas_ku_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
stdev_stack_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
skew_stack_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
kurt_stack_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
i_echoes_ku_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
q_echoes_ku_l1bs_echo_sar_ku 0 1 2 3 4 5 6 7 8 9
i_echoes_ku_l1bs_echo_sar_ku 0.1 0.4 0.2 0.3 0.8 0.7 0.5 0.9 0.6 1.0
q_echoes_ku_l1bs_echo_sar_ku 0.1 0.4 0.2 0.3 0.8 0.7 0.5 0.9 0.6 1.0
2 changes: 1 addition & 1 deletion test_data/data/test_l1bs/inputs/input.txt

Large diffs are not rendered by default.

Binary file modified test_data/data/test_l1bs/temp/output.nc
Binary file not shown.
18 changes: 7 additions & 11 deletions tests/data/test_l1b.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from tests.testing import TestDataLoader
import netCDF4 as nc
from math import degrees
import numpy as np

from dedop.data.output import L1BWriter
from dedop.model.surface_data import SurfaceData, SurfaceType
Expand Down Expand Up @@ -96,7 +97,7 @@ def test_output(self):
stack_skewness=data["stack_skewness"][i],
stack_kurtosis=data["stack_kurtosis"][i],
beam_angles_surf=data["beam_angles_surf"],
waveform_multilooked=data["waveform_multilooked"],
waveform_multilooked=data["waveform_multilooked"][i] * np.ones((256)),
closest_burst_index=0,
stack_bursts=[burst]
)
Expand Down Expand Up @@ -144,12 +145,6 @@ def test_output(self):
degrees(expected["lon_l1b_echo_sar_ku"][i]),
places=6
)

# TODO: there's an offset going on here
# self.assertAlmostEqual(
# output.variables["alt_l1b_echo_sar_ku"][i],
# expected["alt_l1b_echo_sar_ku"][i]
# )
self.assertAlmostEqual(
output.variables["orb_alt_rate_l1b_echo_sar_ku"][i],
expected["orb_alt_rate_l1b_echo_sar_ku"][i]
Expand Down Expand Up @@ -207,7 +202,8 @@ def test_output(self):
# output.variables["beam_ang_l1b_echo_sar_ku"][i],
# expected["beam_ang_l1b_echo_sar_ku"][i]
# )
# self.assertAlmostEqual(
# output.variables["i2q2_meas_ku_l1b_echo_sar_ku"][i],
# expected["i2q2_meas_ku_l1b_echo_sar_ku"][i]
# )
for j in range(256):
self.assertAlmostEqual(
output.variables["i2q2_meas_ku_l1b_echo_sar_ku"][i, j],
expected["i2q2_meas_ku_l1b_echo_sar_ku"][i]
)
53 changes: 22 additions & 31 deletions tests/data/test_l1bs.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,9 +40,14 @@ def test_output(self):
writer.open()

# write outputs
stack_data = np.reshape(
data["beams_range_compr_iq"], (data["count"], 240, 256)
# stack_data = np.reshape(
# data["beams_range_compr_iq"], (data["count"], 240, 256)
# )
a, b = np.meshgrid(
np.linspace(-10, 10, 256),
np.linspace(-5, 5, 240)
)
stack_data = a + 1j * b

for i in range(data["count"]):
burst = L1AProcessingData(
Expand Down Expand Up @@ -80,7 +85,7 @@ def test_output(self):
stack_std=data["stack_std"][i],
stack_skewness=data["stack_skewness"][i],
stack_kurtosis=data["stack_kurtosis"][i],
beams_range_compr_iq=stack_data[i, :, :],
beams_range_compr_iq=stack_data * data["beams_range_compr_iq"][i],
closest_burst_index=0,
stack_bursts=[burst]
)
Expand Down Expand Up @@ -124,10 +129,6 @@ def test_output(self):
output.variables["surf_type_l1bs_echo_sar_ku"][i],
expected["surf_type_l1bs_echo_sar_ku"][i]
)
# self.assertAlmostEqual(
# output.variables["alt_l1bs_echo_sar_ku"][i],
# expected["alt_l1bs_echo_sar_ku"][i]
# )
self.assertAlmostEqual(
output.variables["orb_alt_rate_l1bs_echo_sar_ku"][i],
expected["orb_alt_rate_l1bs_echo_sar_ku"][i]
Expand Down Expand Up @@ -168,18 +169,6 @@ def test_output(self):
output.variables["meas_z_pos_l1bs_echo_sar_ku"][i],
expected["meas_z_pos_l1bs_echo_sar_ku"][i]
)
# self.assertAlmostEqual(
# output.variables["roll_sat_pointing_l1bs_echo_sar_ku"][i],
# expected["roll_sat_pointing_l1bs_echo_sar_ku"][i]
# )
# self.assertAlmostEqual(
# output.variables["pitch_sat_pointing_l1bs_echo_sar_ku"][i],
# expected["pitch_sat_pointing_l1bs_echo_sar_ku"][i]
# )
# self.assertAlmostEqual(
# output.variables["yaw_sat_pointing_l1bs_echo_sar_ku"][i],
# expected["yaw_sat_pointing_l1bs_echo_sar_ku"][i]
# )
self.assertAlmostEqual(
output.variables["stdev_stack_l1bs_echo_sar_ku"][i],
expected["stdev_stack_l1bs_echo_sar_ku"][i]
Expand All @@ -192,15 +181,17 @@ def test_output(self):
output.variables["kurt_stack_l1bs_echo_sar_ku"][i],
expected["kurt_stack_l1bs_echo_sar_ku"][i]
)
# self.assertAlmostEqual(
# output.variables["i2q2_meas_ku_l1bs_echo_sar_ku"][i],
# expected["i2q2_meas_ku_l1bs_echo_sar_ku"][i]
# )
# self.assertAlmostEqual(
# output.variables["i_echoes_ku_l1bs_echo_sar_ku"][i],
# expected["i_echoes_ku_l1bs_echo_sar_ku"][i]
# )
# self.assertAlmostEqual(
# output.variables["q_echoes_ku_l1bs_echo_sar_ku"][i],
# expected["q_echoes_ku_l1bs_echo_sar_ku"][i]
# )
for j in range(256):
self.assertAlmostEqual(
output.variables["i2q2_meas_ku_l1bs_echo_sar_ku"][i, j],
expected["i2q2_meas_ku_l1bs_echo_sar_ku"][i]
)
for k in range(240):
self.assertAlmostEqual(
output.variables["i_echoes_ku_l1bs_echo_sar_ku"][i, j, k],
expected["i_echoes_ku_l1bs_echo_sar_ku"][i] * stack_data[k, j].real
)
self.assertAlmostEqual(
output.variables["q_echoes_ku_l1bs_echo_sar_ku"][i, j, k],
expected["q_echoes_ku_l1bs_echo_sar_ku"][i] * stack_data[j, k].imag
)

0 comments on commit 6d4e4c5

Please sign in to comment.