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

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

@@ -73,36 +73,39 @@ models:
         super().tearDownClass()
 
     def test_invoke(self):
-        DBTCli([
+        DBTCli().invoke([
             'build',
             '--profiles-dir', self.project_dir,
             '--project-dir', self.project_dir,
             '--select', 'mage_test_model'
-        ]).invoke()
+        ])
 
-        DBTCli([
-            'clean',
-            '--project-dir', self.project_dir,
-        ]).invoke()
+        # DBTCli().invoke([
+        #     'clean',
+        #     '--project-dir', self.project_dir,
+        # ])
 
     def test_to_pandas(self):
-        DBTCli([
+        cli = DBTCli()
+
+        cli.invoke([
             'run',
             '--profiles-dir', self.project_dir,
             '--project-dir', self.project_dir,
             '--select', 'mage_test_model'
-        ]).invoke()
+        ])
 
-        df, _res, success = DBTCli([
+        res = cli.invoke([
             'show',
             '--profiles-dir', self.project_dir,
             '--project-dir', self.project_dir,
             '--select', 'mage_test_model',
             '--limit', '1'
-        ]).to_pandas()
+        ])
 
-        self.assertTrue(success)
+        self.assertTrue(res.success)
 
+        df = cli.to_pandas(res)
         self.assertEqual(
             df.to_dict(),
             {'id': {0: 1}}

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
     """