Skip to content

Commit

Permalink
Update test_job_queue test
Browse files Browse the repository at this point in the history
  • Loading branch information
andreas-el committed Sep 21, 2023
1 parent cd2f87d commit 662e95a
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions tests/unit_tests/job_queue/test_job_queue.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from pathlib import Path
from threading import BoundedSemaphore
from typing import Any, Callable, Dict, Optional
from unittest.mock import MagicMock, patch
from unittest.mock import patch

import ert.callbacks
from ert.config import QueueSystem
Expand Down Expand Up @@ -331,10 +331,9 @@ def test_stop_long_running():
job_list[i]._start_time = 0
job_list[i]._end_time = 5

queue = JobQueue(MagicMock())
driver = Driver(driver_type=QueueSystem.LOCAL)
queue = JobQueue(driver)

# We don't need the c layer call here, we only need it added to
# the queue's job_list.
with patch("ert.job_queue.JobQueue._add_job") as _add_job:
for idx, job in enumerate(job_list):
_add_job.return_value = idx
Expand Down

0 comments on commit 662e95a

Please sign in to comment.