def test_open_dataset_duplicated_names(self):
try:
ds_a1 = SimpleDataSource('duplicate')
ds_a2 = SimpleDataSource('duplicate')
duplicated_cat = SimpleDataStore('duplicated_cat', [ds_a1, ds_a2])
ds.DATA_STORE_REGISTRY.add_data_store(duplicated_cat)
with self.assertRaises(ValueError) as cm:
ds.open_dataset('duplicate')
self.assertEqual("2 data_sources found for the given query term 'duplicate'", str(cm.exception))
finally:
ds.DATA_STORE_REGISTRY.remove_data_store('duplicated_cat')
def test_open_dataset(self):
with self.assertRaises(ValueError) as cm:
ds.open_dataset(None)
self.assertTupleEqual(tuple(['No data_source given']), cm.exception.args)
with self.assertRaises(ValueError) as cm:
ds.open_dataset('foo')
self.assertEqual(("No data_source found for the given query term", 'foo'), cm.exception.args)
inmem_data_source = InMemoryDataSource(42)
dataset1 = ds.open_dataset(inmem_data_source)
self.assertIsNotNone(dataset1)
self.assertIsInstance(dataset1, xr.Dataset)
self.assertEqual(42, dataset1.a.values)
dataset2 = inmem_data_source.open_dataset()
self.assertIsInstance(dataset2, xr.Dataset)
self.assertEqual(42, dataset2.a.values)
def test_open_dataset(self):
with self.assertRaises(ValueError) as cm:
# noinspection PyTypeChecker
open_dataset(None)
self.assertTupleEqual(('No data source given',), cm.exception.args)
with self.assertRaises(ValueError) as cm:
open_dataset('foo')
self.assertEqual(("No data sources found for the given ID 'foo'",), cm.exception.args)
inmem_data_source = InMemoryDataSource(42)
dataset1 = open_dataset(inmem_data_source)
self.assertIsNotNone(dataset1)
self.assertIsInstance(dataset1, xr.Dataset)
self.assertEqual(42, dataset1.a.values)
dataset2 = inmem_data_source.open_dataset()
self.assertIsInstance(dataset2, xr.Dataset)
self.assertEqual(42, dataset2.a.values)
def test_open_local(local_dataset):
ds.open_dataset(local_dataset)
pass
