Skip to content

Commit ecf7e09

Browse files
committed
Add test to make sure that priorities work.
1 parent 34573d4 commit ecf7e09

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

src/pytask_parallel/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
1+
"""The entry-point for pytask-parallel."""
12
__version__ = "0.0.4"

tests/test_execute.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -209,11 +209,16 @@ def task_1():
209209
def task_2():
210210
time.sleep(1)
211211
212+
@pytask.mark.try_first
212213
def task_3():
213214
time.sleep(1)
214215
215216
def task_4():
216217
time.sleep(1)
218+
219+
@pytask.mark.try_last
220+
def task_5():
221+
time.sleep(1)
217222
"""
218223
tmp_path.joinpath("task_dummy.py").write_text(textwrap.dedent(source))
219224

@@ -222,5 +227,7 @@ def task_4():
222227
)
223228

224229
assert session.exit_code == 0
225-
assert session.execution_reports[0].task.name.endswith("task_0")
226-
assert session.execution_reports[-1].task.name.endswith("task_2")
230+
first_task_name = session.execution_reports[0].task.name
231+
assert first_task_name.endswith("task_0") or first_task_name.endswith("task_3")
232+
last_task_name = session.execution_reports[-1].task.name
233+
assert last_task_name.endswith("task_2") or last_task_name.endswith("task_5")

0 commit comments

Comments
 (0)