def test_drpsys_2_instruments(drpmocker):
"""Test that two DRPs are returned"""
drpdata1 = pkgutil.get_data('numina.drps.tests', 'drptest1.yaml')
drpdata2 = pkgutil.get_data('numina.drps.tests', 'drptest2.yaml')
drpmocker.add_drp('TEST1', drpdata1)
drpmocker.add_drp('TEST2', drpdata2)
drpsys = DrpSystem()
drpsys.load()
ldrp1 = drpsys.query_by_name('TEST1')
assert ldrp1 is not None
assert ldrp1.name == 'TEST1'
ldrp2 = drpsys.query_by_name('TEST2')
assert ldrp2 is not None
assert ldrp2.name == 'TEST2'
res = drpsys.query_by_name('OTHER')
assert res is None
alldrps = drpsys.query_all()
assert len(alldrps) == 2
assert 'TEST1' in alldrps
# FIXME: We should check that both are equal, not just the name
assert alldrps['TEST1'].name == ldrp1.name
assert 'TEST2' in alldrps
# FIXME: We should check that both are equal, not just the name
assert alldrps['TEST2'].name == ldrp2.name
def test_drpsys_no_instrument():
drpsys = DrpSystem()
drpsys.load()
res = drpsys.query_by_name('TEST1')
assert res is None
res = drpsys.query_by_name('TEST2')
assert res is None
res = drpsys.query_by_name('OTHER')
assert res is None
alldrps = drpsys.query_all()
assert len(alldrps) == 0
def test_drpsys_one_instrument(drpmocker):
"""Test that only one DRP is returned."""
drpdata1 = pkgutil.get_data('numina.drps.tests', 'drptest1.yaml')
drpmocker.add_drp('TEST1', drpdata1)
drpsys = DrpSystem()
drpsys.load()
ldrp = drpsys.query_by_name('TEST1')
assert ldrp is not None
assert ldrp.name == 'TEST1'
res = drpsys.query_by_name('OTHER')
assert res is None
alldrps = drpsys.query_all()
assert len(alldrps) == 1
assert 'TEST1' in alldrps
# FIXME: We should check that both are equal, not just the name
assert alldrps['TEST1'].name == ldrp.name
