def test_get_keys_subset(self): er = ExternalResources('terms') er.add_ref(container='uuid1', key='key1', resource_name='resource1', resource_uri='resource_uri1', entity_id="id11", entity_uri='url11') er.add_ref(container='uuid2', key='key2', resource_name='resource2', resource_uri='resource_uri2', entity_id="id12", entity_uri='url21') er.add_ref(container='uuid1', key=er.get_key(key_name='key1'), resource_name='resource3', resource_uri='resource_uri3', entity_id="id13", entity_uri='url31') key = er.keys.row[0] received = er.get_keys(keys=key) expected = pd.DataFrame( data=[['key1', 0, 'id11', 'url11'], ['key1', 2, 'id13', 'url31']], columns=['key_name', 'resources_idx', 'entity_id', 'entity_uri']) pd.testing.assert_frame_equal(received, expected)
def test_keys_roundtrip(self): er = ExternalResources('terms') keys = pd.DataFrame( data=[['key1', 'resource11', 'resource_id11', 'url11'], ['key1', 'resource12', 'resource_id12', 'url12'], ['key2', 'resource21', 'resource_id21', 'url21']], columns=['key_name', 'resource_name', 'resource_entity_id', 'resource_entity_uri'] ) er.add_keys(keys) received = er.get_keys() pd.testing.assert_frame_equal(received, keys)
def test_get_keys_subset(self): er = ExternalResources('terms') er.add_ref('uuid1', 'field1', 'key1', 'resource11', 'resource_id11', 'url11') er.add_ref('uuid2', 'field2', 'key2', 'resource21', 'resource_id21', 'url21') er.add_ref('uuid1', 'field1', 'key1', 'resource12', 'resource_id12', 'url12') key = er.keys.row[0] received = er.get_keys(keys=key) expected = pd.DataFrame( data=[['key1', 'resource11', 'resource_id11', 'url11'], ['key1', 'resource12', 'resource_id12', 'url12']], columns=['key_name', 'resource_name', 'resource_entity_id', 'resource_entity_uri'] ) pd.testing.assert_frame_equal(received, expected)