Cleaned up unit tests

This commit is contained in:
allegroai 2023-05-25 19:13:10 +03:00
parent c8e4d9eeac
commit 6e777e80b8
26 changed files with 16 additions and 28 deletions

View File

@ -127,7 +127,7 @@ class TestBatchOperations(TestService):
def _temp_task(self):
return self.create_temp(
service="tasks", type="testing", name=self.name, input=dict(view={}),
service="tasks", type="testing", name=self.name,
)
def _temp_task_model(self, task, **kwargs) -> str:

View File

@ -139,6 +139,5 @@ class TestEntityOrdering(TestService):
name="test",
comment=self.test_comment,
type="testing",
input=dict(view=dict()),
**kwargs,
)

View File

@ -109,7 +109,7 @@ class TestModelsService(TestService):
def test_publish_task_no_output_model(self):
task_id = self.create_temp(
service="tasks", type="testing", name="server-test", input=dict(view={})
service="tasks", type="testing", name="server-test"
)
self.api.tasks.started(task=task_id)
self.api.tasks.stopped(task=task_id)
@ -263,7 +263,6 @@ class TestModelsService(TestService):
service="tasks",
type=kwargs.pop("type", "testing"),
name=kwargs.pop("name", "server-test"),
input=kwargs.pop("input", dict(view={})),
**kwargs,
)
@ -277,7 +276,6 @@ class TestModelsService(TestService):
service="tasks",
type="testing",
name="server-test",
input=dict(view={}),
execution=dict(model=execution_model_id),
)
self.api.tasks.started(task=task_id)

View File

@ -45,7 +45,7 @@ class TestMoveUnderProject(TestService):
def _temp_task(self):
task_input = dict(
name=self.entity_name, type="training", input=dict(mapping={}, view=dict(entries=[])),
name=self.entity_name, type="training"
)
return self.create_temp("tasks", **task_input)

View File

@ -75,6 +75,5 @@ class TestPagingAndScrolling(TestService):
name=name,
comment="Test task",
type="testing",
input=dict(view=dict()),
**kwargs,
)

View File

@ -43,7 +43,6 @@ class TestPipelines(TestService):
"tasks",
name=name,
type="testing",
input=dict(view=dict()),
project=project,
system_tags=["pipeline"],
),

View File

@ -10,7 +10,7 @@ class TestProjectsDelete(TestService):
def new_task(self, **kwargs):
return self.create_temp(
"tasks", type="testing", name=db_id(), input=dict(view=dict()), **kwargs
"tasks", type="testing", name=db_id(), **kwargs
)
def new_model(self, **kwargs):

View File

@ -94,7 +94,7 @@ class TestProjectTags(TestService):
def new_task(self, **kwargs):
self.update_missing(
kwargs, type="testing", name="test project tags", input=dict(view=dict())
kwargs, type="testing", name="test project tags"
)
return self.create_temp("tasks", **kwargs)

View File

@ -14,7 +14,6 @@ class TestProjection(TestService):
kwargs,
type="testing",
name="test projection",
input=dict(view=dict()),
delete_params=dict(force=True),
)
return self.create_temp("tasks", **kwargs)

View File

@ -234,7 +234,6 @@ class TestQueues(TestService):
task_input = dict(
name=task_name,
type="testing" if is_testing else "training",
input=dict(mapping={}, view={}),
script={"repository": "test", "entry_point": "test"},
system_tags=["development"] if is_development else None,
)

View File

@ -379,7 +379,6 @@ class TestSubProjects(TestService):
delete_params=self.delete_params,
type=type or "testing",
name=name or db_id(),
input=dict(view=dict()),
client=client,
**kwargs,
)

View File

@ -222,7 +222,7 @@ class TestTags(TestService):
return self.create_temp("models", **kwargs)
def _temp_task(self, **kwargs):
self.update_missing(kwargs, name="Test tags", type="testing", input=dict(view=dict()))
self.update_missing(kwargs, name="Test tags", type="testing")
return self.create_temp("tasks", **kwargs)
def _send(self, service, action, **kwargs):

View File

