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