diff --git a/package/tests/test_launchers/test_cli.py b/package/tests/test_launchers/test_cli.py index d30e651bba..82403cd5e7 100755 --- a/package/tests/test_launchers/test_cli.py +++ b/package/tests/test_launchers/test_cli.py @@ -162,6 +162,38 @@ def mock_project_path(mocker): "extra_params": {"extra_param": "param"}, }, ), + ( + [ + "viz", + "run", + "--host", + "8.8.8.8", + "--port", + "4142", + "--no-browser", + "--save-file", + "save_dir", + "-p", + "data_science", + "-e", + "local", + "--params", + "extra_param=param", + ], + { + "host": "8.8.8.8", + "port": 4142, + "load_file": None, + "save_file": "save_dir", + "pipeline_name": "data_science", + "env": "local", + "project_path": "testPath", + "autoreload": False, + "include_hooks": False, + "package_name": None, + "extra_params": {"extra_param": "param"}, + }, + ), ( ["viz", "run", "--include-hooks"], {