@ -14,7 +14,6 @@ class TestTasksArtifacts(TestService):
kwargs,
type="testing",
name="test artifacts",
input=dict(view=dict()),
delete_params=dict(force=True),
)
return self.create_temp("tasks", **kwargs)

View File

@ -11,7 +11,7 @@ class TestTaskDebugImages(TestService):
def _temp_task(self, name="test task events"):
task_input = dict(
name=name, type="training", input=dict(mapping={}, view=dict(entries=[])),
name=name, type="training"
)
return self.create_temp("tasks", **task_input)

View File

@ -17,7 +17,7 @@ class TestTaskEvents(TestService):
def _temp_task(self, name="test task events"):
task_input = dict(
name=name, type="training", input=dict(mapping={}, view=dict(entries=[])),
name=name, type="training",
)
return self.create_temp(
"tasks", delete_paramse=self.delete_params, **task_input

View File

@ -24,7 +24,6 @@ class TestTasksHyperparams(TestService):
kwargs,
type="testing",
name="test hyperparams",
input=dict(view=dict()),
delete_params=dict(force=True),
)
return self.create_temp("tasks", **kwargs), kwargs["project"]

View File

@ -104,7 +104,7 @@ class TestTaskModels(TestService):
def new_task(self, **kwargs):
self.update_missing(
kwargs, type="testing", name="test task models", input=dict(view=dict())
kwargs, type="testing", name="test task models"
)
return self.create_temp("tasks", **kwargs)

View File

@ -83,6 +83,6 @@ class TestTaskParent(TestService):
def new_task(self, **kwargs):
self.update_missing(
kwargs, type="testing", name="test task parents", input=dict(view=dict())
kwargs, type="testing", name="test task parents"
)
return self.create_temp("tasks", **kwargs)

View File

@ -8,7 +8,7 @@ from apiserver.tests.automated import TestService
class TestTaskPlots(TestService):
def _temp_task(self, name="test task events"):
task_input = dict(
name=name, type="training", input=dict(mapping={}, view=dict(entries=[])),
name=name, type="training"
)
return self.create_temp("tasks", **task_input)

View File

@ -242,7 +242,7 @@ class TestTasksResetDelete(TestService):
def new_task(self, **kwargs):
self.update_missing(
kwargs, name=self.name, type="testing", input=dict(view=dict())
kwargs, name=self.name, type="testing"
)
return self.create_temp("tasks", delete_params=self.delete_params, **kwargs,)

View File

@ -10,7 +10,7 @@ class TestTasksDiff(TestService):
def new_task(self, **kwargs):
return self.create_temp(
"tasks", name="test", type="testing", input=dict(view=dict()), **kwargs
"tasks", name="test", type="testing", **kwargs
)
def _compare_script(self, task_id, script):

View File

@ -13,7 +13,7 @@ class TestTasksEdit(TestService):
def new_task(self, **kwargs):
self.update_missing(
kwargs, type="testing", name="test", input=dict(view=dict())
kwargs, type="testing", name="test"
)
return self.create_temp("tasks", **kwargs)

View File

@ -96,7 +96,6 @@ class TestTasksFiltering(TestService):
kwargs,
type="testing",
name="test tasks filtering",
input=dict(view=dict()),
delete_params=dict(force=True),
)
return self.create_temp("tasks", **kwargs)

View File

@ -46,7 +46,6 @@ class TestTasksRunning(TestService):
task_input = dict(
name="task-1",
type="testing",
input=dict(mapping={}, view=dict()),
)
if is_development:
task_input["system_tags"] = ["development"]

View File

@ -35,7 +35,7 @@ class TestUsersService(TestService):
task = (
self.api.impersonate(user_3)
.tasks.create(
name="test", type="testing", input=dict(view={}), project=project
name="test", type="testing", project=project
)
.id
)

View File

@ -103,7 +103,7 @@ class TestWorkersService(TestService):
def _create_running_task(self, task_name):
task_input = dict(
name=task_name, type="testing", input=dict(mapping={}, view={})
name=task_name, type="testing"
)
task_id = self.create_temp("tasks", **task_input)