def test_function_map_scalar(self):
echo_fn = 'echo_sounding.nc'
echo_ds = xr.open_dataset(os.path.join(DATA_DIR, echo_fn), decode_times=False)
echo_sk = StreamKey('RS01SLBS', 'LJ01A', '05-HPIESA101', 'streamed', 'echo_sounding')
tr = TimeRange(0, 99999999)
sr = StreamRequest(echo_sk, [], {}, tr, {}, request_id='UNIT')
sr.datasets[echo_sk] = StreamDataset(echo_sk, sr.coefficients, sr.uflags, [], sr.request_id)
sr.datasets[echo_sk]._insert_dataset(echo_ds)
sr.calculate_derived_products()
sr._add_location()
expected = {'hpies_travel_time1_L1', 'hpies_travel_time2_L1', 'hpies_travel_time3_L1', 'hpies_travel_time4_L1',
'hpies_bliley_temperature_L1', 'hpies_pressure_L1'}
missing = expected.difference(sr.datasets[echo_sk].datasets[0])
self.assertSetEqual(missing, set())
def test_function_map_scalar(self):
echo_fn = 'echo_sounding.nc'
echo_ds = xr.open_dataset(os.path.join(DATA_DIR, echo_fn),
decode_times=False)
echo_sk = StreamKey('RS01SLBS', 'LJ01A', '05-HPIESA101', 'streamed',
'echo_sounding')
tr = TimeRange(0, 99999999)
sr = StreamRequest(echo_sk, [], {}, tr, {}, request_id='UNIT')
sr.datasets[echo_sk] = StreamDataset(echo_sk, sr.coefficients,
sr.uflags, [], sr.request_id)
sr.datasets[echo_sk]._insert_dataset(echo_ds)
sr.calculate_derived_products()
sr._add_location()
expected = {
'hpies_travel_time1_L1', 'hpies_travel_time2_L1',
'hpies_travel_time3_L1', 'hpies_travel_time4_L1',
'hpies_bliley_temperature_L1', 'hpies_pressure_L1'
}
missing = expected.difference(sr.datasets[echo_sk].datasets[0])
self.assertSetEqual(missing, set())
def test_add_location(self):
echo_fn = 'echo_sounding.nc'
echo_ds = xr.open_dataset(os.path.join(DATA_DIR, echo_fn),
decode_times=False)
echo_ds.deployment.values[:20] = 1
echo_ds.deployment.values[20:] = 2
echo_sk = StreamKey('RS01SLBS', 'LJ01A', '05-HPIESA101', 'streamed',
'echo_sounding')
location_info = {
echo_sk.as_three_part_refdes(): [{
'deployment': 1,
'lat': 1,
'lon': 5
}, {
'deployment': 2,
'lat': 2,
'lon': 6
}]
}
tr = TimeRange(0, 99999999)
sr = StreamRequest(echo_sk, [], {},
tr, {},
location_information=location_info,
request_id='UNIT')
sr.datasets[echo_sk] = StreamDataset(echo_sk, sr.coefficients,
sr.uflags, [], sr.request_id)
sr.datasets[echo_sk]._insert_dataset(echo_ds)
sr.calculate_derived_products()
sr._add_location()
ds = sr.datasets[echo_sk]
for deployment, lat, lon in [(1, 1.0, 5.0), (2, 2.0, 6.0)]:
lats = set(np.unique(ds.datasets[deployment].lat.values))
lons = set(np.unique(ds.datasets[deployment].lon.values))
self.assertSetEqual(lats, {lat})
self.assertSetEqual(lons, {lon})
def test_add_location(self):
echo_fn = 'echo_sounding.nc'
echo_ds = xr.open_dataset(os.path.join(DATA_DIR, echo_fn), decode_times=False)
echo_ds.deployment.values[:20] = 1
echo_ds.deployment.values[20:] = 2
echo_sk = StreamKey('RS01SLBS', 'LJ01A', '05-HPIESA101', 'streamed', 'echo_sounding')
location_info = {echo_sk.as_three_part_refdes(): [{'deployment': 1, 'lat': 1, 'lon': 5},
{'deployment': 2, 'lat': 2, 'lon': 6}]}
tr = TimeRange(0, 99999999)
sr = StreamRequest(echo_sk, [], {}, tr, {}, location_information=location_info, request_id='UNIT')
sr.datasets[echo_sk] = StreamDataset(echo_sk, sr.coefficients, sr.uflags, [], sr.request_id)
sr.datasets[echo_sk]._insert_dataset(echo_ds)
sr.calculate_derived_products()
sr._add_location()
ds = sr.datasets[echo_sk]
for deployment, lat, lon in [(1, 1.0, 5.0), (2, 2.0, 6.0)]:
lats = set(np.unique(ds.datasets[deployment].lat.values))
lons = set(np.unique(ds.datasets[deployment].lon.values))
self.assertSetEqual(lats, {lat})
self.assertSetEqual(lons, {lon})
