def test_bad_good_doc_id():
b = Document()
b.id = 'hello'
b.id = 'abcd' * 4
b.id = 'de09' * 4
b.id = 'af54' * 4
b.id = 'abcdef0123456789'
def test_bad_good_doc_id():
b = Document()
with pytest.raises(BadDocID):
b.id = 'hello'
b.id = 'abcd' * 4
b.id = 'de09' * 4
b.id = 'af54' * 4
b.id = 'abcdef0123456789'
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
doc1 = Document()
doc1.id = '01' * 8
doc1.tags['groundtruth'] = True
doc2 = Document()
doc2.id = '02' * 8
doc2.tags['groundtruth'] = True
doc4 = Document()
doc4.id = '04' * 8
doc4.tags['groundtruth'] = True
self.db = {
int(doc1.id): doc1.SerializeToString(),
int(doc2.id): doc2.SerializeToString(),
int(doc4.id): doc4.SerializeToString()
}
def test_copy_construct():
a = DocumentProto()
b = Document(a, copy=True)
a.id = '1' * 16
assert b.id != '1' * 16
b.id = '2' * 16
assert a.id == '1' * 16
def eval_request():
req = Request()
req.request_type = 'search'
# doc: 1
# doc: 2
# doc: 3
# doc: 4
# doc: 5 - will be missing from KV indexer
for idx in range(5):
dp = Document()
dp.id = f'0{str(idx + 1)}' * 8
req.docs.append(dp)
return req
