mage-ai 0.9.58__py3-none-any.whl → 0.9.59__py3-none-any.whl

mage_ai/tests/data_preparation/models/block/dbt/test_dbt_cli.py

@@ -80,10 +80,10 @@ models:
             '--select', 'mage_test_model'
         ])
 
-        DBTCli().invoke([
-            'clean',
-            '--project-dir', self.project_dir,
-        ])
+        # DBTCli().invoke([
+        #     'clean',
+        #     '--project-dir', self.project_dir,
+        # ])
 
     def test_to_pandas(self):
         cli = DBTCli()

mage_ai/tests/data_preparation/models/block/dbt/test_profiles.py

@@ -5,10 +5,10 @@ from unittest.mock import patch
 import yaml
 
 from mage_ai.data_preparation.models.block.dbt.profiles import Profiles
-from mage_ai.tests.base_test import TestCase
+from mage_ai.tests.base_test import AsyncDBTestCase
 
 
-class ProfilesTest(TestCase):
+class ProfilesTest(AsyncDBTestCase):
     """
     Tests the Profiles class, which is an interface with dbt profiles.yml files
     """

mage_ai/tests/data_preparation/models/block/dbt/test_project.py

@@ -1,10 +1,10 @@
 from pathlib import Path
 
 from mage_ai.data_preparation.models.block.dbt.project import Project
-from mage_ai.tests.base_test import TestCase
+from mage_ai.tests.base_test import AsyncDBTestCase
 
 
-class ProjectTest(TestCase):
+class ProjectTest(AsyncDBTestCase):
     """
     Tests the Project class, which is an interface with dbt dbt_project.yml files
     """

mage_ai/tests/data_preparation/models/block/dbt/test_sources.py

@@ -1,10 +1,10 @@
 from pathlib import Path
 
 from mage_ai.data_preparation.models.block.dbt.sources import Sources
-from mage_ai.tests.base_test import TestCase
+from mage_ai.tests.base_test import AsyncDBTestCase
 
 
-class SourcesTest(TestCase):
+class SourcesTest(AsyncDBTestCase):
     """
     Tests the Project class, which is an interface with dbt dbt_project.yml files
     """