Skip to content

Commit de569e0

Browse files
committed
fixed tests
1 parent 4b98d4e commit de569e0

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

tests/sap/aibus/dar/client/test_model_manager_client.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -464,9 +464,9 @@ def test_is_job_failed(self):
464464
non_failed_job = self._make_job_resource(non_failed_state)
465465
assert ModelManagerClient.is_job_failed(non_failed_job) is False
466466

467-
def test_read_job_by_model_name(self):
468-
job1 = self._make_job_resource()
469-
job2 = self._make_job_resource()
467+
def test_read_job_by_model_name(self, model_manager_client: ModelManagerClient):
468+
job1 = self._make_job_resource("SUCCEEDED")
469+
job2 = self._make_job_resource("SUCCEEDED")
470470
job2["modelName"] = "my-model-2"
471471
job_collection_response = {"count": 2, "jobs": [job1, job2]}
472472
model_manager_client.read_job_collection = create_autospec(
@@ -477,7 +477,9 @@ def test_read_job_by_model_name(self):
477477
assert response["modelName"] == "my-model-1"
478478
model_manager_client.read_job_collection.assert_called_once()
479479

480-
def test_read_job_by_model_name_job_not_found(self):
480+
def test_read_job_by_model_name_job_not_found(
481+
self, model_manager_client: ModelManagerClient
482+
):
481483
job1 = self._make_job_resource("SUCCEEDED")
482484
job2 = self._make_job_resource("SUCCEEDED")
483485
job2["modelName"] = "my-model-2"

0 commit comments

Comments
 (0)