def main() -> int: parser = argparse.ArgumentParser() parser.add_argument('test_name', type=str) parser.add_argument('--clean-up-first', action='store_true', default=False) parser.add_argument('--bless', action='store_true', default=False) if 'PLZ_HOST' in os.environ: plz_host = os.environ['PLZ_HOST'] if 'PLZ_PORT' not in os.environ: raise ValueError('PLZ_HOST is defined but PLZ_PORT is not!') plz_port = int(os.environ['PLZ_PORT']) network = get_network() else: plz_host = 'localhost' plz_port = 5123 network = 'host' test_utils.print_warning( f'Using default controller host and port: {plz_host}:{plz_port} ' '(outside docker)') options = parser.parse_args(sys.argv[1:]) if options.clean_up_first: test_utils.cleanup(interrupted=False) signal.signal(signal.SIGTERM, test_utils.sig_cleanup) with test_utils.DoCleanupContextManager() as cleanup_context: try: success = run_end_to_end_test(network, plz_host, plz_port, options.test_name, options.bless) return 0 if success else 1 except InterruptedError: cleanup_context.interrupted = True raise
def test_multiple_mxnet_integ(tmpdir): tmpdir = str(tmpdir) utils._download_file( tmpdir, "https://s3.amazonaws.com/model-server/inputs/kitten.jpg") model_urls = utils.filtered_urls(mxnet_models, utils.mxnet_model_urls) utils.start_test(tmpdir, model_urls, port='8080', model_type='mxnet') utils.cleanup(tmpdir)
def test_onnx_integ(tmpdir, model_data): tmpdir = str(tmpdir) model_urls = utils.filtered_urls([model_data], utils.onnx_model_urls) utils._download_file( tmpdir, "https://s3.amazonaws.com/model-server/inputs/kitten.jpg") utils.start_test(tmpdir, model_urls, port='8080', model_type='onnx') utils.cleanup(tmpdir)