def test_query_data_package_get_datasets(self):
models = [CanCM4TestDataset,MaurerTas,MaurerTasmax]
with db.session_scope() as session:
for m in models: m().insert(session)
dataset = session.query(db.Dataset).filter_by(name='Maurer 2010').one()
category = session.query(db.DatasetCategory).filter_by(name='Observational').one()
fields = [c.field[0] for c in dataset.container]
dp = db.DataPackage(field=fields,name='Test Package',description='For testing! Duh...',dataset_category=category)
session.add(dp)
dataset = session.query(db.Dataset).filter_by(name='Maurer 2010').one()
category = session.query(db.DatasetCategory).filter_by(name='GCMs').one()
fields = [c.field[0] for c in dataset.container]
dp = db.DataPackage(field=fields,name='Test Package GCMs',description='For testing! Duh...',dataset_category=category)
session.add(dp)
session.commit()
dq = DataQuery()
ret = dq.get_package_datasets(package_name='Test Package')
rds = [ocgis.RequestDataset(**k) for k in ret['dataset']]
for rd in rds: rd.inspect_as_dct()
with self.assertRaises(MultipleResultsFound):
dq.get_package_datasets()
