merged.

Author: tobiasraabe

CHANGES.rst

@@ -11,6 +11,7 @@ all releases are available on `Anaconda.org
 ------------------
 
 - :gh:`5` fixes the CI and other smaller issues.
+- :gh:`8` aligns pytask-parallel with task priorities in pytask v0.0.11.
 - :gh:`9` enables --max-failures. Closes :gh:`7`.
 
 

src/pytask_parallel/execute.py

@@ -118,6 +118,8 @@ def pytask_execute_build(session):
 
 
 class ProcessesNameSpace:
+    """The name space for hooks related to processes."""
+
     @hookimpl(tryfirst=True)
     def pytask_execute_task(session, task):  # noqa: N805
         """Execute a task.
@@ -142,6 +144,8 @@ def unserialize_and_execute_task(bytes_):
 
 
 class DefaultBackendNameSpace:
+    """The name space for hooks related to threads."""
+
     @hookimpl(tryfirst=True)
     def pytask_execute_task(session, task):  # noqa: N805
         """Execute a task.

tests/test_execute.py

@@ -2,6 +2,7 @@
 import textwrap
 from time import time
 
+import attr
 import pytest
 from pytask import cli
 from pytask import main
@@ -10,9 +11,9 @@
 from pytask_parallel.execute import ProcessesNameSpace
 
 
+@attr.s
 class DummyTask:
-    def __init__(self, function):
-        self.function = function
+    function = attr.ib()
 
     def execute(self):
         self.function()
@@ -188,3 +189,38 @@ def task_1(): pass
 
     assert len(session.tasks) == 3
     assert len(session.execution_reports) == 2
+
+
+@pytest.mark.end_to_end
+@pytest.mark.parametrize("parallel_backend", PARALLEL_BACKENDS)
+def test_task_priorities(tmp_path, parallel_backend):
+    source = """
+    import pytask
+    import time
+
+    @pytask.mark.try_first
+    def task_0():
+        time.sleep(1)
+
+    def task_1():
+        time.sleep(1)
+
+    @pytask.mark.try_last
+    def task_2():
+        time.sleep(1)
+
+    def task_3():
+        time.sleep(1)
+
+    def task_4():
+        time.sleep(1)
+    """
+    tmp_path.joinpath("task_dummy.py").write_text(textwrap.dedent(source))
+
+    session = main(
+        {"paths": tmp_path, "parallel_backend": parallel_backend, "n_workers": 2}
+    )
+
+    assert session.exit_code == 0
+    assert session.execution_reports[0].task.name.endswith("task_0")
+    assert session.execution_reports[-1].task.name.endswith("task_2")