Explorar o código

fixed rest conftest

Toan Quach hai 1 ano
pai
achega
cc93b309b9
Modificáronse 1 ficheiros con 10 adicións e 10 borrados
  1. 10 10
      tests/rest/conftest.py

+ 10 - 10
tests/rest/conftest.py

@@ -23,7 +23,7 @@ from taipy.config.common.frequency import Frequency
 from taipy.config.common.scope import Scope
 from taipy.config.common.scope import Scope
 from taipy.core import Cycle, DataNodeId, Job, JobId, Scenario, Sequence, Task
 from taipy.core import Cycle, DataNodeId, Job, JobId, Scenario, Sequence, Task
 from taipy.core.cycle._cycle_manager import _CycleManager
 from taipy.core.cycle._cycle_manager import _CycleManager
-from taipy.core.data.in_memory import InMemoryDataNode
+from taipy.core.data.pickle import PickleDataNode
 from taipy.core.job._job_manager import _JobManager
 from taipy.core.job._job_manager import _JobManager
 from taipy.core.task._task_manager import _TaskManager
 from taipy.core.task._task_manager import _TaskManager
 from taipy.rest.app import create_app
 from taipy.rest.app import create_app
@@ -122,24 +122,24 @@ def scenario_data():
 
 
 @pytest.fixture
 @pytest.fixture
 def default_datanode():
 def default_datanode():
-    return InMemoryDataNode(
+    return PickleDataNode(
         "input_ds",
         "input_ds",
         Scope.SCENARIO,
         Scope.SCENARIO,
         DataNodeId("f"),
         DataNodeId("f"),
-        "my name",
         "owner_id",
         "owner_id",
+        None,
         properties={"default_data": [1, 2, 3, 4, 5, 6]},
         properties={"default_data": [1, 2, 3, 4, 5, 6]},
     )
     )
 
 
 
 
 @pytest.fixture
 @pytest.fixture
 def default_df_datanode():
 def default_df_datanode():
-    return InMemoryDataNode(
+    return PickleDataNode(
         "input_ds",
         "input_ds",
         Scope.SCENARIO,
         Scope.SCENARIO,
         DataNodeId("id_uio2"),
         DataNodeId("id_uio2"),
-        "my name",
         "owner_id",
         "owner_id",
+        None,
         properties={"default_data": pd.DataFrame([{"a": 1, "b": 2}, {"a": 3, "b": 4}, {"a": 5, "b": 6}])},
         properties={"default_data": pd.DataFrame([{"a": 1, "b": 2}, {"a": 3, "b": 4}, {"a": 5, "b": 6}])},
     )
     )
 
 
@@ -158,21 +158,21 @@ def default_datanode_config_list():
 
 
 
 
 def __default_task():
 def __default_task():
-    input_ds = InMemoryDataNode(
+    input_ds = PickleDataNode(
         "input_ds",
         "input_ds",
         Scope.SCENARIO,
         Scope.SCENARIO,
         DataNodeId("id_uio"),
         DataNodeId("id_uio"),
-        "my name",
         "owner_id",
         "owner_id",
+        {"TASK_task_id"},
         properties={"default_data": "In memory Data Source"},
         properties={"default_data": "In memory Data Source"},
     )
     )
 
 
-    output_ds = InMemoryDataNode(
+    output_ds = PickleDataNode(
         "output_ds",
         "output_ds",
         Scope.SCENARIO,
         Scope.SCENARIO,
         DataNodeId("id_uio"),
         DataNodeId("id_uio"),
-        "my name",
         "owner_id",
         "owner_id",
+        {"TASK_task_id"},
         properties={"default_data": "In memory Data Source"},
         properties={"default_data": "In memory Data Source"},
     )
     )
     return Task(
     return Task(
@@ -181,7 +181,7 @@ def __default_task():
         function=print,
         function=print,
         input=[input_ds],
         input=[input_ds],
         output=[output_ds],
         output=[output_ds],
-        id=None,
+        id="TASK_task_id",
     )
     )