Bläddra i källkod

minor cleaning

jrobinAV 1 år sedan
förälder
incheckning
b17aac4331
1 ändrade filer med 9 tillägg och 8 borttagningar
  1. 9 8
      tests/core/cycle/test_cycle_manager.py

+ 9 - 8
tests/core/cycle/test_cycle_manager.py

@@ -192,16 +192,17 @@ def test_get_cycle_start_date_and_end_date():
 def test_hard_delete_shared_entities():
     Config.configure_job_executions(mode=JobConfig._DEVELOPMENT_MODE)
 
-    dn_config_1 = Config.configure_data_node("my_input_1", "in_memory", scope=Scope.SCENARIO, default_data="testing")
-    dn_config_2 = Config.configure_data_node("my_input_2", "in_memory", scope=Scope.SCENARIO, default_data="testing")
-    dn_config_3 = Config.configure_data_node("my_input_3", "in_memory", scope=Scope.CYCLE, default_data="testing")
-    dn_config_4 = Config.configure_data_node("my_input_4", "in_memory", scope=Scope.GLOBAL, default_data="testing")
+    dn_config_1 = Config.configure_data_node("my_input_1", "pickle", scope=Scope.SCENARIO, default_data="testing")
+    dn_config_2 = Config.configure_data_node("my_input_2", "pickle", scope=Scope.SCENARIO, default_data="testing")
+    dn_config_3 = Config.configure_data_node("my_input_3", "pickle", scope=Scope.CYCLE, default_data="testing")
+    dn_config_4 = Config.configure_data_node("my_input_4", "pickle", scope=Scope.GLOBAL, default_data="testing")
     task_config_1 = Config.configure_task("task_config_1", print, dn_config_1, dn_config_2)
     task_config_2 = Config.configure_task("task_config_2", print, dn_config_2, dn_config_3)
     task_config_3 = Config.configure_task("task_config_3", print, dn_config_3, dn_config_4)  # scope = global
     creation_date = datetime.now()
+    # Daily frequency so cycle attached to scenarios
     scenario_config_1 = Config.configure_scenario(
-        "scenario_config_1",
+        "scenario_1",
         [task_config_1, task_config_2, task_config_3],
         creation_date=creation_date,
         frequency=Frequency.DAILY,
@@ -213,9 +214,9 @@ def test_hard_delete_shared_entities():
             "sequence_3": [task_config_3],
         }
     )
-    scenario_config_2 = Config.configure_scenario(
-        "scenario_config_2", [task_config_2, task_config_3]
-    )  # No Frequency so cycle attached to scenarios
+
+    # No Frequency so no cycle attached to scenarios
+    scenario_config_2 = Config.configure_scenario("scenario_config_2", [task_config_2, task_config_3])
     scenario_config_2.add_sequences({"sequence_3": [task_config_3]})
 
     _OrchestratorFactory._build_dispatcher()