diff --git a/mgraph_ai/providers/mermaid/test_data/__init__.py b/mgraph_ai/providers/mermaid/test_data/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/mgraph_ai/providers/mermaid/test_data/Mermaid_Examples__FlowChart.py b/mgraph_ai/providers/mermaid/utils/Mermaid_Examples__FlowChart.py similarity index 100% rename from mgraph_ai/providers/mermaid/test_data/Mermaid_Examples__FlowChart.py rename to mgraph_ai/providers/mermaid/utils/Mermaid_Examples__FlowChart.py diff --git a/mgraph_ai/providers/mermaid/test_data/Test_Data__Mermaid.py b/mgraph_ai/providers/mermaid/utils/Test_Data__Mermaid.py similarity index 97% rename from mgraph_ai/providers/mermaid/test_data/Test_Data__Mermaid.py rename to mgraph_ai/providers/mermaid/utils/Test_Data__Mermaid.py index a463dad..3954471 100644 --- a/mgraph_ai/providers/mermaid/test_data/Test_Data__Mermaid.py +++ b/mgraph_ai/providers/mermaid/utils/Test_Data__Mermaid.py @@ -1,7 +1,6 @@ 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 @@ -9,6 +8,7 @@ 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 index dde6370..441bb28 100644 --- 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 index f992b6b..0eb09fe 100644 --- 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/tests/unit/providers/mermaid/examples/test_Mermain_Examples__FlowChart.py b/tests/unit/providers/mermaid/utils/test_Mermain_Examples__FlowChart.py similarity index 98% rename from tests/unit/providers/mermaid/examples/test_Mermain_Examples__FlowChart.py rename to tests/unit/providers/mermaid/utils/test_Mermain_Examples__FlowChart.py index 9cdef79..4664726 100644 --- a/tests/unit/providers/mermaid/examples/test_Mermain_Examples__FlowChart.py +++ b/tests/unit/providers/mermaid/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