Пример #1
0
    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)
Пример #2
0
    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)