def test_find_data_sources_default_data_store(self): size_before = len(DATA_STORE_REGISTRY) orig_stores = list(DATA_STORE_REGISTRY.get_data_stores()) try: DATA_STORE_REGISTRY._data_stores.clear() self.assertEqual(0, len(DATA_STORE_REGISTRY)) from cate.ds.esa_cci_ftp import set_default_data_store as set_default_data_store_ftp set_default_data_store_ftp() self.assertEqual(1, len(DATA_STORE_REGISTRY)) data_sources = find_data_sources() self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 98) self.assertEqual(data_sources[0].id, "AEROSOL_ATSR2_SU_L3_V4.2_DAILY") data_sources = find_data_sources( ds_id="AEROSOL_ATSR2_SU_L3_V4.2_DAILY") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 1) data_sources = find_data_sources(ds_id="ZZ") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 0) finally: DATA_STORE_REGISTRY._data_stores.clear() for data_store in orig_stores: DATA_STORE_REGISTRY.add_data_store(data_store) self.assertEqual(size_before, len(DATA_STORE_REGISTRY))
def test_find_data_sources_default_data_store(self): size_before = len(DATA_STORE_REGISTRY) orig_stores = list(DATA_STORE_REGISTRY.get_data_stores()) try: DATA_STORE_REGISTRY._data_stores.clear() self.assertEqual(0, len(DATA_STORE_REGISTRY)) from cate.ds.esa_cci_ftp import set_default_data_store as set_default_data_store_ftp set_default_data_store_ftp() self.assertEqual(1, len(DATA_STORE_REGISTRY)) data_sources = find_data_sources() self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 98) self.assertEqual(data_sources[0].id, "AEROSOL_ATSR2_SU_L3_V4.2_DAILY") data_sources = find_data_sources(ds_id="AEROSOL_ATSR2_SU_L3_V4.2_DAILY") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 1) data_sources = find_data_sources(ds_id="ZZ") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 0) finally: DATA_STORE_REGISTRY._data_stores.clear() for data_store in orig_stores: DATA_STORE_REGISTRY.add_data_store(data_store) self.assertEqual(size_before, len(DATA_STORE_REGISTRY))
def test_find_data_sources_with_id_and_query_expr_constrains(self): data_sources = find_data_sources(data_stores=self.TEST_DATA_STORE, ds_id="foo", query_expr="aerosol") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 1) self.assertEqual(data_sources[0].id, "aerosol") data_sources = find_data_sources(data_stores=self.TEST_DATA_STORE, ds_id="aerosol", query_expr="foo") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 1) self.assertEqual(data_sources[0].id, "aerosol") data_sources = find_data_sources(data_stores=self.TEST_DATA_STORE, ds_id="foo", query_expr="bar") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 0)
def test_find_data_sources_with_query_expr_constrains(self): data_sources = find_data_sources(data_stores=self.TEST_DATA_STORE, query_expr="aerosol") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 1) self.assertEqual(data_sources[0].id, "aerosol") data_sources = find_data_sources(data_stores=self.TEST_DATA_STORE, query_expr="Z") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 1) self.assertEqual(data_sources[0].id, "ozone") data_sources = find_data_sources(data_stores=self.TEST_DATA_STORE, query_expr="x") self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 0)
def test_find_data_sources_with_data_store_list(self): data_stores = [self.TEST_DATA_STORE, self.TEST_DATA_STORE_SST] data_sources = find_data_sources(data_stores=data_stores) self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 3) self.assertEqual(data_sources[0].id, "aerosol") self.assertEqual(data_sources[1].id, "ozone") self.assertEqual(data_sources[2].id, "sst")
def test_find_data_sources_with_data_store_value(self): data_sources = find_data_sources(data_stores=self.TEST_DATA_STORE) self.assertIsNotNone(data_sources) self.assertEqual(len(data_sources), 2) self.assertEqual(data_sources[0].id, "aerosol") self.assertEqual(data_sources[1].id, "ozone")