def test_fashion(helloworld_args, query_document, mocker): """Regression test for fashion example.""" def validate_response(resp): assert len(resp.search.docs) == 1 for doc in resp.search.docs: assert len(doc.matches) == 10 hello_world(helloworld_args) flow_query_path = os.path.join(resource_filename('jina', 'resources'), 'fashion') mock_on_done = mocker.Mock() mock_on_fail = mocker.Mock() with Flow.load_config( os.path.join(flow_query_path, 'helloworld.flow.query.yml')) as f: f.search( inputs=[query_document], on_done=mock_on_done, on_fail=mock_on_fail, top_k=10, ) mock_on_fail.assert_not_called() validate_callback(mock_on_done, validate_response)
def test_helloworld_py_multimodal(tmpdir): from jina.helloworld.multimodal import hello_world hello_world(set_hw_multimodal_parser().parse_args([ '--workdir', str(tmpdir), '--unblock-query-flow', '--port-expose', str(random_port()) ]))
def hello_world(args: 'Namespace'): """ Run the fashion hello world example :param args: arguments coming from the CLI. """ from jina.helloworld.fashion import hello_world hello_world(args)
def test_helloworld_py_chatbot(tmpdir): from jina.helloworld.chatbot import hello_world hello_world(set_hw_chatbot_parser().parse_args([ '--workdir', str(tmpdir), '--unblock-query-flow', '--port-expose', str(helper.random_port()), ]))
def hello(args: 'Namespace'): if args.hello == 'fashion': from jina.helloworld.fashion import hello_world elif args.hello == 'chatbot': from jina.helloworld.chatbot import hello_world elif args.hello == 'multimodal': from jina.helloworld.multimodal import hello_world else: raise ValueError(f'must be one of [`fashion`, `chatbot`, `multimodal`]') hello_world(args)
def hello(args: 'Namespace'): """ Run any of the hello world examples :param args: arguments coming from the CLI. """ if args.hello == 'fashion': from jina.helloworld.fashion import hello_world elif args.hello == 'chatbot': from jina.helloworld.chatbot import hello_world elif args.hello == 'multimodal': from jina.helloworld.multimodal import hello_world else: raise ValueError( f'must be one of [`fashion`, `chatbot`, `multimodal`]') hello_world(args)
def hello_world(args: 'Namespace'): from jina.helloworld.fashion import hello_world hello_world(args)
def test_helloworld_py(tmpdir): from jina.helloworld.fashion import hello_world hello_world(set_hw_parser().parse_args(['--workdir', str(tmpdir)])) check_hello_world_results(os.path.join(str(tmpdir), 'hello-world.html'))