update test for task from workgraph to check the pickled function

tests/test_tasks.py

@@ -44,20 +44,26 @@ def test_task_collection(decorated_add: Callable) -> None:
 
 
 @pytest.mark.usefixtures("started_daemon_client")
-def test_build_task_from_workgraph(
-    wg_calcfunction: Callable, decorated_add: Callable
-) -> None:
-
+def test_build_task_from_workgraph(decorated_add: Callable) -> None:
+    # create a sub workgraph
+    sub_wg = WorkGraph("build_task_from_workgraph")
+    sub_wg.add_task(decorated_add, name="add1", x=1, y=3)
+    sub_wg.add_task(
+        decorated_add, name="add2", x=2, y=sub_wg.tasks["add1"].outputs["result"]
+    )
+    #
     wg = WorkGraph("build_task_from_workgraph")
     add1_task = wg.add_task(decorated_add, name="add1", x=1, y=3)
-    wg_task = wg.add_task(wg_calcfunction, name="wg_calcfunction")
-    assert wg_task.inputs["sumdiff1"].value is None
+    wg_task = wg.add_task(sub_wg, name="sub_wg")
+    # the default value of the namespace is None
+    assert wg_task.inputs["add1"].value is None
     wg.add_task(decorated_add, name="add2", y=3)
-    wg.add_link(add1_task.outputs["result"], wg_task.inputs["sumdiff1.x"])
-    wg.add_link(wg_task.outputs["sumdiff2.sum"], wg.tasks["add2"].inputs["x"])
-    assert len(wg_task.inputs) == 7
-    assert len(wg_task.outputs) == 8
+    wg.add_link(add1_task.outputs["result"], wg_task.inputs["add1.x"])
+    wg.add_link(wg_task.outputs["add2.result"], wg.tasks["add2"].inputs["x"])
+    assert len(wg_task.inputs) == 21
+    assert len(wg_task.outputs) == 6
     wg.submit(wait=True)
+    # wg.run()
     assert wg.tasks["add2"].outputs["result"].value.value == 14