fixed imports

owasp-sbot · Jan 11, 2025 · 5ab66b8 · 5ab66b8
1 parent 8bcd4c5
commit 5ab66b8
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 6 deletions.
diff --git a/mgraph_ai/providers/mermaid/test_data/__init__.py b/mgraph_ai/providers/mermaid/test_data/__init__.py
diff --git a/.../test_data/Mermaid_Examples__FlowChart.py → ...maid/utils/Mermaid_Examples__FlowChart.py b/.../test_data/Mermaid_Examples__FlowChart.py → ...maid/utils/Mermaid_Examples__FlowChart.py
diff --git a/...s/mermaid/test_data/Test_Data__Mermaid.py → ...iders/mermaid/utils/Test_Data__Mermaid.py b/...s/mermaid/test_data/Test_Data__Mermaid.py → ...iders/mermaid/utils/Test_Data__Mermaid.py
@@ -1,14 +1,14 @@
 from mgraph_ai.providers.mermaid.domain.Mermaid__Graph                   import Mermaid__Graph
 from osbot_utils.helpers.Safe_Id                                         import Safe_Id
 from osbot_utils.helpers.Random_Guid                                     import Random_Guid
-from mgraph_ai.mgraph.domain.MGraph__Graph                               import MGraph__Graph
 from mgraph_ai.providers.mermaid.schemas.Schema__Mermaid__Node           import Schema__Mermaid__Node
 from mgraph_ai.providers.mermaid.schemas.Schema__Mermaid__Node__Config   import Schema__Mermaid__Node__Config
 from mgraph_ai.providers.mermaid.models.Model__Mermaid__Graph            import Model__Mermaid__Graph
 from mgraph_ai.providers.mermaid.schemas.Schema__Mermaid__Graph          import Schema__Mermaid__Graph
 from mgraph_ai.providers.mermaid.schemas.Schema__Mermaid__Default__Types import Schema__Mermaid__Default__Types
 from mgraph_ai.providers.mermaid.schemas.Schema__Mermaid__Graph__Config  import Schema__Mermaid__Graph__Config
 
+# todo: refactor this out, and use Mermaid__Random_Graph instead
 class Test_Data_Mermaid:
     @staticmethod
     def create_test_graph(num_nodes=3):

diff --git a/tests/unit/providers/mermaid/actions/test_Mermaid__Data.py b/tests/unit/providers/mermaid/actions/test_Mermaid__Data.py
@@ -1,7 +1,8 @@
-from unittest                                                  import TestCase
-from mgraph_ai.providers.mermaid.actions.Mermaid__Data         import Mermaid__Data
-from mgraph_ai.providers.mermaid.test_data.Test_Data__Mermaid  import Test_Data_Mermaid
+from unittest                                             import TestCase
+from mgraph_ai.providers.mermaid.actions.Mermaid__Data    import Mermaid__Data
+from mgraph_ai.providers.mermaid.utils.Test_Data__Mermaid import Test_Data_Mermaid
 
+# todo: refactor this out with Mermaid__Random_Graph
 
 class test_Mermaid__Data(TestCase):
 

diff --git a/tests/unit/providers/mermaid/domain/test_Mermaid__Node__config.py b/tests/unit/providers/mermaid/domain/test_Mermaid__Node__config.py
@@ -1,7 +1,7 @@
 from unittest                                                         import TestCase
 from mgraph_ai.providers.mermaid.domain.Mermaid__Node                 import Mermaid__Node
 from mgraph_ai.providers.mermaid.schemas.Schema__Mermaid__Node__Shape import Schema__Mermaid__Node__Shape
-from mgraph_ai.providers.mermaid.test_data.Test_Data__Mermaid         import Test_Data_Mermaid
+from mgraph_ai.providers.mermaid.utils.Test_Data__Mermaid             import Test_Data_Mermaid
 
 
 class test_Mermaid__Node__config(TestCase):     # todo: refactor these tests to the correct location since most are related to the node's .config()

diff --git a/...mples/test_Mermain_Examples__FlowChart.py → ...utils/test_Mermain_Examples__FlowChart.py b/...mples/test_Mermain_Examples__FlowChart.py → ...utils/test_Mermain_Examples__FlowChart.py
@@ -1,7 +1,7 @@
 from unittest                                                           import TestCase
 from mgraph_ai.providers.mermaid.domain.Mermaid                         import Mermaid
-from mgraph_ai.providers.mermaid.examples.Mermaid_Examples__FlowChart   import Mermain_Examples__FlowChart
 from mgraph_ai.providers.mermaid.schemas.Schema__Mermaid__Diagram__Type import Schema__Mermaid__Diagram__Type
+from mgraph_ai.providers.mermaid.utils.Mermaid_Examples__FlowChart      import Mermain_Examples__FlowChart
 from osbot_utils.utils.Str                                              import str_dedent