def test_helloworld_flow(tmpdir):
args = set_hw_parser().parse_args([])
os.environ['RESOURCE_DIR'] = resource_filename('jina', 'resources')
os.environ['SHARDS'] = str(args.shards)
os.environ['PARALLEL'] = str(args.parallel)
os.environ['HW_WORKDIR'] = str(tmpdir)
f = Flow.load_config(resource_filename('jina', '/'.join(('resources', 'helloworld.flow.index.yml'))))
targets = {
'index': {
'url': args.index_data_url,
'filename': os.path.join(tmpdir, 'index-original')
},
'query': {
'url': args.query_data_url,
'filename': os.path.join(tmpdir, 'query-original')
}
}
# download the data
Path(tmpdir).mkdir(parents=True, exist_ok=True)
download_data(targets)
# run it!
with f:
f.index(_input_ndarray(targets['index']['data']), request_size=args.index_request_size)
def test_input_numpy(array):
Client.check_input(_input_ndarray(array))