def test_split_database(split_function_database, database):
#Send all entries with instrument='EIA' to destination_database
split_function_database, database = split_database(split_function_database, database, vso.attrs.Instrument('EIA'))
observed_source_entries = split_function_database.query(vso.attrs.Provider('xyz'), sortby='id')
observed_destination_entries = database.query(vso.attrs.Provider('xyz'))
assert observed_source_entries == [
DatabaseEntry(id=1,instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=2,instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=3,instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=5,instrument='AIA_3', provider='xyz'),
DatabaseEntry(id=6,instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=7,instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=9,instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=10,instrument='AIA_3', provider='xyz'),
]
assert observed_destination_entries == [
DatabaseEntry(id=4,instrument='EIA', provider='xyz'),
DatabaseEntry(id=8,instrument='EIA', provider='xyz'),
]
def test_split_database(split_function_database, database):
# Send all entries with instrument='EIA' to destination_database
split_function_database, database = split_database(
split_function_database, database, vso.attrs.Instrument('EIA'))
observed_source_entries = split_function_database.query(
vso.attrs.Provider('xyz'), sortby='id')
observed_destination_entries = database.query(vso.attrs.Provider('xyz'))
assert observed_source_entries == [
DatabaseEntry(id=1, instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=2, instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=3, instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=5, instrument='AIA_3', provider='xyz'),
DatabaseEntry(id=6, instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=7, instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=9, instrument='RHESSI', provider='xyz'),
DatabaseEntry(id=10, instrument='AIA_3', provider='xyz'),
]
assert observed_destination_entries == [
DatabaseEntry(id=4, instrument='EIA', provider='xyz'),
DatabaseEntry(id=8, instrument='EIA', provider='xyz'),
]
