test_scenario.py 68 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522
  1. # Copyright 2021-2024 Avaiga Private Limited
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
  4. # the License. You may obtain a copy of the License at
  5. #
  6. # http://www.apache.org/licenses/LICENSE-2.0
  7. #
  8. # Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
  9. # an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
  10. # specific language governing permissions and limitations under the License.
  11. from datetime import datetime, timedelta
  12. from unittest import mock
  13. import pytest
  14. from taipy.config import Frequency
  15. from taipy.config.common.scope import Scope
  16. from taipy.config.exceptions.exceptions import InvalidConfigurationId
  17. from taipy.core.common._utils import _Subscriber
  18. from taipy.core.cycle._cycle_manager_factory import _CycleManagerFactory
  19. from taipy.core.cycle.cycle import Cycle, CycleId
  20. from taipy.core.data._data_manager_factory import _DataManagerFactory
  21. from taipy.core.data.in_memory import DataNode, InMemoryDataNode
  22. from taipy.core.data.pickle import PickleDataNode
  23. from taipy.core.exceptions.exceptions import SequenceAlreadyExists, SequenceTaskDoesNotExistInScenario
  24. from taipy.core.scenario._scenario_manager_factory import _ScenarioManagerFactory
  25. from taipy.core.scenario.scenario import Scenario
  26. from taipy.core.scenario.scenario_id import ScenarioId
  27. from taipy.core.sequence.sequence import Sequence
  28. from taipy.core.sequence.sequence_id import SequenceId
  29. from taipy.core.task._task_manager_factory import _TaskManagerFactory
  30. from taipy.core.task.task import Task, TaskId
  31. def test_scenario_equals(scenario):
  32. scenario_manager = _ScenarioManagerFactory()._build_manager()
  33. scenario_id = scenario.id
  34. scenario_manager._set(scenario)
  35. # To test if instance is same type
  36. task = Task("task", {}, print, [], [], scenario_id)
  37. scenario_2 = scenario_manager._get(scenario_id)
  38. assert scenario == scenario_2
  39. assert scenario != scenario_id
  40. assert scenario != task
  41. def test_create_primary_scenario(cycle):
  42. scenario = Scenario("foo", set(), {"key": "value"}, is_primary=True, cycle=cycle)
  43. assert scenario.id is not None
  44. assert scenario.config_id == "foo"
  45. assert scenario.tasks == {}
  46. assert scenario.additional_data_nodes == {}
  47. assert scenario.data_nodes == {}
  48. assert scenario.sequences == {}
  49. assert scenario.properties == {"key": "value"}
  50. assert scenario.key == "value"
  51. assert scenario.creation_date is not None
  52. assert scenario.is_primary
  53. assert scenario.cycle == cycle
  54. assert scenario.tags == set()
  55. assert scenario.get_simple_label() == scenario.config_id
  56. with mock.patch("taipy.core.get") as get_mck:
  57. class MockOwner:
  58. label = "owner_label"
  59. def get_label(self):
  60. return self.label
  61. get_mck.return_value = MockOwner()
  62. assert scenario.get_label() == "owner_label > " + scenario.config_id
  63. def test_create_scenario_at_time(current_datetime):
  64. scenario = Scenario("bar", set(), {}, set(), ScenarioId("baz"), creation_date=current_datetime)
  65. assert scenario.id == "baz"
  66. assert scenario.config_id == "bar"
  67. assert scenario.tasks == {}
  68. assert scenario.additional_data_nodes == {}
  69. assert scenario.data_nodes == {}
  70. assert scenario.sequences == {}
  71. assert scenario.properties == {}
  72. assert scenario.creation_date == current_datetime
  73. assert not scenario.is_primary
  74. assert scenario.cycle is None
  75. assert scenario.tags == set()
  76. assert scenario.get_simple_label() == scenario.config_id
  77. assert scenario.get_label() == scenario.config_id
  78. def test_create_scenario_with_task_and_additional_dn_and_sequence():
  79. dn_1 = PickleDataNode("xyz", Scope.SCENARIO)
  80. dn_2 = PickleDataNode("abc", Scope.SCENARIO)
  81. task = Task("qux", {}, print, [dn_1])
  82. scenario = Scenario("quux", {task}, {}, {dn_2}, sequences={"acb": {"tasks": [task]}})
  83. sequence = scenario.sequences["acb"]
  84. assert scenario.id is not None
  85. assert scenario.config_id == "quux"
  86. assert len(scenario.tasks) == 1
  87. assert len(scenario.additional_data_nodes) == 1
  88. assert len(scenario.data_nodes) == 2
  89. assert len(scenario.sequences) == 1
  90. assert scenario.qux == task
  91. assert scenario.xyz == dn_1
  92. assert scenario.abc == dn_2
  93. assert scenario.acb == sequence
  94. assert scenario.properties == {}
  95. assert scenario.tags == set()
  96. def test_create_scenario_invalid_config_id():
  97. with pytest.raises(InvalidConfigurationId):
  98. Scenario("foo bar", set(), {})
  99. def test_create_scenario_and_add_sequences():
  100. input_1 = PickleDataNode("input_1", Scope.SCENARIO)
  101. output_1 = PickleDataNode("output_1", Scope.SCENARIO)
  102. output_2 = PickleDataNode("output_2", Scope.SCENARIO)
  103. additional_dn_1 = PickleDataNode("additional_1", Scope.SCENARIO)
  104. additional_dn_2 = PickleDataNode("additional_2", Scope.SCENARIO)
  105. task_1 = Task("task_1", {}, print, [input_1], [output_1], TaskId("task_id_1"))
  106. task_2 = Task("task_2", {}, print, [output_1], [output_2], TaskId("task_id_2"))
  107. data_manager = _DataManagerFactory._build_manager()
  108. task_manager = _TaskManagerFactory._build_manager()
  109. data_manager._set(input_1)
  110. data_manager._set(output_1)
  111. data_manager._set(output_2)
  112. data_manager._set(additional_dn_1)
  113. data_manager._set(additional_dn_2)
  114. task_manager._set(task_1)
  115. task_manager._set(task_2)
  116. scenario = Scenario("scenario", {task_1}, {})
  117. scenario.sequences = {"sequence_1": {"tasks": [task_1]}, "sequence_2": {"tasks": []}}
  118. assert scenario.id is not None
  119. assert scenario.config_id == "scenario"
  120. assert len(scenario.tasks) == 1
  121. assert scenario.tasks.keys() == {task_1.config_id}
  122. assert len(scenario.additional_data_nodes) == 0
  123. assert scenario.additional_data_nodes == {}
  124. assert len(scenario.data_nodes) == 2
  125. assert scenario.data_nodes == {
  126. input_1.config_id: input_1,
  127. output_1.config_id: output_1,
  128. }
  129. assert len(scenario.sequences) == 2
  130. assert scenario.sequence_1 == scenario.sequences["sequence_1"]
  131. assert scenario.sequence_2 == scenario.sequences["sequence_2"]
  132. assert scenario.sequences == {"sequence_1": scenario.sequence_1, "sequence_2": scenario.sequence_2}
  133. def test_create_scenario_overlapping_sequences():
  134. input_1 = PickleDataNode("input_1", Scope.SCENARIO)
  135. output_1 = PickleDataNode("output_1", Scope.SCENARIO)
  136. output_2 = PickleDataNode("output_2", Scope.SCENARIO)
  137. additional_dn_1 = PickleDataNode("additional_1", Scope.SCENARIO)
  138. additional_dn_2 = PickleDataNode("additional_2", Scope.SCENARIO)
  139. task_1 = Task("task_1", {}, print, [input_1], [output_1], TaskId("task_id_1"))
  140. task_2 = Task("task_2", {}, print, [output_1], [output_2], TaskId("task_id_2"))
  141. data_manager = _DataManagerFactory._build_manager()
  142. task_manager = _TaskManagerFactory._build_manager()
  143. data_manager._set(input_1)
  144. data_manager._set(output_1)
  145. data_manager._set(output_2)
  146. data_manager._set(additional_dn_1)
  147. data_manager._set(additional_dn_2)
  148. task_manager._set(task_1)
  149. task_manager._set(task_2)
  150. scenario = Scenario("scenario", {task_1, task_2}, {})
  151. scenario.add_sequence("sequence_1", [task_1])
  152. scenario.add_sequence("sequence_2", [task_1, task_2])
  153. assert scenario.id is not None
  154. assert scenario.config_id == "scenario"
  155. assert len(scenario.tasks) == 2
  156. assert scenario.tasks.keys() == {task_1.config_id, task_2.config_id}
  157. assert len(scenario.additional_data_nodes) == 0
  158. assert scenario.additional_data_nodes == {}
  159. assert len(scenario.data_nodes) == 3
  160. assert scenario.data_nodes == {
  161. input_1.config_id: input_1,
  162. output_1.config_id: output_1,
  163. output_2.config_id: output_2,
  164. }
  165. sequence_1 = scenario.sequences["sequence_1"]
  166. sequence_2 = scenario.sequences["sequence_2"]
  167. assert scenario.sequences == {"sequence_1": sequence_1, "sequence_2": sequence_2}
  168. scenario.remove_sequences(["sequence_2"])
  169. assert scenario.sequences == {"sequence_1": sequence_1}
  170. scenario.remove_sequences(["sequence_1"])
  171. assert scenario.sequences == {}
  172. def test_create_scenario_one_additional_dn():
  173. input_1 = PickleDataNode("input_1", Scope.SCENARIO)
  174. input_2 = PickleDataNode("input_2", Scope.SCENARIO)
  175. output_1 = PickleDataNode("output_1", Scope.SCENARIO)
  176. output_2 = PickleDataNode("output_2", Scope.SCENARIO)
  177. additional_dn_1 = PickleDataNode("additional_1", Scope.SCENARIO)
  178. additional_dn_2 = PickleDataNode("additional_2", Scope.SCENARIO)
  179. task_1 = Task("task_1", {}, print, [input_1], [output_1], TaskId("task_id_1"))
  180. task_2 = Task("task_2", {}, print, [input_2], [output_2], TaskId("task_id_2"))
  181. data_manager = _DataManagerFactory._build_manager()
  182. task_manager = _TaskManagerFactory._build_manager()
  183. data_manager._set(input_1)
  184. data_manager._set(output_1)
  185. data_manager._set(input_2)
  186. data_manager._set(output_2)
  187. data_manager._set(additional_dn_1)
  188. data_manager._set(additional_dn_2)
  189. task_manager._set(task_1)
  190. task_manager._set(task_2)
  191. scenario = Scenario("scenario", set(), {}, {additional_dn_1})
  192. assert scenario.id is not None
  193. assert scenario.config_id == "scenario"
  194. assert len(scenario.tasks) == 0
  195. assert len(scenario.additional_data_nodes) == 1
  196. assert len(scenario.data_nodes) == 1
  197. assert scenario.tasks == {}
  198. assert scenario.additional_data_nodes == {additional_dn_1.config_id: additional_dn_1}
  199. assert scenario.data_nodes == {additional_dn_1.config_id: additional_dn_1}
  200. def test_create_scenario_wth_additional_dns():
  201. input_1 = PickleDataNode("input_1", Scope.SCENARIO)
  202. input_2 = PickleDataNode("input_2", Scope.SCENARIO)
  203. output_1 = PickleDataNode("output_1", Scope.SCENARIO)
  204. output_2 = PickleDataNode("output_2", Scope.SCENARIO)
  205. additional_dn_1 = PickleDataNode("additional_1", Scope.SCENARIO)
  206. additional_dn_2 = PickleDataNode("additional_2", Scope.SCENARIO)
  207. task_1 = Task("task_1", {}, print, [input_1], [output_1], TaskId("task_id_1"))
  208. task_2 = Task("task_2", {}, print, [input_2], [output_2], TaskId("task_id_2"))
  209. data_manager = _DataManagerFactory._build_manager()
  210. task_manager = _TaskManagerFactory._build_manager()
  211. data_manager._set(input_1)
  212. data_manager._set(output_1)
  213. data_manager._set(input_2)
  214. data_manager._set(output_2)
  215. data_manager._set(additional_dn_1)
  216. data_manager._set(additional_dn_2)
  217. task_manager._set(task_1)
  218. task_manager._set(task_2)
  219. scenario = Scenario("scenario", set(), {}, {additional_dn_1, additional_dn_2})
  220. assert scenario.id is not None
  221. assert scenario.config_id == "scenario"
  222. assert len(scenario.tasks) == 0
  223. assert len(scenario.additional_data_nodes) == 2
  224. assert len(scenario.data_nodes) == 2
  225. assert scenario.tasks == {}
  226. assert scenario.additional_data_nodes == {
  227. additional_dn_1.config_id: additional_dn_1,
  228. additional_dn_2.config_id: additional_dn_2,
  229. }
  230. assert scenario.data_nodes == {
  231. additional_dn_1.config_id: additional_dn_1,
  232. additional_dn_2.config_id: additional_dn_2,
  233. }
  234. scenario_1 = Scenario("scenario_1", {task_1}, {}, {additional_dn_1})
  235. assert scenario_1.id is not None
  236. assert scenario_1.config_id == "scenario_1"
  237. assert len(scenario_1.tasks) == 1
  238. assert len(scenario_1.additional_data_nodes) == 1
  239. assert len(scenario_1.data_nodes) == 3
  240. assert scenario_1.tasks.keys() == {task_1.config_id}
  241. assert scenario_1.additional_data_nodes == {
  242. additional_dn_1.config_id: additional_dn_1,
  243. }
  244. assert scenario_1.data_nodes == {
  245. input_1.config_id: input_1,
  246. output_1.config_id: output_1,
  247. additional_dn_1.config_id: additional_dn_1,
  248. }
  249. scenario_2 = Scenario("scenario_2", {task_1, task_2}, {}, {additional_dn_1, additional_dn_2})
  250. assert scenario_2.id is not None
  251. assert scenario_2.config_id == "scenario_2"
  252. assert len(scenario_2.tasks) == 2
  253. assert len(scenario_2.additional_data_nodes) == 2
  254. assert len(scenario_2.data_nodes) == 6
  255. assert scenario_2.tasks.keys() == {task_1.config_id, task_2.config_id}
  256. assert scenario_2.additional_data_nodes == {
  257. additional_dn_1.config_id: additional_dn_1,
  258. additional_dn_2.config_id: additional_dn_2,
  259. }
  260. assert {dn_config_id: dn.id for dn_config_id, dn in scenario_2.data_nodes.items()} == {
  261. input_1.config_id: input_1.id,
  262. output_1.config_id: output_1.id,
  263. input_2.config_id: input_2.id,
  264. output_2.config_id: output_2.id,
  265. additional_dn_1.config_id: additional_dn_1.id,
  266. additional_dn_2.config_id: additional_dn_2.id,
  267. }
  268. def test_raise_sequence_tasks_not_in_scenario(data_node):
  269. task_1 = Task("task_1", {}, print, output=[data_node])
  270. task_2 = Task("task_2", {}, print, input=[data_node])
  271. with pytest.raises(SequenceTaskDoesNotExistInScenario) as err:
  272. Scenario("scenario", set(), {}, sequences={"sequence": {"tasks": [task_1]}}, scenario_id="SCENARIO_scenario")
  273. assert err.value.args == ([task_1.id], "sequence", "SCENARIO_scenario")
  274. with pytest.raises(SequenceTaskDoesNotExistInScenario) as err:
  275. Scenario(
  276. "scenario",
  277. [task_1],
  278. {},
  279. sequences={"sequence": {"tasks": [task_1, task_2]}},
  280. scenario_id="SCENARIO_scenario",
  281. )
  282. assert err.value.args == ([task_2.id], "sequence", "SCENARIO_scenario")
  283. Scenario("scenario", {task_1}, {}, sequences={"sequence": {"tasks": [task_1]}})
  284. Scenario(
  285. "scenario",
  286. [task_1, task_2],
  287. {},
  288. sequences={"sequence_1": {"tasks": [task_1]}, "sequence_2": {"tasks": [task_1, task_2]}},
  289. )
  290. def test_adding_sequence_raises_tasks_not_in_scenario(data_node):
  291. task_1 = Task("task_1", {}, print, output=[data_node])
  292. task_2 = Task("task_2", {}, print, input=[data_node])
  293. scenario = Scenario("scenario", [task_1], {})
  294. scenario_manager = _ScenarioManagerFactory._build_manager()
  295. task_manager = _TaskManagerFactory._build_manager()
  296. scenario_manager._set(scenario)
  297. task_manager._set(task_1)
  298. task_manager._set(task_2)
  299. scenario.add_sequences({"sequence_1": {}})
  300. with pytest.raises(SequenceTaskDoesNotExistInScenario) as err:
  301. scenario.add_sequence("sequence_2", [task_2])
  302. assert err.value.args == ([task_2.id], "sequence_2", scenario.id)
  303. scenario.add_sequence("sequence_3", [task_1])
  304. with pytest.raises(SequenceTaskDoesNotExistInScenario) as err:
  305. scenario.add_sequences({"sequence_4": [task_2]})
  306. assert err.value.args == ([task_2.id], "sequence_4", scenario.id)
  307. with pytest.raises(SequenceTaskDoesNotExistInScenario) as err:
  308. scenario.add_sequences({"sequence_5": [task_1, task_2]})
  309. assert err.value.args == ([task_2.id], "sequence_5", scenario.id)
  310. scenario.tasks = [task_1, task_2]
  311. scenario.add_sequence("sequence_6", [task_1, task_2])
  312. def test_adding_existing_sequence_raises_exception(data_node):
  313. task_1 = Task("task_1", {}, print, output=[data_node])
  314. _TaskManagerFactory._build_manager()._set(task_1)
  315. task_2 = Task("task_2", {}, print, input=[data_node])
  316. _TaskManagerFactory._build_manager()._set(task_2)
  317. scenario = Scenario("scenario", tasks={task_1, task_2}, properties={})
  318. _ScenarioManagerFactory._build_manager()._set(scenario)
  319. scenario.add_sequence("sequence_1", [task_1])
  320. with pytest.raises(SequenceAlreadyExists):
  321. scenario.add_sequence("sequence_1", [task_2])
  322. def test_renaming_existing_sequence_raises_exception(data_node):
  323. task_1 = Task("task_1", {}, print, output=[data_node])
  324. _TaskManagerFactory._build_manager()._set(task_1)
  325. task_2 = Task("task_2", {}, print, input=[data_node])
  326. _TaskManagerFactory._build_manager()._set(task_2)
  327. scenario = Scenario("scenario", {task_1, task_2}, {})
  328. _ScenarioManagerFactory._build_manager()._set(scenario)
  329. scenario.add_sequence("sequence_1", [task_1])
  330. scenario.add_sequence("sequence_2", [task_2])
  331. with pytest.raises(SequenceAlreadyExists):
  332. scenario.rename_sequence("sequence_1", "sequence_2")
  333. def test_add_rename_and_remove_sequences():
  334. data_node_1 = InMemoryDataNode("foo", Scope.SCENARIO, "s1")
  335. data_node_2 = InMemoryDataNode("bar", Scope.SCENARIO, "s2")
  336. data_node_3 = InMemoryDataNode("qux", Scope.SCENARIO, "s3")
  337. data_node_4 = InMemoryDataNode("quux", Scope.SCENARIO, "s4")
  338. data_node_5 = InMemoryDataNode("quuz", Scope.SCENARIO, "s5")
  339. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_3], TaskId("t1"))
  340. task_2 = Task("garply", {}, print, [data_node_3], id=TaskId("t2"))
  341. task_3 = Task("waldo", {}, print, [data_node_3], None, id=TaskId("t3"))
  342. task_4 = Task("fred", {}, print, [data_node_3], [data_node_4], TaskId("t4"))
  343. task_5 = Task("bob", {}, print, [data_node_5], [data_node_3], TaskId("t5"))
  344. scenario = Scenario("quest", {task_1, task_2, task_3, task_4, task_5}, {}, [], scenario_id=ScenarioId("s1"))
  345. sequence_1 = Sequence({"name": "seq_1"}, [task_1], SequenceId(f"SEQUENCE_seq_1_{scenario.id}"))
  346. sequence_2 = Sequence({"name": "seq_2"}, [task_1, task_2], SequenceId(f"SEQUENCE_seq_2_{scenario.id}"))
  347. new_seq_2 = Sequence({"name": "seq_2"}, [task_1, task_2], SequenceId(f"SEQUENCE_new_seq_2_{scenario.id}"))
  348. seq_3 = Sequence({"name": "seq 3"}, [task_1, task_5, task_3], SequenceId(f"SEQUENCE_seqTPSPACE3_{scenario.id}"))
  349. task_manager = _TaskManagerFactory._build_manager()
  350. data_manager = _DataManagerFactory._build_manager()
  351. scenario_manager = _ScenarioManagerFactory._build_manager()
  352. for dn in [data_node_1, data_node_2, data_node_3, data_node_4, data_node_5]:
  353. data_manager._set(dn)
  354. for t in [task_1, task_2, task_3, task_4, task_5]:
  355. task_manager._set(t)
  356. scenario_manager._set(scenario)
  357. assert scenario.get_inputs() == {data_node_1, data_node_2, data_node_5}
  358. assert scenario._get_set_of_tasks() == {task_1, task_2, task_3, task_4, task_5}
  359. assert len(scenario.sequences) == 0
  360. scenario.sequences = {"seq_1": {"tasks": [task_1]}}
  361. assert scenario.sequences == {"seq_1": sequence_1}
  362. scenario.add_sequences({"seq_2": [task_1, task_2]})
  363. assert scenario.sequences == {"seq_1": sequence_1, "seq_2": sequence_2}
  364. scenario.remove_sequences(["seq_1"])
  365. assert scenario.sequences == {"seq_2": sequence_2}
  366. scenario.add_sequences({"seq_1": [task_1], "seq 3": [task_1, task_5, task_3]})
  367. assert scenario.sequences == {"seq_2": sequence_2, "seq_1": sequence_1, "seq 3": seq_3}
  368. scenario.remove_sequences(["seq_2", "seq 3"])
  369. assert scenario.sequences == {"seq_1": sequence_1}
  370. scenario.add_sequence("seq_2", [task_1, task_2])
  371. assert scenario.sequences == {"seq_1": sequence_1, "seq_2": sequence_2}
  372. scenario.add_sequence("seq 3", [task_1.id, task_5.id, task_3.id])
  373. assert scenario.sequences == {"seq_1": sequence_1, "seq_2": sequence_2, "seq 3": seq_3}
  374. scenario.remove_sequence("seq_1")
  375. assert scenario.sequences == {"seq_2": sequence_2, "seq 3": seq_3}
  376. scenario.rename_sequence("seq_2", "new_seq_2")
  377. assert scenario.sequences == {"new_seq_2": new_seq_2, "seq 3": seq_3}
  378. def test_update_sequence(data_node):
  379. task_1 = Task("foo", {}, print, [data_node], [], TaskId("t1"))
  380. task_2 = Task("bar", {}, print, [], [data_node], id=TaskId("t2"))
  381. scenario = Scenario("baz", {task_1, task_2}, {})
  382. scenario.add_sequence("seq_1", [task_1])
  383. assert len(scenario.sequences) == 1
  384. assert scenario.sequences["seq_1"].tasks == {"foo": task_1}
  385. assert scenario.sequences["seq_1"].name == "seq_1"
  386. scenario.update_sequence("seq_1", [task_2], {"new_key": "new_value"}, [])
  387. assert len(scenario.sequences) == 1
  388. assert scenario.sequences["seq_1"].tasks == {"bar": task_2}
  389. assert scenario.sequences["seq_1"].name == "seq_1"
  390. assert scenario.sequences["seq_1"].properties["new_key"] == "new_value"
  391. def test_add_rename_and_remove_sequences_within_context(data_node):
  392. task_1 = Task("task_1", {}, print, output=[data_node])
  393. task_2 = Task("task_2", {}, print, input=[data_node])
  394. _TaskManagerFactory._build_manager()._set(task_1)
  395. scenario = Scenario(config_id="scenario", tasks={task_1, task_2}, properties={})
  396. _ScenarioManagerFactory._build_manager()._set(scenario)
  397. with scenario as sc:
  398. sc.add_sequence("seq_name", [task_1])
  399. assert len(scenario.sequences) == 1
  400. assert scenario.sequences["seq_name"].tasks == {"task_1": task_1}
  401. with scenario as sc:
  402. sc.update_sequence("seq_name", [task_2])
  403. assert len(scenario.sequences) == 1
  404. assert scenario.sequences["seq_name"].tasks == {"task_2": task_2}
  405. with scenario as sc:
  406. sc.rename_sequence("seq_name", "seq name")
  407. assert scenario.sequences["seq name"].tasks == {"task_2": task_2}
  408. with scenario as sc:
  409. sc.remove_sequence("seq name")
  410. assert len(scenario.sequences) == 0
  411. def test_add_property_to_scenario():
  412. scenario = Scenario("foo", set(), {"key": "value"})
  413. assert scenario.properties == {"key": "value"}
  414. assert scenario.key == "value"
  415. scenario.properties["new_key"] = "new_value"
  416. assert scenario.properties == {"key": "value", "new_key": "new_value"}
  417. assert scenario.key == "value"
  418. assert scenario.new_key == "new_value"
  419. def test_add_cycle_to_scenario(cycle):
  420. scenario = Scenario("foo", set(), {})
  421. assert scenario.cycle is None
  422. _CycleManagerFactory._build_manager()._set(cycle)
  423. scenario.cycle = cycle
  424. assert scenario.cycle == cycle
  425. def test_add_and_remove_subscriber():
  426. scenario = Scenario("foo", set(), {})
  427. scenario._add_subscriber(print)
  428. assert len(scenario.subscribers) == 1
  429. scenario._remove_subscriber(print)
  430. assert len(scenario.subscribers) == 0
  431. def test_add_and_remove_tag():
  432. scenario = Scenario("foo", set(), {})
  433. assert len(scenario.tags) == 0
  434. scenario._add_tag("tag")
  435. assert len(scenario.tags) == 1
  436. scenario._remove_tag("tag")
  437. assert len(scenario.tags) == 0
  438. def test_auto_set_and_reload(cycle, current_datetime, task, data_node):
  439. scenario_1 = Scenario(
  440. "foo",
  441. set(),
  442. {"name": "bar"},
  443. set(),
  444. creation_date=current_datetime,
  445. is_primary=True,
  446. cycle=None,
  447. )
  448. additional_dn = InMemoryDataNode("additional_dn", Scope.SCENARIO)
  449. example_date = datetime.fromisoformat("2021-11-11T11:11:01.000001")
  450. tmp_cycle = Cycle(
  451. Frequency.WEEKLY,
  452. {},
  453. creation_date=example_date,
  454. start_date=example_date,
  455. end_date=example_date,
  456. name="cc",
  457. id=CycleId("tmp_cc_id"),
  458. )
  459. sequence_1_name = "sequence_1"
  460. sequence_1 = Sequence({}, [], SequenceId(f"SEQUENCE_{sequence_1_name}_{scenario_1.id}"))
  461. tmp_sequence_name = "tmp_sequence"
  462. tmp_sequence = Sequence(
  463. {},
  464. [],
  465. SequenceId(f"SEQUENCE_{tmp_sequence_name}_{scenario_1.id}"),
  466. )
  467. _TaskManagerFactory._build_manager()._set(task)
  468. _DataManagerFactory._build_manager()._set(data_node)
  469. _DataManagerFactory._build_manager()._set(additional_dn)
  470. _CycleManagerFactory._build_manager()._set(cycle)
  471. scenario_manager = _ScenarioManagerFactory._build_manager()
  472. cycle_manager = _CycleManagerFactory._build_manager()
  473. cycle_manager._set(cycle)
  474. cycle_manager._set(tmp_cycle)
  475. scenario_manager._set(scenario_1)
  476. scenario_2 = scenario_manager._get(scenario_1)
  477. assert scenario_1.config_id == "foo"
  478. assert scenario_2.config_id == "foo"
  479. # auto set & reload on name attribute
  480. assert scenario_1.name == "bar"
  481. assert scenario_2.name == "bar"
  482. scenario_1.name = "zab"
  483. assert scenario_1.name == "zab"
  484. assert scenario_2.name == "zab"
  485. scenario_2.name = "baz"
  486. assert scenario_1.name == "baz"
  487. assert scenario_2.name == "baz"
  488. # auto set & reload on sequences attribute
  489. assert len(scenario_1.sequences) == 0
  490. assert len(scenario_2.sequences) == 0
  491. scenario_1.sequences = {tmp_sequence_name: {}}
  492. assert len(scenario_1.sequences) == 1
  493. assert scenario_1.sequences[tmp_sequence_name] == tmp_sequence
  494. assert len(scenario_2.sequences) == 1
  495. assert scenario_2.sequences[tmp_sequence_name] == tmp_sequence
  496. scenario_2.add_sequences({sequence_1_name: []})
  497. assert len(scenario_1.sequences) == 2
  498. assert scenario_1.sequences == {sequence_1_name: sequence_1, tmp_sequence_name: tmp_sequence}
  499. assert len(scenario_2.sequences) == 2
  500. assert scenario_2.sequences == {sequence_1_name: sequence_1, tmp_sequence_name: tmp_sequence}
  501. scenario_2.remove_sequences([tmp_sequence_name])
  502. assert len(scenario_1.sequences) == 1
  503. assert scenario_1.sequences == {sequence_1_name: sequence_1}
  504. assert len(scenario_2.sequences) == 1
  505. assert scenario_2.sequences == {sequence_1_name: sequence_1}
  506. assert len(scenario_1.tasks) == 0
  507. assert len(scenario_1.data_nodes) == 0
  508. scenario_1.tasks = {task}
  509. assert len(scenario_1.tasks) == 1
  510. assert scenario_1.tasks[task.config_id] == task
  511. assert len(scenario_1.data_nodes) == 2
  512. assert len(scenario_2.tasks) == 1
  513. assert scenario_2.tasks[task.config_id] == task
  514. assert len(scenario_2.data_nodes) == 2
  515. assert len(scenario_1.additional_data_nodes) == 0
  516. scenario_1.additional_data_nodes = {additional_dn}
  517. assert len(scenario_1.additional_data_nodes) == 1
  518. assert scenario_1.additional_data_nodes[additional_dn.config_id] == additional_dn
  519. assert len(scenario_1.data_nodes) == 3
  520. assert len(scenario_2.additional_data_nodes) == 1
  521. assert scenario_2.additional_data_nodes[additional_dn.config_id] == additional_dn
  522. assert len(scenario_2.data_nodes) == 3
  523. new_datetime = current_datetime + timedelta(1)
  524. new_datetime_1 = current_datetime + timedelta(2)
  525. # auto set & reload on name attribute
  526. assert scenario_1.creation_date == current_datetime
  527. assert scenario_2.creation_date == current_datetime
  528. scenario_1.creation_date = new_datetime_1
  529. assert scenario_1.creation_date == new_datetime_1
  530. assert scenario_2.creation_date == new_datetime_1
  531. scenario_2.creation_date = new_datetime
  532. assert scenario_1.creation_date == new_datetime
  533. assert scenario_2.creation_date == new_datetime
  534. # auto set & reload on cycle attribute
  535. assert scenario_1.cycle is None
  536. assert scenario_2.cycle is None
  537. scenario_1.cycle = tmp_cycle
  538. assert scenario_1.cycle == tmp_cycle
  539. assert scenario_2.cycle == tmp_cycle
  540. scenario_2.cycle = cycle
  541. assert scenario_1.cycle == cycle
  542. assert scenario_2.cycle == cycle
  543. # auto set & reload on is_primary attribute
  544. assert scenario_1.is_primary
  545. assert scenario_2.is_primary
  546. scenario_1.is_primary = False
  547. assert not scenario_1.is_primary
  548. assert not scenario_2.is_primary
  549. scenario_2.is_primary = True
  550. assert scenario_1.is_primary
  551. assert scenario_2.is_primary
  552. # auto set & reload on subscribers attribute
  553. assert len(scenario_1.subscribers) == 0
  554. assert len(scenario_2.subscribers) == 0
  555. scenario_1.subscribers.append(_Subscriber(print, []))
  556. assert len(scenario_1.subscribers) == 1
  557. assert len(scenario_2.subscribers) == 1
  558. scenario_2.subscribers.append(_Subscriber(print, []))
  559. assert len(scenario_1.subscribers) == 2
  560. assert len(scenario_2.subscribers) == 2
  561. scenario_1.subscribers.clear()
  562. assert len(scenario_1.subscribers) == 0
  563. assert len(scenario_2.subscribers) == 0
  564. scenario_1.subscribers.extend([_Subscriber(print, []), _Subscriber(map, [])])
  565. assert len(scenario_1.subscribers) == 2
  566. assert len(scenario_2.subscribers) == 2
  567. scenario_1.subscribers.remove(_Subscriber(print, []))
  568. assert len(scenario_1.subscribers) == 1
  569. assert len(scenario_2.subscribers) == 1
  570. scenario_1.subscribers + print + len
  571. assert len(scenario_1.subscribers) == 3
  572. assert len(scenario_2.subscribers) == 3
  573. scenario_1.subscribers = []
  574. assert len(scenario_1.subscribers) == 0
  575. assert len(scenario_2.subscribers) == 0
  576. assert len(scenario_1.tags) == 0
  577. scenario_1.tags = {"hi"}
  578. assert len(scenario_1.tags) == 1
  579. assert len(scenario_2.tags) == 1
  580. with scenario_1 as scenario:
  581. assert scenario.config_id == "foo"
  582. assert len(scenario.tasks) == 1
  583. assert len(scenario.sequences) == 1
  584. assert scenario.sequences["sequence_1"] == sequence_1
  585. assert scenario.tasks[task.config_id] == task
  586. assert len(scenario.additional_data_nodes) == 1
  587. assert scenario.additional_data_nodes[additional_dn.config_id] == additional_dn
  588. assert scenario.creation_date == new_datetime
  589. assert scenario.cycle == cycle
  590. assert scenario.is_primary
  591. assert len(scenario.subscribers) == 0
  592. assert len(scenario.tags) == 1
  593. assert scenario._is_in_context
  594. assert scenario.name == "baz"
  595. new_datetime_2 = new_datetime + timedelta(5)
  596. scenario._config_id = "foo"
  597. scenario.tasks = set()
  598. scenario.additional_data_nodes = set()
  599. scenario.remove_sequences([sequence_1_name])
  600. scenario.creation_date = new_datetime_2
  601. scenario.cycle = None
  602. scenario.is_primary = False
  603. scenario.subscribers = [print]
  604. scenario.tags = None
  605. scenario.name = "qux"
  606. assert scenario.config_id == "foo"
  607. assert len(scenario.sequences) == 1
  608. assert scenario.sequences[sequence_1_name] == sequence_1
  609. assert len(scenario.tasks) == 1
  610. assert scenario.tasks[task.config_id] == task
  611. assert len(scenario.additional_data_nodes) == 1
  612. assert scenario.additional_data_nodes[additional_dn.config_id] == additional_dn
  613. assert scenario.creation_date == new_datetime
  614. assert scenario.cycle == cycle
  615. assert scenario.is_primary
  616. assert len(scenario.subscribers) == 0
  617. assert len(scenario.tags) == 1
  618. assert scenario._is_in_context
  619. assert scenario.name == "baz"
  620. assert scenario_1.config_id == "foo"
  621. assert len(scenario_1.sequences) == 0
  622. assert len(scenario_1.tasks) == 0
  623. assert len(scenario_1.additional_data_nodes) == 0
  624. assert scenario_1.tasks == {}
  625. assert scenario_1.additional_data_nodes == {}
  626. assert scenario_1.creation_date == new_datetime_2
  627. assert scenario_1.cycle is None
  628. assert not scenario_1.is_primary
  629. assert len(scenario_1.subscribers) == 1
  630. assert len(scenario_1.tags) == 0
  631. assert not scenario_1._is_in_context
  632. def test_auto_set_and_reload_properties():
  633. scenario_1 = Scenario(
  634. "foo",
  635. set(),
  636. {"name": "baz"},
  637. )
  638. scenario_manager = _ScenarioManagerFactory._build_manager()
  639. scenario_manager._set(scenario_1)
  640. scenario_2 = scenario_manager._get(scenario_1)
  641. # auto set & reload on properties attribute
  642. assert scenario_1.properties == {"name": "baz"}
  643. assert scenario_2.properties == {"name": "baz"}
  644. scenario_1._properties["qux"] = 4
  645. assert scenario_1.properties["qux"] == 4
  646. assert scenario_2.properties["qux"] == 4
  647. assert scenario_1.properties == {"name": "baz", "qux": 4}
  648. assert scenario_2.properties == {"name": "baz", "qux": 4}
  649. scenario_2._properties["qux"] = 5
  650. assert scenario_1.properties["qux"] == 5
  651. assert scenario_2.properties["qux"] == 5
  652. scenario_1.properties["temp_key_1"] = "temp_value_1"
  653. scenario_1.properties["temp_key_2"] = "temp_value_2"
  654. assert scenario_1.properties == {
  655. "name": "baz",
  656. "qux": 5,
  657. "temp_key_1": "temp_value_1",
  658. "temp_key_2": "temp_value_2",
  659. }
  660. assert scenario_2.properties == {
  661. "name": "baz",
  662. "qux": 5,
  663. "temp_key_1": "temp_value_1",
  664. "temp_key_2": "temp_value_2",
  665. }
  666. scenario_1.properties.pop("temp_key_1")
  667. assert "temp_key_1" not in scenario_1.properties.keys()
  668. assert "temp_key_1" not in scenario_1.properties.keys()
  669. assert scenario_1.properties == {
  670. "name": "baz",
  671. "qux": 5,
  672. "temp_key_2": "temp_value_2",
  673. }
  674. assert scenario_2.properties == {
  675. "name": "baz",
  676. "qux": 5,
  677. "temp_key_2": "temp_value_2",
  678. }
  679. scenario_2.properties.pop("temp_key_2")
  680. assert scenario_1.properties == {"name": "baz", "qux": 5}
  681. assert scenario_2.properties == {"name": "baz", "qux": 5}
  682. assert "temp_key_2" not in scenario_1.properties.keys()
  683. assert "temp_key_2" not in scenario_2.properties.keys()
  684. scenario_1.properties["temp_key_3"] = 0
  685. assert scenario_1.properties == {"name": "baz", "qux": 5, "temp_key_3": 0}
  686. assert scenario_2.properties == {"name": "baz", "qux": 5, "temp_key_3": 0}
  687. scenario_1.properties.update({"temp_key_3": 1})
  688. assert scenario_1.properties == {"name": "baz", "qux": 5, "temp_key_3": 1}
  689. assert scenario_2.properties == {"name": "baz", "qux": 5, "temp_key_3": 1}
  690. scenario_1.properties.update({})
  691. assert scenario_1.properties == {"name": "baz", "qux": 5, "temp_key_3": 1}
  692. assert scenario_2.properties == {"name": "baz", "qux": 5, "temp_key_3": 1}
  693. scenario_1.properties["temp_key_4"] = 0
  694. scenario_1.properties["temp_key_5"] = 0
  695. with scenario_1 as scenario:
  696. assert scenario.properties["qux"] == 5
  697. assert scenario.properties["temp_key_3"] == 1
  698. assert scenario.properties["temp_key_4"] == 0
  699. assert scenario.properties["temp_key_5"] == 0
  700. scenario.properties["qux"] = 9
  701. scenario.properties.pop("temp_key_3")
  702. scenario.properties.pop("temp_key_4")
  703. scenario.properties.update({"temp_key_4": 1})
  704. scenario.properties.update({"temp_key_5": 2})
  705. scenario.properties.pop("temp_key_5")
  706. scenario.properties.update({})
  707. assert scenario._is_in_context
  708. assert scenario.properties["qux"] == 5
  709. assert scenario.properties["temp_key_3"] == 1
  710. assert scenario.properties["temp_key_4"] == 0
  711. assert scenario.properties["temp_key_5"] == 0
  712. assert not scenario_1._is_in_context
  713. assert scenario_1.properties["qux"] == 9
  714. assert "temp_key_3" not in scenario_1.properties.keys()
  715. assert scenario_1.properties["temp_key_4"] == 1
  716. assert "temp_key_5" not in scenario_1.properties.keys()
  717. def test_is_deletable():
  718. with mock.patch("taipy.core.scenario._scenario_manager._ScenarioManager._is_deletable") as mock_submit:
  719. scenario = Scenario("foo", set(), {})
  720. scenario.is_deletable()
  721. mock_submit.assert_called_once_with(scenario)
  722. def test_submit_scenario():
  723. with mock.patch("taipy.core.scenario._scenario_manager._ScenarioManager._submit") as mock_submit:
  724. scenario = Scenario("foo", set(), {})
  725. scenario.submit(force=False)
  726. mock_submit.assert_called_once_with(scenario, None, False, False, None)
  727. def test_subscribe_scenario():
  728. with mock.patch("taipy.core.subscribe_scenario") as mock_subscribe:
  729. scenario = Scenario("foo", set(), {})
  730. scenario.subscribe(None)
  731. mock_subscribe.assert_called_once_with(None, None, scenario)
  732. def test_unsubscribe_scenario():
  733. with mock.patch("taipy.core.unsubscribe_scenario") as mock_unsubscribe:
  734. scenario = Scenario("foo", set(), {})
  735. scenario.unsubscribe(None)
  736. mock_unsubscribe.assert_called_once_with(None, None, scenario)
  737. def test_add_tag_scenario():
  738. with mock.patch("taipy.core.tag") as mock_add_tag:
  739. scenario = Scenario("foo", set(), {})
  740. scenario.add_tag("tag")
  741. mock_add_tag.assert_called_once_with(scenario, "tag")
  742. def test_remove_tag_scenario():
  743. with mock.patch("taipy.core.untag") as mock_remove_tag:
  744. scenario = Scenario("foo", set(), {})
  745. scenario.remove_tag("tag")
  746. mock_remove_tag.assert_called_once_with(scenario, "tag")
  747. def test_get_inputs_outputs_intermediate_data_nodes():
  748. data_node_1 = DataNode("foo", Scope.SCENARIO, "s1")
  749. data_node_2 = DataNode("bar", Scope.SCENARIO, "s2")
  750. data_node_3 = DataNode("baz", Scope.SCENARIO, "s3")
  751. data_node_4 = DataNode("qux", Scope.SCENARIO, "s4")
  752. data_node_5 = DataNode("quux", Scope.SCENARIO, "s5")
  753. data_node_6 = DataNode("quuz", Scope.SCENARIO, "s6")
  754. data_node_7 = DataNode("corge", Scope.SCENARIO, "s7")
  755. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_3, data_node_4], TaskId("t1"))
  756. task_2 = Task("garply", {}, print, [data_node_3], [data_node_5], TaskId("t2"))
  757. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], [data_node_6], TaskId("t3"))
  758. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  759. scenario = Scenario("scenario", {task_1, task_2, task_3, task_4}, {}, set(), ScenarioId("s1"))
  760. # s1 --- ---> s3 ---> t2 ---> s5 ----
  761. # | | |
  762. # |---> t1 ---| -------------------------> t3 ---> s6
  763. # | | |
  764. # s2 --- ---> s4 ---> t4 ---> s7
  765. assert scenario.get_inputs() == {data_node_1, data_node_2}
  766. assert scenario.get_outputs() == {data_node_6, data_node_7}
  767. assert scenario.get_intermediate() == {data_node_3, data_node_4, data_node_5}
  768. data_node_1 = DataNode("foo", Scope.SCENARIO, "s1")
  769. data_node_2 = DataNode("bar", Scope.SCENARIO, "s2")
  770. data_node_4 = DataNode("qux", Scope.SCENARIO, "s4")
  771. data_node_5 = DataNode("quux", Scope.SCENARIO, "s5")
  772. data_node_6 = DataNode("quuz", Scope.SCENARIO, "s6")
  773. data_node_7 = DataNode("corge", Scope.SCENARIO, "s7")
  774. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  775. task_2 = Task("garply", {}, print, None, [data_node_5], TaskId("t2"))
  776. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], [data_node_6], TaskId("t3"))
  777. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  778. scenario = Scenario("scenario", {task_1, task_2, task_3, task_4}, {}, set(), ScenarioId("s1"))
  779. # s1 --- t2 ---> s5 ------
  780. # | |
  781. # |---> t1 ---| -----> t3 ---> s6
  782. # | | |
  783. # s2 --- ---> s4 ---> t4 ---> s7
  784. assert scenario.get_inputs() == {data_node_1, data_node_2}
  785. assert scenario.get_outputs() == {data_node_6, data_node_7}
  786. assert scenario.get_intermediate() == {data_node_4, data_node_5}
  787. data_node_1 = DataNode("foo", Scope.SCENARIO, "s1")
  788. data_node_2 = DataNode("bar", Scope.SCENARIO, "s2")
  789. data_node_4 = DataNode("qux", Scope.SCENARIO, "s4")
  790. data_node_5 = DataNode("quux", Scope.SCENARIO, "s5")
  791. data_node_6 = DataNode("quuz", Scope.SCENARIO, "s6")
  792. data_node_7 = DataNode("corge", Scope.SCENARIO, "s7")
  793. data_node_8 = DataNode("d8", Scope.SCENARIO, "s8")
  794. data_node_9 = DataNode("d9", Scope.SCENARIO, "s9")
  795. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  796. task_2 = Task("garply", {}, print, [data_node_6], [data_node_5], TaskId("t2"))
  797. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], id=TaskId("t3"))
  798. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  799. task_5 = Task("t5", {}, print, [data_node_8], [data_node_9], TaskId("t5"))
  800. task_6 = Task("t6", {}, print, [data_node_7, data_node_9], id=TaskId("t6"))
  801. scenario = Scenario("scenario", {task_1, task_2, task_3, task_4, task_5, task_6}, {}, set(), ScenarioId("s1"))
  802. # s1 --- s6 ---> t2 ---> s5
  803. # | |
  804. # |---> t1 ---| -----> t3
  805. # | | |
  806. # s2 --- ---> s4 ---> t4 ---> s7 ---> t6
  807. # |
  808. # s8 -------> t5 -------> s9 ------------------
  809. assert scenario.get_inputs() == {data_node_1, data_node_2, data_node_6, data_node_8}
  810. assert scenario.get_outputs() == set()
  811. assert scenario.get_intermediate() == {data_node_5, data_node_4, data_node_7, data_node_9}
  812. data_node_1 = DataNode("foo", Scope.SCENARIO, "s1")
  813. data_node_2 = DataNode("bar", Scope.SCENARIO, "s2")
  814. data_node_4 = DataNode("qux", Scope.SCENARIO, "s4")
  815. data_node_5 = DataNode("quux", Scope.SCENARIO, "s5")
  816. data_node_6 = DataNode("quuz", Scope.SCENARIO, "s6")
  817. data_node_7 = DataNode("corge", Scope.SCENARIO, "s7")
  818. data_node_8 = DataNode("hugh", Scope.SCENARIO, "s8")
  819. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  820. task_2 = Task("garply", {}, print, output=[data_node_5], id=TaskId("t2"))
  821. task_3 = Task("waldo", {}, print, [data_node_4], None, id=TaskId("t3"))
  822. task_4 = Task("fred", {}, print, [data_node_4, data_node_6], [data_node_7], TaskId("t4"))
  823. task_5 = Task("bob", {}, print, [data_node_8], None, TaskId("t5"))
  824. scenario = Scenario("scenario", {task_1, task_2, task_3, task_4, task_5}, {}, set(), ScenarioId("sc1"))
  825. # s1 ---
  826. # |
  827. # |---> t1 ---| -----> t3
  828. # | | |
  829. # s2 --- ---> s4 ---> t4 ---> s7
  830. # t2 ---> s5 |
  831. # s8 ---> t5 s6 --|
  832. assert scenario.get_inputs() == {data_node_1, data_node_2, data_node_8, data_node_6}
  833. assert scenario.get_outputs() == {data_node_5, data_node_7}
  834. assert scenario.get_intermediate() == {data_node_4}
  835. def test_is_ready_to_run():
  836. task_1_id, task_2_id, task_3_id, task_4_id, task_5_id, task_6_id = (
  837. TaskId("TASK_t1"),
  838. TaskId("TASK_t2"),
  839. TaskId("TASK_t3"),
  840. TaskId("TASK_t4"),
  841. TaskId("TASK_t5"),
  842. TaskId("TASK_t6"),
  843. )
  844. sc_id = ScenarioId("SCENARIO_s1")
  845. data_node_1 = PickleDataNode("foo", Scope.SCENARIO, "d1", parent_ids={task_1_id}, properties={"default_data": 1})
  846. data_node_2 = PickleDataNode("bar", Scope.SCENARIO, "d2", parent_ids={task_1_id}, properties={"default_data": 2})
  847. data_node_4 = PickleDataNode(
  848. "qux", Scope.SCENARIO, "d4", parent_ids={task_1_id, task_4_id, task_3_id}, properties={"default_data": 4}
  849. )
  850. data_node_5 = PickleDataNode(
  851. "quux", Scope.SCENARIO, "d5", parent_ids={task_2_id, task_3_id}, properties={"default_data": 5}
  852. )
  853. data_node_6 = PickleDataNode("quuz", Scope.SCENARIO, "d6", parent_ids={task_2_id}, properties={"default_data": 6})
  854. data_node_7 = PickleDataNode(
  855. "corge", Scope.SCENARIO, "d7", parent_ids={task_4_id, task_6_id}, properties={"default_data": 7}
  856. )
  857. data_node_8 = PickleDataNode("d8", Scope.SCENARIO, "d8", parent_ids={task_5_id}, properties={"default_data": 8})
  858. data_node_9 = PickleDataNode(
  859. "d9", Scope.SCENARIO, "d9", parent_ids={task_5_id, task_6_id}, properties={"default_data": 9}
  860. )
  861. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], id=task_1_id, parent_ids={sc_id})
  862. task_2 = Task("garply", {}, print, [data_node_6], [data_node_5], id=task_2_id, parent_ids={sc_id})
  863. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], id=task_3_id, parent_ids={sc_id})
  864. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], id=task_4_id, parent_ids={sc_id})
  865. task_5 = Task("t5", {}, print, [data_node_8], [data_node_9], id=task_5_id, parent_ids={sc_id})
  866. task_6 = Task("t6", {}, print, [data_node_7, data_node_9], id=task_6_id, parent_ids={sc_id})
  867. scenario = Scenario("scenario", {task_1, task_2, task_3, task_4, task_5, task_6}, {}, set(), scenario_id=sc_id)
  868. # d1 --- d6 ---> t2 ---> d5
  869. # | |
  870. # |---> t1 ---| -----> t3
  871. # | | |
  872. # d2 --- ---> d4 ---> t4 ---> d7 ---> t6
  873. # |
  874. # d8 -------> t5 -------> d9 ------------------
  875. assert scenario.get_inputs() == {data_node_1, data_node_2, data_node_6, data_node_8}
  876. data_manager = _DataManagerFactory._build_manager()
  877. data_manager._delete_all()
  878. for dn in [data_node_1, data_node_2, data_node_4, data_node_5, data_node_6, data_node_7, data_node_8, data_node_9]:
  879. data_manager._set(dn)
  880. task_manager = _TaskManagerFactory._build_manager()
  881. for task in [task_1, task_2, task_3, task_4, task_5, task_6]:
  882. task_manager._set(task)
  883. _ScenarioManagerFactory._build_manager()._set(scenario)
  884. assert scenario.is_ready_to_run()
  885. data_node_1.edit_in_progress = True
  886. assert not scenario.is_ready_to_run()
  887. data_node_2.edit_in_progress = True
  888. assert not scenario.is_ready_to_run()
  889. data_node_6.edit_in_progress = True
  890. data_node_8.edit_in_progress = True
  891. assert not scenario.is_ready_to_run()
  892. data_node_1.edit_in_progress = False
  893. data_node_2.edit_in_progress = False
  894. data_node_6.edit_in_progress = False
  895. data_node_8.edit_in_progress = False
  896. assert scenario.is_ready_to_run()
  897. def test_data_nodes_being_edited():
  898. data_node_1 = PickleDataNode("foo", Scope.SCENARIO, "s1", properties={"default_data": 1})
  899. data_node_2 = PickleDataNode("bar", Scope.SCENARIO, "s2", properties={"default_data": 2})
  900. data_node_4 = PickleDataNode("qux", Scope.SCENARIO, "s4", properties={"default_data": 4})
  901. data_node_5 = PickleDataNode("quux", Scope.SCENARIO, "s5", properties={"default_data": 5})
  902. data_node_6 = PickleDataNode("quuz", Scope.SCENARIO, "s6", properties={"default_data": 6})
  903. data_node_7 = PickleDataNode("corge", Scope.SCENARIO, "s7", properties={"default_data": 7})
  904. data_node_8 = PickleDataNode("d8", Scope.SCENARIO, "s8", properties={"default_data": 8})
  905. data_node_9 = PickleDataNode("d9", Scope.SCENARIO, "s9", properties={"default_data": 9})
  906. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  907. task_2 = Task("garply", {}, print, [data_node_6], [data_node_5], TaskId("t2"))
  908. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], id=TaskId("t3"))
  909. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  910. task_5 = Task("t5", {}, print, [data_node_8], [data_node_9], TaskId("t5"))
  911. task_6 = Task("t6", {}, print, [data_node_7, data_node_9], id=TaskId("t6"))
  912. scenario = Scenario("scenario", {task_1, task_2, task_3, task_4, task_5, task_6}, {}, set(), ScenarioId("s1"))
  913. # s1 --- s6 ---> t2 ---> s5
  914. # | |
  915. # |---> t1 ---| -----> t3
  916. # | | |
  917. # s2 --- ---> s4 ---> t4 ---> s7 ---> t6
  918. # |
  919. # s8 -------> t5 -------> s9 ------------------
  920. data_manager = _DataManagerFactory._build_manager()
  921. for dn in [data_node_1, data_node_2, data_node_4, data_node_5, data_node_6, data_node_7, data_node_8, data_node_9]:
  922. data_manager._set(dn)
  923. assert len(scenario.data_nodes_being_edited()) == 0
  924. assert scenario.data_nodes_being_edited() == set()
  925. data_node_1.edit_in_progress = True
  926. assert len(scenario.data_nodes_being_edited()) == 1
  927. assert scenario.data_nodes_being_edited() == {data_node_1}
  928. data_node_2.edit_in_progress = True
  929. data_node_6.edit_in_progress = True
  930. data_node_8.edit_in_progress = True
  931. assert len(scenario.data_nodes_being_edited()) == 4
  932. assert scenario.data_nodes_being_edited() == {data_node_1, data_node_2, data_node_6, data_node_8}
  933. data_node_4.edit_in_progress = True
  934. data_node_5.edit_in_progress = True
  935. data_node_9.edit_in_progress = True
  936. assert len(scenario.data_nodes_being_edited()) == 7
  937. assert scenario.data_nodes_being_edited() == {
  938. data_node_1,
  939. data_node_2,
  940. data_node_4,
  941. data_node_5,
  942. data_node_6,
  943. data_node_8,
  944. data_node_9,
  945. }
  946. data_node_1.edit_in_progress = False
  947. data_node_2.edit_in_progress = False
  948. data_node_6.edit_in_progress = False
  949. data_node_8.edit_in_progress = False
  950. assert len(scenario.data_nodes_being_edited()) == 3
  951. assert scenario.data_nodes_being_edited() == {data_node_4, data_node_5, data_node_9}
  952. data_node_4.edit_in_progress = False
  953. data_node_5.edit_in_progress = False
  954. data_node_7.edit_in_progress = True
  955. assert len(scenario.data_nodes_being_edited()) == 2
  956. assert scenario.data_nodes_being_edited() == {data_node_7, data_node_9}
  957. data_node_7.edit_in_progress = False
  958. data_node_9.edit_in_progress = False
  959. assert len(scenario.data_nodes_being_edited()) == 0
  960. assert scenario.data_nodes_being_edited() == set()
  961. def test_get_tasks():
  962. task_1 = Task("grault", {}, print, id=TaskId("t1"))
  963. task_2 = Task("garply", {}, print, id=TaskId("t2"))
  964. task_3 = Task("waldo", {}, print, id=TaskId("t3"))
  965. task_4 = Task("fred", {}, print, id=TaskId("t4"))
  966. scenario_1 = Scenario("scenario_1", {task_1, task_2, task_3, task_4}, {}, set(), ScenarioId("s1"))
  967. assert scenario_1.tasks == {"grault": task_1, "garply": task_2, "waldo": task_3, "fred": task_4}
  968. task_5 = Task("wallo", {}, print, id=TaskId("t5"))
  969. scenario_2 = Scenario("scenario_2", {task_1, task_2, task_3, task_4, task_5}, {}, set(), ScenarioId("s2"))
  970. assert scenario_2.tasks == {"grault": task_1, "garply": task_2, "waldo": task_3, "fred": task_4, "wallo": task_5}
  971. def test_get_set_of_tasks():
  972. task_1 = Task("grault", {}, print, id=TaskId("t1"))
  973. task_2 = Task("garply", {}, print, id=TaskId("t2"))
  974. task_3 = Task("waldo", {}, print, id=TaskId("t3"))
  975. task_4 = Task("fred", {}, print, id=TaskId("t4"))
  976. scenario_1 = Scenario("scenario_1", {task_1, task_2, task_3, task_4}, {}, set(), ScenarioId("s1"))
  977. assert scenario_1._get_set_of_tasks() == {task_1, task_2, task_3, task_4}
  978. task_5 = Task("wallo", {}, print, id=TaskId("t5"))
  979. scenario_2 = Scenario("scenario_2", {task_1, task_2, task_3, task_4, task_5}, {}, set(), ScenarioId("s2"))
  980. assert scenario_2._get_set_of_tasks() == {task_1, task_2, task_3, task_4, task_5}
  981. def test_get_sorted_tasks():
  982. def _assert_equal(tasks_a, tasks_b) -> bool:
  983. if len(tasks_a) != len(tasks_b):
  984. return False
  985. for i in range(len(tasks_a)):
  986. task_a, task_b = tasks_a[i], tasks_b[i]
  987. if isinstance(task_a, list) and isinstance(task_b, list):
  988. if not _assert_equal(task_a, task_b):
  989. return False
  990. elif isinstance(task_a, list) or isinstance(task_b, list):
  991. return False
  992. else:
  993. index_task_b = tasks_b.index(task_a)
  994. if any(isinstance(task_b, list) for task_b in tasks_b[i : index_task_b + 1]):
  995. return False
  996. return True
  997. # s1 --- ---> s3 ---> t2 ---> s5 ----
  998. # | | |
  999. # |---> t1 ---| -------------------------> t3 ---> s6
  1000. # | | |
  1001. # s2 --- ---> s4 ---> t4 ---> s7
  1002. data_node_1 = InMemoryDataNode("foo", Scope.SCENARIO, "s1")
  1003. data_node_2 = InMemoryDataNode("bar", Scope.SCENARIO, "s2")
  1004. data_node_3 = InMemoryDataNode("baz", Scope.SCENARIO, "s3")
  1005. data_node_4 = InMemoryDataNode("qux", Scope.SCENARIO, "s4")
  1006. data_node_5 = InMemoryDataNode("quux", Scope.SCENARIO, "s5")
  1007. data_node_6 = InMemoryDataNode("quuz", Scope.SCENARIO, "s6")
  1008. data_node_7 = InMemoryDataNode("corge", Scope.SCENARIO, "s7")
  1009. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_3, data_node_4], TaskId("t1"))
  1010. task_2 = Task("garply", {}, print, [data_node_3], [data_node_5], TaskId("t2"))
  1011. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], [data_node_6], TaskId("t3"))
  1012. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  1013. scenario_1 = Scenario("scenario_1", {task_1, task_2, task_3, task_4}, {}, [], ScenarioId("s1"))
  1014. assert scenario_1.get_inputs() == {data_node_1, data_node_2}
  1015. assert scenario_1._get_set_of_tasks() == {task_1, task_2, task_3, task_4}
  1016. _assert_equal(scenario_1._get_sorted_tasks(), [[task_1], [task_2, task_4], [task_3]])
  1017. # s1 --- t2 ---> s5
  1018. # | |
  1019. # |---> t1 ---| -----> t3 ---> s6
  1020. # | | |
  1021. # s2 --- ---> s4 ---> t4 ---> s7
  1022. data_node_1 = InMemoryDataNode("foo", Scope.SCENARIO, "s1")
  1023. data_node_2 = InMemoryDataNode("bar", Scope.SCENARIO, "s2")
  1024. data_node_4 = InMemoryDataNode("qux", Scope.SCENARIO, "s4")
  1025. data_node_5 = InMemoryDataNode("quux", Scope.SCENARIO, "s5")
  1026. data_node_6 = InMemoryDataNode("quuz", Scope.SCENARIO, "s6")
  1027. data_node_7 = InMemoryDataNode("corge", Scope.SCENARIO, "s7")
  1028. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  1029. task_2 = Task("garply", {}, print, None, [data_node_5], TaskId("t2"))
  1030. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], [data_node_6], TaskId("t3"))
  1031. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  1032. scenario_2 = Scenario("scenario_2", {task_1, task_2, task_3, task_4}, {}, [], ScenarioId("s2"))
  1033. assert scenario_2.get_inputs() == {data_node_1, data_node_2}
  1034. assert scenario_2._get_set_of_tasks() == {task_1, task_2, task_3, task_4}
  1035. _assert_equal(scenario_2._get_sorted_tasks(), [[task_1, task_2], [task_3, task_4]])
  1036. # s1 --- s6 ---> t2 ---> s5
  1037. # | |
  1038. # |---> t1 ---| -----> t3
  1039. # | | |
  1040. # s2 --- ---> s4 ---> t4 ---> s7
  1041. data_node_1 = DataNode("foo", Scope.SCENARIO, "s1")
  1042. data_node_2 = DataNode("bar", Scope.SCENARIO, "s2")
  1043. data_node_4 = DataNode("qux", Scope.SCENARIO, "s4")
  1044. data_node_5 = DataNode("quux", Scope.SCENARIO, "s5")
  1045. data_node_6 = DataNode("quuz", Scope.SCENARIO, "s6")
  1046. data_node_7 = DataNode("corge", Scope.SCENARIO, "s7")
  1047. task_1 = Task(
  1048. "grault",
  1049. {},
  1050. print,
  1051. [data_node_1, data_node_2],
  1052. [data_node_4],
  1053. TaskId("t1"),
  1054. )
  1055. task_2 = Task("garply", {}, print, [data_node_6], [data_node_5], TaskId("t2"))
  1056. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], id=TaskId("t3"))
  1057. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  1058. scenario_3 = Scenario("quest", [task_4, task_2, task_1, task_3], {}, [], scenario_id=ScenarioId("s3"))
  1059. assert scenario_3.get_inputs() == {data_node_1, data_node_2, data_node_6}
  1060. assert scenario_3._get_set_of_tasks() == {task_1, task_2, task_3, task_4}
  1061. assert _assert_equal(scenario_3._get_sorted_tasks(), [[task_2, task_1], [task_4, task_3]])
  1062. # s1 --- s6 ---> t2 ---> s5
  1063. # | |
  1064. # |---> t1 ---| -----> t3
  1065. # | | |
  1066. # s2 --- ---> s4 ---> t4 ---> s7 ---> t6
  1067. # |
  1068. # s8 -------> t5 -------> s9 ------------------
  1069. data_node_1 = InMemoryDataNode("foo", Scope.SCENARIO, "s1")
  1070. data_node_2 = InMemoryDataNode("bar", Scope.SCENARIO, "s2")
  1071. data_node_4 = InMemoryDataNode("qux", Scope.SCENARIO, "s4")
  1072. data_node_5 = InMemoryDataNode("quux", Scope.SCENARIO, "s5")
  1073. data_node_6 = InMemoryDataNode("quuz", Scope.SCENARIO, "s6")
  1074. data_node_7 = InMemoryDataNode("corge", Scope.SCENARIO, "s7")
  1075. data_node_8 = InMemoryDataNode("d8", Scope.SCENARIO, "s8")
  1076. data_node_9 = InMemoryDataNode("d9", Scope.SCENARIO, "s9")
  1077. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  1078. task_2 = Task("garply", {}, print, [data_node_6], [data_node_5], TaskId("t2"))
  1079. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], id=TaskId("t3"))
  1080. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  1081. task_5 = Task("t5", {}, print, [data_node_8], [data_node_9], TaskId("t5"))
  1082. task_6 = Task("t6", {}, print, [data_node_7, data_node_9], id=TaskId("t6"))
  1083. scenario_4 = Scenario("scenario_3", [task_1, task_2, task_3, task_4, task_5, task_6], {}, [], ScenarioId("s4"))
  1084. assert scenario_4.get_inputs() == {data_node_1, data_node_2, data_node_6, data_node_8}
  1085. assert scenario_4._get_set_of_tasks() == {task_1, task_2, task_3, task_4, task_5, task_6}
  1086. _assert_equal(scenario_4._get_sorted_tasks(), [[task_1, task_2, task_5], [task_3, task_4], [task_6]])
  1087. # s1 ---
  1088. # |
  1089. # |---> t1 ---| -----> t3
  1090. # | | |
  1091. # s2 --- ---> s4 ---> t4 ---> s7
  1092. # t2 ---> s5 |
  1093. # s8 ---> t5 s6 --|
  1094. data_node_1 = InMemoryDataNode("foo", Scope.SCENARIO, "s1")
  1095. data_node_2 = InMemoryDataNode("bar", Scope.SCENARIO, "s2")
  1096. data_node_4 = InMemoryDataNode("qux", Scope.SCENARIO, "s4")
  1097. data_node_5 = InMemoryDataNode("quux", Scope.SCENARIO, "s5")
  1098. data_node_6 = InMemoryDataNode("quuz", Scope.SCENARIO, "s6")
  1099. data_node_7 = InMemoryDataNode("corge", Scope.SCENARIO, "s7")
  1100. data_node_8 = InMemoryDataNode("hugh", Scope.SCENARIO, "s8")
  1101. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  1102. task_2 = Task("garply", {}, print, output=[data_node_5], id=TaskId("t2"))
  1103. task_3 = Task("waldo", {}, print, [data_node_4], None, id=TaskId("t3"))
  1104. task_4 = Task("fred", {}, print, [data_node_4, data_node_6], [data_node_7], TaskId("t4"))
  1105. task_5 = Task("bob", {}, print, [data_node_8], None, TaskId("t5"))
  1106. scenario_5 = Scenario("scenario_4", [task_1, task_2, task_3, task_4, task_5], {}, [], ScenarioId("s5"))
  1107. assert scenario_5.get_inputs() == {data_node_1, data_node_2, data_node_8, data_node_6}
  1108. assert scenario_5._get_set_of_tasks() == {task_1, task_2, task_3, task_4, task_5}
  1109. _assert_equal(scenario_5._get_sorted_tasks(), [[task_1, task_2, task_5], [task_3, task_4]])
  1110. # p1 s1 ---
  1111. # |
  1112. # |---> t1 ---| -----> t3
  1113. # | | |
  1114. # s2 --- ---> s4 ---> t4 ---> s5
  1115. # p2 t2 ---> s4 ---> t3
  1116. # p3 s6 ---> t5
  1117. data_node_1 = DataNode("foo", Scope.SCENARIO, "s1")
  1118. data_node_2 = DataNode("bar", Scope.SCENARIO, "s2")
  1119. data_node_4 = DataNode("qux", Scope.SCENARIO, "s4")
  1120. data_node_5 = DataNode("quux", Scope.SCENARIO, "s5")
  1121. data_node_6 = DataNode("quuz", Scope.SCENARIO, "s6")
  1122. task_1 = Task(
  1123. "grault",
  1124. {},
  1125. print,
  1126. [data_node_1, data_node_2],
  1127. [data_node_4],
  1128. TaskId("t1"),
  1129. )
  1130. task_2 = Task("garply", {}, print, output=[data_node_4], id=TaskId("t2"))
  1131. task_3 = Task("waldo", {}, print, [data_node_4], None, id=TaskId("t3"))
  1132. task_4 = Task("fred", {}, print, [data_node_4], [data_node_5], TaskId("t4"))
  1133. task_5 = Task("bob", {}, print, [data_node_6], None, TaskId("t5"))
  1134. scenario_6 = Scenario("quest", [task_1, task_2, task_3, task_4, task_5], {}, [], ScenarioId("s6"))
  1135. assert scenario_6.get_inputs() == {data_node_1, data_node_2, data_node_6}
  1136. assert scenario_6._get_set_of_tasks() == {task_1, task_2, task_3, task_4, task_5}
  1137. _assert_equal(scenario_6._get_sorted_tasks(), [[task_5, task_2, task_1], [task_4, task_3]])
  1138. # p1 s1 ---
  1139. # |
  1140. # |---> t1 ---| -----> t3
  1141. # | | |
  1142. # s2 --- ---> s4 ---> t4 ---> s5
  1143. # p2 t2 ---> s4 ---> t3
  1144. # p3 s6 ---> t5 ---> s4 ---> t4 ---> s5
  1145. data_node_1 = DataNode("foo", Scope.SCENARIO, "s1")
  1146. data_node_2 = DataNode("bar", Scope.SCENARIO, "s2")
  1147. data_node_4 = DataNode("qux", Scope.SCENARIO, "s4")
  1148. data_node_5 = DataNode("quux", Scope.SCENARIO, "s5")
  1149. data_node_6 = DataNode("quuz", Scope.SCENARIO, "s6")
  1150. task_1 = Task(
  1151. "grault",
  1152. {},
  1153. print,
  1154. [data_node_1, data_node_2],
  1155. [data_node_4],
  1156. TaskId("t1"),
  1157. )
  1158. task_2 = Task("garply", {}, print, output=[data_node_4], id=TaskId("t2"))
  1159. task_3 = Task("waldo", {}, print, [data_node_4], None, id=TaskId("t3"))
  1160. task_4 = Task("fred", {}, print, [data_node_4], [data_node_5], TaskId("t4"))
  1161. task_5 = Task("bob", {}, print, [data_node_6], [data_node_4], None, TaskId("t5"))
  1162. scenario_7 = Scenario("quest", [task_4, task_1, task_2, task_3, task_5], {}, [], scenario_id=ScenarioId("s7"))
  1163. assert scenario_7.get_inputs() == {data_node_1, data_node_2, data_node_6}
  1164. assert scenario_7._get_set_of_tasks() == {task_1, task_2, task_3, task_4, task_5}
  1165. _assert_equal(scenario_7._get_sorted_tasks(), [[task_5, task_2, task_1], [task_4, task_3]])
  1166. # p1 s1 ---
  1167. # |
  1168. # |---> t1 ---| -----> t3
  1169. # | | |
  1170. # s2 --- ---> s3 ---> t4 ---> s4
  1171. # p2 t2 ---> s3 ---> t3
  1172. # p3 s5 ---> t5 ---> s3 ---> t4 ---> s4
  1173. # p4 s3 ---> t4 ---> s4
  1174. data_node_1 = DataNode("foo", Scope.SCENARIO, "s1")
  1175. data_node_2 = DataNode("bar", Scope.SCENARIO, "s2")
  1176. data_node_3 = DataNode("qux", Scope.SCENARIO, "s3")
  1177. data_node_4 = DataNode("quux", Scope.SCENARIO, "s4")
  1178. data_node_5 = DataNode("quuz", Scope.SCENARIO, "s5")
  1179. task_1 = Task(
  1180. "grault",
  1181. {},
  1182. print,
  1183. [data_node_1, data_node_2],
  1184. [data_node_3],
  1185. TaskId("t1"),
  1186. )
  1187. task_2 = Task("garply", {}, print, output=[data_node_3], id=TaskId("t2"))
  1188. task_3 = Task("waldo", {}, print, [data_node_3], None, id=TaskId("t3"))
  1189. task_4 = Task("fred", {}, print, [data_node_3], [data_node_4], TaskId("t4"))
  1190. task_5 = Task("bob", {}, print, [data_node_5], [data_node_3], TaskId("t5"))
  1191. scenario_8 = Scenario("quest", [task_1, task_2, task_3, task_4, task_5], {}, [], scenario_id=ScenarioId("s8"))
  1192. assert scenario_8.get_inputs() == {data_node_1, data_node_2, data_node_5}
  1193. assert scenario_8._get_set_of_tasks() == {task_1, task_2, task_3, task_4, task_5}
  1194. _assert_equal(scenario_8._get_sorted_tasks(), [[task_5, task_2, task_1], [task_3, task_4]])
  1195. def test_check_consistency():
  1196. data_node_1 = InMemoryDataNode("foo", Scope.SCENARIO, "s1")
  1197. data_node_2 = InMemoryDataNode("bar", Scope.SCENARIO, "s2")
  1198. data_node_3 = InMemoryDataNode("bar", Scope.SCENARIO, "s3")
  1199. data_node_4 = InMemoryDataNode("qux", Scope.SCENARIO, "s4")
  1200. data_node_5 = InMemoryDataNode("quux", Scope.SCENARIO, "s5")
  1201. data_node_6 = InMemoryDataNode("quuz", Scope.SCENARIO, "s6")
  1202. data_node_7 = InMemoryDataNode("corge", Scope.SCENARIO, "s7")
  1203. data_node_8 = InMemoryDataNode("d8", Scope.SCENARIO, "s8")
  1204. data_node_9 = InMemoryDataNode("d9", Scope.SCENARIO, "s9")
  1205. scenario_0 = Scenario("scenario_0", [], {})
  1206. assert scenario_0._is_consistent()
  1207. task_1 = Task("foo", {}, print, [data_node_1], [data_node_2], TaskId("t1"))
  1208. scenario_1 = Scenario("scenario_1", [task_1], {})
  1209. assert scenario_1._is_consistent()
  1210. # s1 --- ---> s3 ---> t2 ---> s5 ----
  1211. # | | |
  1212. # |---> t1 ---| -------------------------> t3 ---> s6
  1213. # | | |
  1214. # s2 --- ---> s4 ---> t4 ---> s7
  1215. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_3, data_node_4], TaskId("t1"))
  1216. task_2 = Task("garply", {}, print, [data_node_3], [data_node_5], TaskId("t2"))
  1217. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], [data_node_6], TaskId("t3"))
  1218. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  1219. scenario_2 = Scenario("scenario_2", {task_1, task_2, task_3, task_4}, {}, [], ScenarioId("s1"))
  1220. assert scenario_2._is_consistent()
  1221. # s1 --- t2 ---> s5
  1222. # | |
  1223. # |---> t1 ---| -----> t3 ---> s6
  1224. # | | |
  1225. # s2 --- ---> s4 ---> t4 ---> s7
  1226. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  1227. task_2 = Task("garply", {}, print, None, [data_node_5], TaskId("t2"))
  1228. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], [data_node_6], TaskId("t3"))
  1229. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  1230. scenario_3 = Scenario("scenario_3", {task_1, task_2, task_3, task_4}, {}, [], ScenarioId("s2"))
  1231. assert scenario_3._is_consistent()
  1232. # s1 --- s6 ---> t2 ---> s5
  1233. # | |
  1234. # |---> t1 ---| -----> t3
  1235. # | | |
  1236. # s2 --- ---> s4 ---> t4 ---> s7
  1237. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  1238. task_2 = Task("garply", {}, print, [data_node_6], [data_node_5], TaskId("t2"))
  1239. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], id=TaskId("t3"))
  1240. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  1241. scenario_4 = Scenario("scenario_4", [task_4, task_2, task_1, task_3], {}, [], scenario_id=ScenarioId("s3"))
  1242. assert scenario_4._is_consistent()
  1243. # s1 --- s6 ---> t2 ---> s5
  1244. # | |
  1245. # |---> t1 ---| -----> t3
  1246. # | | |
  1247. # s2 --- ---> s4 ---> t4 ---> s7 ---> t6
  1248. # |
  1249. # s8 -------> t5 -------> s9 ------------------
  1250. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  1251. task_2 = Task("garply", {}, print, [data_node_6], [data_node_5], TaskId("t2"))
  1252. task_3 = Task("waldo", {}, print, [data_node_5, data_node_4], id=TaskId("t3"))
  1253. task_4 = Task("fred", {}, print, [data_node_4], [data_node_7], TaskId("t4"))
  1254. task_5 = Task("t5", {}, print, [data_node_8], [data_node_9], TaskId("t5"))
  1255. task_6 = Task("t6", {}, print, [data_node_7, data_node_9], id=TaskId("t6"))
  1256. scenario_5 = Scenario("scenario_5", [task_1, task_2, task_3, task_4, task_5, task_6], {}, [], ScenarioId("s4"))
  1257. assert scenario_5._is_consistent()
  1258. # s1 ---
  1259. # |
  1260. # |---> t1 ---| -----> t3
  1261. # | | |
  1262. # s2 --- ---> s4 ---> t4 ---> s7
  1263. # t2 ---> s5 |
  1264. # s8 ---> t5 s6 --|
  1265. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  1266. task_2 = Task("garply", {}, print, output=[data_node_5], id=TaskId("t2"))
  1267. task_3 = Task("waldo", {}, print, [data_node_4], None, id=TaskId("t3"))
  1268. task_4 = Task("fred", {}, print, [data_node_4, data_node_6], [data_node_7], TaskId("t4"))
  1269. task_5 = Task("bob", {}, print, [data_node_8], None, TaskId("t5"))
  1270. scenario_6 = Scenario("scenario_6", [task_1, task_2, task_3, task_4, task_5], {}, [], ScenarioId("s5"))
  1271. assert scenario_6._is_consistent()
  1272. # p1 s1 ---
  1273. # |
  1274. # |---> t1 ---| -----> t3
  1275. # | | |
  1276. # s2 --- ---> s4 ---> t4 ---> s5
  1277. # p2 t2 ---> s4 ---> t3
  1278. # p3 s6 ---> t5
  1279. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  1280. task_2 = Task("garply", {}, print, output=[data_node_4], id=TaskId("t2"))
  1281. task_3 = Task("waldo", {}, print, [data_node_4], None, id=TaskId("t3"))
  1282. task_4 = Task("fred", {}, print, [data_node_4], [data_node_5], TaskId("t4"))
  1283. task_5 = Task("bob", {}, print, [data_node_6], None, TaskId("t5"))
  1284. scenario_7 = Scenario("scenario_7", [task_1, task_2, task_3, task_4, task_5], {}, [], ScenarioId("s6"))
  1285. assert scenario_7._is_consistent()
  1286. # p1 s1 ---
  1287. # |
  1288. # |---> t1 ---| -----> t3
  1289. # | | |
  1290. # s2 --- ---> s4 ---> t4 ---> s5
  1291. # p2 t2 ---> s4 ---> t3
  1292. # p3 s6 ---> t5 ---> s4 ---> t4 ---> s5
  1293. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_4], TaskId("t1"))
  1294. task_2 = Task("garply", {}, print, output=[data_node_4], id=TaskId("t2"))
  1295. task_3 = Task("waldo", {}, print, [data_node_4], None, id=TaskId("t3"))
  1296. task_4 = Task("fred", {}, print, [data_node_4], [data_node_5], TaskId("t4"))
  1297. task_5 = Task("bob", {}, print, [data_node_6], [data_node_4], None, TaskId("t5"))
  1298. scenario_8 = Scenario("scenario_8", [task_4, task_1, task_2, task_3, task_5], {}, [], scenario_id=ScenarioId("s7"))
  1299. assert scenario_8._is_consistent()
  1300. # p1 s1 ---
  1301. # |
  1302. # |---> t1 ---| -----> t3
  1303. # | | |
  1304. # s2 --- ---> s3 ---> t4 ---> s4
  1305. # p2 t2 ---> s3 ---> t3
  1306. # p3 s5 ---> t5 ---> s3 ---> t4 ---> s4
  1307. # p4 s3 ---> t4 ---> s4
  1308. task_1 = Task("grault", {}, print, [data_node_1, data_node_2], [data_node_3], TaskId("t1"))
  1309. task_2 = Task("garply", {}, print, output=[data_node_3], id=TaskId("t2"))
  1310. task_3 = Task("waldo", {}, print, [data_node_3], None, id=TaskId("t3"))
  1311. task_4 = Task("fred", {}, print, [data_node_3], [data_node_4], TaskId("t4"))
  1312. task_5 = Task("bob", {}, print, [data_node_5], [data_node_3], TaskId("t5"))
  1313. scenario_9 = Scenario("scenario_9", [task_1, task_2, task_3, task_4, task_5], {}, [], scenario_id=ScenarioId("s8"))
  1314. assert scenario_9._is_consistent()