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