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_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_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")
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")
