Fix typing in lockfile state tests

Author: tobiasraabe

tests/test_database.py

@@ -5,6 +5,8 @@
 from _pytask.lockfile import build_portable_node_id
 from _pytask.lockfile import build_portable_task_id
 from _pytask.lockfile import read_lockfile
+from _pytask.node_protocols import PNode
+from _pytask.tree_util import tree_leaves
 from pytask import ExitCode
 from pytask import build
 from pytask import cli
@@ -40,8 +42,14 @@ def task_write(path=Path("in.txt"), produces=Path("out.txt")):
     produces = task.produces
     assert depends_on is not None
     assert produces is not None
-    in_node = depends_on["path"]  # type: ignore[union-attr]
-    out_node = produces["produces"]  # type: ignore[union-attr]
+    in_nodes = tree_leaves(depends_on["path"])
+    out_nodes = tree_leaves(produces["produces"])
+    assert len(in_nodes) == 1
+    assert len(out_nodes) == 1
+    in_node = in_nodes[0]
+    out_node = out_nodes[0]
+    assert isinstance(in_node, PNode)
+    assert isinstance(out_node, PNode)
 
     in_id = build_portable_node_id(in_node, tmp_path)
     out_id = build_portable_node_id(out_node, tmp_path)

tests/test_persist.py

@@ -8,7 +8,9 @@
 from _pytask.lockfile import build_portable_node_id
 from _pytask.lockfile import build_portable_task_id
 from _pytask.lockfile import read_lockfile
+from _pytask.node_protocols import PNode
 from _pytask.persist import pytask_execute_task_process_report
+from _pytask.tree_util import tree_leaves
 from pytask import ExitCode
 from pytask import Persisted
 from pytask import SkippedUnchanged
@@ -69,7 +71,10 @@ def task_dummy(path=Path("in.txt"), produces=Path("out.txt")):
     entry = tasks_by_id[build_portable_task_id(task, tmp_path)]
     produces = task.produces
     assert produces is not None
-    node = produces["produces"]  # type: ignore[union-attr]
+    nodes = tree_leaves(produces["produces"])
+    assert len(nodes) == 1
+    node = nodes[0]
+    assert isinstance(node, PNode)
     node_id = build_portable_node_id(node, tmp_path)
     assert entry.produces[node_id] == node.state()