use process node for test

aiidateam · Dec 13, 2024 · 936548d · 936548d
1 parent 18397b9
commit 936548d
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 9 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -91,7 +91,6 @@ workgraph = "aiida_workgraph.cli.cmd_workgraph:workgraph"
 "workgraph.while" = "aiida_workgraph.tasks.builtins:While"
 "workgraph.if" = "aiida_workgraph.tasks.builtins:If"
 "workgraph.select" = "aiida_workgraph.tasks.builtins:Select"
-"workgraph.gather" = "aiida_workgraph.tasks.builtins:Gather"
 "workgraph.set_context" = "aiida_workgraph.tasks.builtins:SetContext"
 "workgraph.get_context" = "aiida_workgraph.tasks.builtins:GetContext"
 "workgraph.time_monitor" = "aiida_workgraph.tasks.monitors:TimeMonitor"

diff --git a/src/aiida_workgraph/engine/workgraph.py b/src/aiida_workgraph/engine/workgraph.py
@@ -301,7 +301,6 @@ def setup_ctx_workgraph(self, wgdata: t.Dict[str, t.Any]) -> None:
         self.ctx._tasks = wgdata["tasks"]
         self.ctx._links = wgdata["links"]
         self.ctx._connectivity = wgdata["connectivity"]
-        self.ctx._ctrl_links = wgdata["ctrl_links"]
         self.ctx._workgraph = wgdata
         self.ctx._error_handlers = wgdata["error_handlers"]
 

diff --git a/src/aiida_workgraph/utils/analysis.py b/src/aiida_workgraph/utils/analysis.py
@@ -22,8 +22,13 @@ def __init__(
         """
         self.process = process
         self.restart_process = restart_process
+        wgdata.setdefault("uuid", "")
+        wgdata.setdefault("tasks", {})
+        wgdata.setdefault("links", [])
+        wgdata.setdefault("ctrl_links", [])
+        wgdata.setdefault("error_handlers", {})
+        wgdata.setdefault("context", {})
         self.wgdata = wgdata
-        self.uuid = wgdata["uuid"]
         self.name = wgdata["name"]
         self.wait_to_link()
         self.clean_hanging_links()

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -270,13 +270,7 @@ def process_node(state="finished", exit_status=0):
             inputs={
                 "wg": {
                     "name": "test",
-                    "uuid": "",
-                    "tasks": {},
-                    "links": [],
-                    "ctrl_links": [],
                     "state": "",
-                    "error_handlers": {},
-                    "context": {},
                 }
             }
         )