Unit тест графа c подграфами проведен и отлажен

test_funcs/simplest.py

@@ -64,4 +64,27 @@ def branch_selector(data):
     condition = check_condition(data)
     return [condition, not condition]
 
+def init_data(data):
+    data['initialized'] = True
+    data['processed'] = False
+    data['value'] = data.get('input', 0)
+
+def validate_data(data):
+    if 'value' not in data:
+        raise ValueError("Data not initialized")
+    data['valid'] = True
+
+def process_data(data):
+    if not data.get('valid', False):
+        raise ValueError("Invalid data")
+    data['processed'] = True
+    data['value'] *= 2
+
+def save_result(data):
+    data['saved'] = True
+    data['result'] = data['value']
+
+def cleanup(data):
+    data['clean'] = True
+
 

tests/test_aDOT/test_adot_files/main_graph.adot

+digraph MAIN_GRAPH {
+    // Функции
+    MAIN_PROCESS [module=test_funcs.simplest, entry_func=process_data]
+
+    / Функции-предикаты
+    ALWAYS_TRUE [module=test_funcs.simplest, entry_func=true_predicate, comment = pred_ALWAYS_TRUE]
+
+    // Морфизмы
+    EDGE_MAIN [predicate=ALWAYS_TRUE, function=MAIN_PROCESS, comment = edge_1]
+
+        // Главный граф
+    PREPROCESS [subgraph=./test_adot_files/preprocess.adot]
+    POSTPROCESS [subgraph=./test_adot_files/postprocess.adot]
+
+
+    __BEGIN__ -> PREPROCESS
+    PREPROCESS -> MAIN_PROCESS [morphism=EDGE_MAIN]
+    MAIN_PROCESS -> POSTPROCESS
+    POSTPROCESS -> __END__
+}
\ No newline at end of file

tests/test_aDOT/test_adot_files/postprocess.adot

+digraph POSTPROCESS {
+    // Функции постобработки
+    SAVE_RESULT [module=test_funcs.simplest, entry_func=save_result]
+    CLEANUP [module=test_funcs.simplest, entry_func=cleanup]
+
+    / Функции-предикаты
+    ALWAYS_TRUE [module=test_funcs.simplest, entry_func=true_predicate, comment = pred_ALWAYS_TRUE_postprocess]
+
+    // Морфизмы
+    EDGE_1 [predicate=ALWAYS_TRUE, function=SAVE_RESULT, comment = edge_1_postprocess]
+    EDGE_2 [predicate=ALWAYS_TRUE, function=CLEANUP, comment = edge_2_postprocess]
+
+    __BEGIN__ -> SAVE
+    SAVE -> FINAL [morphism=EDGE_1]
+    FINAL -> __END__ [morphism=EDGE_2]
+}
\ No newline at end of file

tests/test_aDOT/test_adot_files/preprocess.adot

+digraph PREPROCESS {
+    // Функции предобработки
+    INIT_DATA [module=test_funcs.simplest, entry_func=init_data, comment=""]
+    VALIDATE [module=test_funcs.simplest, entry_func=validate_data]
+
+    / Функции-предикаты
+    ALWAYS_TRUE [module=test_funcs.simplest, entry_func=true_predicate, comment = pred_ALWAYS_TRUE]
+
+    // Морфизмы
+    EDGE_1 [predicate=ALWAYS_TRUE, function=INIT_DATA, comment = edge_1]
+    EDGE_2 [predicate=ALWAYS_TRUE, function=VALIDATE, comment = edge_2]
+
+
+    __BEGIN__ -> INIT
+    INIT -> VALIDATE [morphism=EDGE_1]
+    VALIDATE -> __END__ [morphism=EDGE_2]
+}
\ No newline at end of file

tests/test_aDOT/unit_test_aDOT.py

@@ -38,7 +38,7 @@ class TestADOTParser(unittest.TestCase):
                          "Граф должен уменьшать a до 0")
 
         # Дополнительная проверка с другим начальным значением
-        data = {"a": 3}
+        data = {"a": 10}
         result = graph.run(data)
         self.assertTrue(result)
         self.assertEqual(data["a"], 0)
@@ -62,14 +62,21 @@ class TestADOTParser(unittest.TestCase):
 
     def test_subgraph_integration(self):
         """Тест интеграции подграфов"""
-        graph = self.parser.parse_file(os.path.join(self.test_files_dir, "subgraph.adot"))
+        graph = self.parser.parse_file(os.path.join(self.test_files_dir, "main_graph.adot"))
         self.assertIsInstance(graph, Graph)
 
-        data = {}
+        data = {'input': 5}
         result = graph.run(data)
+        print(data)
         self.assertTrue(result)
-        self.assertTrue(data["preprocessed"])
-        self.assertTrue(data["postprocessed"])
+        self.assertEqual(data['value'], 10)  # 5 * 2 = 10
+        self.assertTrue(data['initialized'])
+        self.assertTrue(data['valid'])
+        self.assertTrue(data['processed'])
+        self.assertTrue(data['saved'])
+        self.assertTrue(data['clean'])
+        self.assertEqual(data['result'], 10)
+
 
     def test_edge_types(self):
         """Тест разных типов ребер (-> и =>)"""