def test_cspp_nutnr_uses_ctd_pressure(self): nut_fn = 'nutnr_j_cspp_instrument_recovered.nc' ctd_fn = 'ctdpf_j_cspp_instrument_recovered.nc' nut_ds = xr.open_dataset(os.path.join(DATA_DIR, nut_fn), decode_times=False) ctd_ds = xr.open_dataset(os.path.join(DATA_DIR, ctd_fn), decode_times=False) tr = TimeRange(nut_ds.time.values[0], nut_ds.time.values[-1]) sr = StreamRequest(self.nut_sk2, [], tr, {}, request_id='UNIT') nut_ds = nut_ds[self.base_params + [p.name for p in sr.stream_parameters[self.nut_sk2]]] sr.datasets[self.ctd_sk2] = StreamDataset(self.ctd_sk2, sr.uflags, [self.nut_sk2], sr.request_id) sr.datasets[self.nut_sk2] = StreamDataset(self.nut_sk2, sr.uflags, [self.ctd_sk2], sr.request_id) sr.datasets[self.ctd_sk2]._insert_dataset(ctd_ds) sr.datasets[self.nut_sk2]._insert_dataset(nut_ds) sr.calculate_derived_products() sr.import_extra_externals() sr.rename_parameters() self.assertIn('int_ctd_pressure', sr.datasets[self.nut_sk2].datasets[1]) data = json.loads(JsonResponse(sr).json()) for each in data: self.assertIn('int_ctd_pressure', each)
def test_pressure_depth_renamed(self): dosta_fn = 'dosta_abcdjm_cspp_instrument_recovered.nc' ctd_fn = 'ctdpf_j_cspp_instrument_recovered.nc' dosta_ds = xr.open_dataset(os.path.join(DATA_DIR, dosta_fn), decode_times=False) ctd_ds = xr.open_dataset(os.path.join(DATA_DIR, ctd_fn), decode_times=False) tr = TimeRange(dosta_ds.time.values[0], dosta_ds.time.values[-1]) sr = StreamRequest(self.dosta_sk, [], tr, {}, request_id='UNIT') dosta_ds = dosta_ds[ self.base_params + [p.name for p in sr.stream_parameters[self.dosta_sk]]] sr.datasets[self.dosta_sk] = StreamDataset(self.dosta_sk, sr.uflags, [self.ctd_sk2], sr.request_id) sr.datasets[self.ctd_sk2] = StreamDataset(self.ctd_sk2, sr.uflags, [self.dosta_sk], sr.request_id) sr.datasets[self.dosta_sk]._insert_dataset(dosta_ds) sr.datasets[self.ctd_sk2]._insert_dataset(ctd_ds) sr.calculate_derived_products() sr.import_extra_externals() sr.rename_parameters() self.assertNotIn('pressure_depth', sr.datasets[self.dosta_sk].datasets[1]) self.assertIn('pressure', sr.datasets[self.dosta_sk].datasets[1]) data = json.loads(JsonResponse(sr).json()) for each in data: self.assertNotIn('pressure_depth', each) self.assertIn('pressure', each)