def _query(flow_fn, query_fn):
f = Flow.load_config(flow_fn)
f.use_grpc_gateway()
with f:
with TimeContext(f'QPS: query ', logger=f.logger):
# read image file and load into uri
d = Document(uri=query_fn)
d.convert_uri_to_data_uri()
f.search(inputs=[
d,
], on_done=get_image)
def test_convert_uri_to_data_uri(uri, mimetype):
doc = Document(uri=uri, mime_type=mimetype)
intialiazed_buffer = doc.buffer
intialiazed_uri = doc.uri
doc.convert_uri_to_data_uri()
converted_buffer = doc.buffer
converted_uri = doc.uri
print(doc.content_type)
assert doc.uri.startswith(f'data:{mimetype}')
assert intialiazed_uri != converted_uri
assert converted_buffer != intialiazed_buffer
assert doc.mime_type == mimetype
def test_datauri_to_blob():
doc = Document(uri=os.path.join(cur_dir, 'test.png'))
doc.convert_uri_to_data_uri()
doc.convert_data_uri_to_blob()
assert isinstance(doc.blob, np.ndarray)
assert doc.blob.shape == (85, 152, 3) # h,w,c
