def test_pipeline_adhoc_with_yaml_path_run_success(runner, logged_in_and_linked): add_valid_pipeline_yaml(yaml_path='bark.yaml') args = ['--adhoc', '--yaml=bark.yaml', 'training'] with RunAPIMock(PROJECT_DATA['id']): output = runner.invoke(run, args).output assert 'Success' in output assert 'Uploaded ad-hoc code' in output
def test_run_no_git(runner, logged_in_and_linked): project_id = PROJECT_DATA['id'] with open(get_project().get_config_filename(), 'w') as yaml_fp: yaml_fp.write(CONFIG_YAML) args = ['train'] with RunAPIMock(project_id, 'f' * 16, {}): output = runner.invoke(run, args, catch_exceptions=False).output assert 'is not a Git repository' in output
def test_remote_project(request, runner, monkeypatch, tmpdir): project_id = PROJECT_DATA['id'] request.addfinalizer(lambda: settings.reset()) monkeypatch.chdir(tmpdir) with RunAPIMock(project_id, 'f' * 40, {}) as mock: configure_token_login(None, test_token) configure_project_override(project_id=project_id, mode=None) assert isinstance(settings.override_project, RemoteProject) # Regular execution run_output = runner.invoke(run, ['train'], catch_exceptions=False).output assert 'created' in run_output assert 'Using remote project' in run_output # Ad-hocs don't work assert 'can not be used' in runner.invoke( run, ['train', '--adhoc'], catch_exceptions=False).output
def test_pipeline_run_no_name(runner, logged_in_and_linked): add_valid_pipeline_yaml() args = [''] with RunAPIMock(PROJECT_DATA['id']): output = runner.invoke(run, args).output assert 'Usage: ' in output
def test_pipeline_run_success(runner, logged_in_and_linked): add_valid_pipeline_yaml() args = ['training'] with RunAPIMock(PROJECT_DATA['id']): output = runner.invoke(run, args).output assert 'Success' in output
def default_run_api_mock(): with RunAPIMock(PROJECT_DATA['id'], 'f' * 16, {}) as am: yield am