def test_get_attr_values():
d = Document({
'id': '123',
'text': 'document',
'feature1': 121,
'name': 'name',
'tags': {
'id': 'identity',
'a': 'b',
'c': 'd'
},
})
d.score = NamedScore(value=42)
required_keys = [
'id',
'text',
'tags__name',
'tags__feature1',
'score__value',
'tags__c',
'tags__id',
'tags__inexistant',
'inexistant',
]
res = d.get_attributes(*required_keys)
assert len(res) == len(required_keys)
assert res[required_keys.index('id')] == '123'
assert res[required_keys.index('tags__feature1')] == 121
assert res[required_keys.index('tags__name')] == 'name'
assert res[required_keys.index('text')] == 'document'
assert res[required_keys.index('tags__c')] == 'd'
assert res[required_keys.index('tags__id')] == 'identity'
assert res[required_keys.index('score__value')] == 42
assert res[required_keys.index('tags__inexistant')] is None
assert res[required_keys.index('inexistant')] is None
required_keys_2 = ['tags', 'text']
res2 = d.get_attributes(*required_keys_2)
assert len(res2) == 2
assert res2[required_keys_2.index('text')] == 'document'
assert res2[required_keys_2.index('tags')] == d.tags
d = Document({
'id': '123',
'tags': {
'outterkey': {
'innerkey': 'real_value'
}
}
})
required_keys_3 = ['tags__outterkey__innerkey']
res3 = d.get_attributes(*required_keys_3)
assert res3 == 'real_value'
d = Document(content=np.array([1, 2, 3]))
res4 = np.stack(d.get_attributes(*['blob']))
np.testing.assert_equal(res4, np.array([1, 2, 3]))
