ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-01-31 09:06:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix test

Browse Source
This commit is contained in:
allegroai 2020-03-20 10:36:20 +02:00
parent 58ab67ea31
commit 4158d08f6f
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/package/test_task_script.py
View File

@ -11,7 +11,7 @@ from contextlib import contextmanager
from typing import Iterator, ContextManager, Sequence, IO, Text from typing import Iterator, ContextManager, Sequence, IO, Text
from uuid import uuid4 from uuid import uuid4
from trains_agent.backend_api.services.tasks import Script from trains_agent.backend_api.services import tasks
from trains_agent.backend_api.session.client import APIClient from trains_agent.backend_api.session.client import APIClient
from pathlib2 import Path from pathlib2 import Path
from pytest import fixture from pytest import fixture
@ -154,7 +154,7 @@ def test_entry_point_warning(client):
""" """
with create_task( with create_task(
client, client,
script=Script(diff="print('hello')", entry_point="foo.py", repository=""), script=tasks.Script(diff="print('hello')", entry_point="foo.py", repository=""),
**DEFAULT_TASK_ARGS **DEFAULT_TASK_ARGS
) as task, iterate_output(SHORT_TIMEOUT, run_task(task)) as output: ) as task, iterate_output(SHORT_TIMEOUT, run_task(task)) as output:
for line in output: for line in output:
@ -172,7 +172,7 @@ def test_run_no_dirs(client):
script = "print('{}')".format(uuid) script = "print('{}')".format(uuid)
with create_task( with create_task(
client, client,
script=Script(diff=script, entry_point="", repository="", working_dir=""), script=tasks.Script(diff=script, entry_point="", repository="", working_dir=""),
**DEFAULT_TASK_ARGS **DEFAULT_TASK_ARGS
) as task, iterate_output(SHORT_TIMEOUT, run_task(task)) as output: ) as task, iterate_output(SHORT_TIMEOUT, run_task(task)) as output:
search_lines( search_lines(
@ -196,7 +196,7 @@ def test_run_working_dir(client):
script = "print('{}')".format(uuid) script = "print('{}')".format(uuid)
with create_task( with create_task(
client, client,
script=Script( script=tasks.Script(
diff=script, diff=script,
entry_point="", entry_point="",
repository="git@bitbucket.org:seematics/roee_test_git.git", repository="git@bitbucket.org:seematics/roee_test_git.git",
@ -223,7 +223,7 @@ def test_regular_task(client):
""" """
with create_task( with create_task(
client, client,
script=Script( script=tasks.Script(
entry_point="noop.py", entry_point="noop.py",
repository="git@bitbucket.org:seematics/roee_test_git.git", repository="git@bitbucket.org:seematics/roee_test_git.git",
), ),
@ -241,7 +241,7 @@ def test_regular_task_nested(client):
""" """
with create_task( with create_task(
client, client,
script=Script( script=tasks.Script(
entry_point="noop_nested.py", entry_point="noop_nested.py",
working_dir="no_reqs", working_dir="no_reqs",
repository="git@bitbucket.org:seematics/roee_test_git.git", repository="git@bitbucket.org:seematics/roee_test_git.git",