def test_update_task_with_models(meta: MetadataRepository, project: Project,
task: Task, model: Model):
task.project = meta.create_project(project)
task = meta.create_task(task)
assert task is not None
id = task.id
model.task = task
model = meta.create_model(model)
task.add_model(model)
task = update_object_fields(task,
excepted_fields=['id', 'models', 'project_id'])
model = update_object_fields(model,
excepted_fields=[
'id', 'wrapper', 'artifact',
'output_meta', 'input_meta',
'requirements', 'transformer', 'task_id'
])
updated_task = meta.update_task(task)
assert id == task.id
assert updated_task is task
assert task == meta.get_task_by_id(task.id)
assert len(task.models) == 1
assert model.id in task.models
assert model == meta.get_model_by_id(model.id)
assert meta.get_model_by_id(model.id).name == 'Test Model2'
assert task.has_meta_repo
def test_push_model_with_task_and_task_argument_with_different_project(ebnt: Ebonite, model: Model):
task = ebnt.get_or_create_task("Project", "Task1")
task2 = ebnt.get_or_create_task("Project2", "Task1")
model.task = task2 # For test purpose
with pytest.raises(ValueError):
ebnt.push_model(model, task)
def test_get_models(meta: MetadataRepository, project: Project, task: Task,
model: Model):
task.project = meta.create_project(project)
created_task = meta.create_task(task)
model.task = created_task
created_model = meta.create_model(model)
actual_models = meta.get_models(created_task)
assert actual_models == [created_model]
def test_update_task_source_is_changed(meta: MetadataRepository,
project: Project, task: Task,
model: Model):
task.project = meta.create_project(project)
saved_task = meta.create_task(task)
assert saved_task is task
id = saved_task.id
model.task = saved_task
model = meta.create_model(model)
saved_task = update_object_fields(
saved_task, excepted_fields=['id', 'models', 'project_id'])
saved_task.add_model(model)
saved_task = meta.update_task(saved_task)
assert id == saved_task.id
assert saved_task == meta.get_task_by_id(saved_task.id)
assert model == saved_task.models.get(model.id)
assert task is saved_task
