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'), ]