mage-ai 0.9.73__py3-none-any.whl → 0.9.75__py3-none-any.whl

mage_ai/tests/data_cleaner/transformer_actions/test_trim_transformer.py

@@ -0,0 +1,161 @@
+import unittest
+
+import pandas as pd
+
+from mage_ai.data_cleaner.transformer_actions.base import BaseAction
+from mage_ai.data_cleaner.transformer_actions.constants import ActionType, Axis
+from mage_ai.data_cleaner.transformer_actions.utils import build_transformer_action
+
+
+class TestTrimTransformer(unittest.TestCase):
+    def test_basic_trimming(self):
+        df = pd.DataFrame({
+            'category': ['  A  ', '  B  ', '  C  ', '  D  ']
+        })
+
+        expected_output = pd.DataFrame({
+            'category': ['A', 'B', 'C', 'D']
+        })
+
+        action = build_transformer_action(
+            df,
+            action_type=ActionType.REFORMAT,
+            arguments=['category'],
+            axis=Axis.COLUMN,
+            options={'reformat': 'trim'}
+        )
+
+        output = BaseAction(action).execute(df)
+
+        pd.testing.assert_frame_equal(output, expected_output)
+        print("Test 1 (Basic Trimming): Passed")
+
+    def test_no_trimming_needed(self):
+        df = pd.DataFrame({
+            'category': ['A', 'B', 'C', 'D']
+        })
+
+        expected_output = df.copy()
+
+        action = build_transformer_action(
+            df,
+            action_type=ActionType.REFORMAT,
+            arguments=['category'],
+            axis=Axis.COLUMN,
+            options={'reformat': 'trim'}
+        )
+
+        output = BaseAction(action).execute(df)
+
+        pd.testing.assert_frame_equal(output, expected_output)
+        print("Test 2 (No Trimming Needed): Passed")
+
+    def test_mixed_content(self):
+        df = pd.DataFrame({
+            'category': ['  A  ', 'B', '  C', 'D  ']
+        })
+
+        expected_output = pd.DataFrame({
+            'category': ['A', 'B', 'C', 'D']
+        })
+
+        action = build_transformer_action(
+            df,
+            action_type=ActionType.REFORMAT,
+            arguments=['category'],
+            axis=Axis.COLUMN,
+            options={'reformat': 'trim'}
+        )
+
+        output = BaseAction(action).execute(df)
+
+        pd.testing.assert_frame_equal(output, expected_output)
+        print("Test 3 (Mixed Content): Passed")
+
+    def test_multiple_columns(self):
+        df = pd.DataFrame({
+            'category': ['  A  ', '  B  ', '  C  ', '  D  '],
+            'subcategory': ['  X  ', 'Y  ', '  Z', ' W ']
+        })
+
+        expected_output = pd.DataFrame({
+            'category': ['A', 'B', 'C', 'D'],
+            'subcategory': ['X', 'Y', 'Z', 'W']
+        })
+
+        action = build_transformer_action(
+            df,
+            action_type=ActionType.REFORMAT,
+            arguments=['category', 'subcategory'],
+            axis=Axis.COLUMN,
+            options={'reformat': 'trim'}
+        )
+
+        output = BaseAction(action).execute(df)
+
+        pd.testing.assert_frame_equal(output, expected_output)
+        print("Test 4 (Multiple Columns): Passed")
+
+    def test_empty_strings(self):
+        df = pd.DataFrame({
+            'category': ['  ', '  B  ', '  ', '  D  ']
+        })
+
+        expected_output = pd.DataFrame({
+            'category': ['', 'B', '', 'D']
+        })
+
+        action = build_transformer_action(
+            df,
+            action_type=ActionType.REFORMAT,
+            arguments=['category'],
+            axis=Axis.COLUMN,
+            options={'reformat': 'trim'}
+        )
+
+        output = BaseAction(action).execute(df)
+
+        pd.testing.assert_frame_equal(output, expected_output)
+        print("Test 5 (Empty Strings): Passed")
+
+    def test_no_columns_to_trim(self):
+        df = pd.DataFrame({
+            'category': ['  A  ', '  B  ', '  C  ', '  D  ']
+        })
+
+        expected_output = df.copy()
+
+        action = build_transformer_action(
+            df,
+            action_type=ActionType.REFORMAT,
+            arguments=[],
+            axis=Axis.COLUMN,
+            options={'reformat': 'trim'}
+        )
+
+        output = BaseAction(action).execute(df)
+
+        pd.testing.assert_frame_equal(output, expected_output)
+        print("Test 6 (No Columns to Trim): Passed")
+
+    def test_column_not_found(self):
+        df = pd.DataFrame({
+            'category': ['  A  ', '  B  ', '  C  ', '  D  ']
+        })
+
+        with self.assertRaises(KeyError):
+            action = build_transformer_action(
+                df,
+                action_type=ActionType.REFORMAT,
+                arguments=['nonexistent_column'],
+                axis=Axis.COLUMN,
+                options={'reformat': 'trim'}
+            )
+
+            BaseAction(action).execute(df)
+
+        print("Test 7 (Column Not Found): Passed")
+
+
+if __name__ == '__main__':
+    unittest.main()

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

@@ -85,7 +85,7 @@ target: dev
             lambda: MockUUID(),
         ):
             with Profiles(self.repo_path, self.variables) as profiles:
-                self.assertNotEquals(
+                self.assertNotEqual(
                     profiles.profiles_dir,
                     self.repo_path
                 )

mage_ai/tests/data_preparation/models/block/dynamic/test_combos.py

@@ -110,7 +110,7 @@ class DynamicBlockCombinationTest(BaseApiTestCase):
             content = inspect.getsource(func)
 
         if content:
-            block.update_content(f'@{block.type}\n{content}')
+            block.update_content(block.get_typed_content(content))
 
         return block
 

mage_ai/tests/data_preparation/models/block/hook/test_hook_block.py

@@ -5,6 +5,7 @@ from faker import Faker
 from mage_ai.data_preparation.models.block.hook.block import HookBlock
 from mage_ai.data_preparation.models.constants import BlockType
 from mage_ai.data_preparation.models.global_hooks.models import HookStatus, HookStrategy
+from mage_ai.shared.constants import ENV_TEST_MAGE
 from mage_ai.tests.api.operations.test_base import BaseApiTestCase
 from mage_ai.tests.factory import create_pipeline_with_blocks
 from mage_ai.tests.shared.mixins import build_hooks
@@ -45,7 +46,7 @@ class HookBlockTest(BaseApiTestCase):
                 check_status=True,
                 configuration={},
                 context={},
-                env='test',
+                env=ENV_TEST_MAGE,
                 error_on_failure=True,
                 mage=1,
                 pipeline_uuid='',
@@ -84,7 +85,7 @@ class HookBlockTest(BaseApiTestCase):
                 check_status=True,
                 configuration={},
                 context={},
-                env='test',
+                env=ENV_TEST_MAGE,
                 error_on_failure=True,
                 mage=1,
                 pipeline_uuid='',

mage_ai/tests/data_preparation/models/test_block.py

@@ -746,6 +746,13 @@ def test_output(output, *args) -> None:
             ),
         )
 
+    def test_none_block_type(self):
+        block = Block(name='test block', uuid='test uuid', block_type=None)
+        self.assertEqual(
+            block.get_typed_content('test content'),
+            '@None\ntest content',
+        )
+
 
 class BlockProjectPlatformTests(ProjectPlatformMixin):
     def test_file_path(self):

mage_ai/tests/data_preparation/models/test_blocks_helper.py

@@ -107,7 +107,7 @@ class BlockHelperTest(BaseApiTestCase):
                 '\n'.join([
                     'from mage_ai.data.models.generator import DataGenerator',
                     'from mage_ai.data.tabular.mocks import create_dataframe',
-                    f'@{block.type}\n{content}',
+                    block.get_typed_content(content),
                 ])
             )
 

mage_ai/tests/data_preparation/models/variables/test_summarizer.py

@@ -110,7 +110,7 @@ class VariableSummarizerTest(BaseApiTestCase):
                 '\n'.join([
                     'from mage_ai.data.models.generator import DataGenerator',
                     'from mage_ai.data.tabular.mocks import create_dataframe',
-                    f'@{block.type}\n{content}',
+                    block.get_typed_content(content),
                 ])
             )
 

mage_ai/tests/data_preparation/sync/test_git_sync.py

@@ -22,8 +22,8 @@ class GitSyncTest(DBTestCase):
 
         git_sync.sync_data()
 
-        git_manager_instance_mock.reset_hard.called_once_with(branch='dev')
-        git_manager_instance_mock.submodules_update.not_called()
+        git_manager_instance_mock.reset_hard.assert_called_once_with(branch='dev')
+        git_manager_instance_mock.submodules_update.assert_not_called()
 
         with open(os.path.join(self.repo_path, '.preferences.yaml'), 'r') as f:
             preferences = yaml.safe_load(f)
@@ -42,8 +42,8 @@ class GitSyncTest(DBTestCase):
 
         git_sync.sync_data()
 
-        git_manager_instance_mock.reset_hard.called_once_with(branch='dev')
-        git_manager_instance_mock.submodules_update.called_once()
+        git_manager_instance_mock.reset_hard.assert_called_once_with(branch='dev')
+        git_manager_instance_mock.submodules_update.assert_called_once()
 
         with open(os.path.join(self.repo_path, '.preferences.yaml'), 'r') as f:
             preferences = yaml.safe_load(f)
@@ -63,7 +63,7 @@ class GitSyncTest(DBTestCase):
 
         git_sync.sync_data()
 
-        git_manager_instance_mock.reset_hard.called_once_with(branch='dev')
+        git_manager_instance_mock.reset_hard.assert_called_once_with(branch='dev')
 
         self.assertFalse(os.path.exists(preferences_file_path))
 
@@ -77,4 +77,4 @@ class GitSyncTest(DBTestCase):
 
         git_sync.reset()
 
-        git_manager_instance_mock.clone.called_once_with(sync_submodules=False)
+        git_manager_instance_mock.clone.assert_called_once_with(sync_submodules=False)

mage_ai/tests/data_preparation/test_repo_manager.py

@@ -10,13 +10,10 @@ from mage_ai.data_preparation.repo_manager import (
     get_project_uuid,
     get_repo_config,
     init_project_uuid,
-    init_repo,
     set_project_uuid_from_metadata,
 )
 from mage_ai.settings.repo import MAGE_DATA_DIR_ENV_VAR
-from mage_ai.settings.utils import base_repo_path
 from mage_ai.tests.base_test import DBTestCase
-from mage_ai.tests.shared.mixins import ProjectPlatformMixin
 
 
 def mock_uuid_value():
@@ -134,63 +131,3 @@ class RepoManagerTest(DBTestCase):
         self.assertEqual(get_project_uuid(), '000000')
 
         os.remove(metadata_path)
-
-
-@patch('mage_ai.settings.platform.project_platform_activated', lambda: True)
-@patch('mage_ai.settings.repo.project_platform_activated', lambda: True)
-class RepoManagerProjectPlatformTest(ProjectPlatformMixin):
-    def test_init(self):
-        repo = RepoConfig(root_project=False)
-        self.assertFalse(repo.root_project)
-        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
-        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
-
-        repo = RepoConfig(root_project=True)
-        self.assertTrue(repo.root_project)
-        self.assertEqual(repo.repo_path, base_repo_path())
-        self.assertEqual(repo.variables_dir, base_repo_path())
-
-    def test_from_dict(self):
-        repo = RepoConfig.from_dict(dict(), root_project=False)
-        self.assertFalse(repo.root_project)
-        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
-        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
-
-        repo = RepoConfig.from_dict(dict(), root_project=True)
-        self.assertTrue(repo.root_project)
-        self.assertEqual(repo.repo_path, base_repo_path())
-        self.assertEqual(repo.variables_dir, base_repo_path())
-
-    def test_metadata_path(self):
-        repo = RepoConfig(root_project=False)
-        self.assertEqual(
-            repo.metadata_path, os.path.join(base_repo_path(), 'mage_platform/metadata.yaml'),
-        )
-
-        repo = RepoConfig(root_project=True)
-        self.assertEqual(repo.metadata_path, os.path.join(base_repo_path(), 'metadata.yaml'))
-
-    def test_get_repo_config(self):
-        repo = get_repo_config(root_project=False)
-        self.assertFalse(repo.root_project)
-        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
-        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
-
-        repo = get_repo_config(root_project=True)
-        self.assertTrue(repo.root_project)
-        self.assertEqual(repo.repo_path, base_repo_path())
-        self.assertEqual(repo.variables_dir, base_repo_path())
-
-    def test_init_repo(self):
-        path = os.path.join(os.path.dirname(base_repo_path()), 'test2')
-        try:
-            shutil.rmtree(path)
-        except Exception:
-            pass
-        with patch('mage_ai.data_preparation.repo_manager.get_repo_config') as mock_get_repo_config:
-            init_repo(path, root_project=True)
-            mock_get_repo_config.assert_called_once_with(path, root_project=True)
-        try:
-            shutil.rmtree(path)
-        except Exception:
-            pass

mage_ai/tests/data_preparation/test_repo_manager_project_platform.py

@@ -0,0 +1,67 @@
+import os
+import shutil
+from unittest.mock import patch
+
+from mage_ai.data_preparation.repo_manager import RepoConfig, get_repo_config, init_repo
+from mage_ai.settings.utils import base_repo_path
+from mage_ai.tests.shared.mixins import ProjectPlatformMixin
+
+
+@patch('mage_ai.settings.platform.project_platform_activated', lambda: True)
+@patch('mage_ai.settings.repo.project_platform_activated', lambda: True)
+class RepoManagerProjectPlatformTest(ProjectPlatformMixin):
+    def test_init(self):
+        repo = RepoConfig(root_project=False)
+        self.assertFalse(repo.root_project)
+        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
+        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
+
+        repo = RepoConfig(root_project=True)
+        self.assertTrue(repo.root_project)
+        self.assertEqual(repo.repo_path, base_repo_path())
+        self.assertEqual(repo.variables_dir, base_repo_path())
+
+    def test_from_dict(self):
+        repo = RepoConfig.from_dict(dict(), root_project=False)
+        self.assertFalse(repo.root_project)
+        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
+        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
+
+        repo = RepoConfig.from_dict(dict(), root_project=True)
+        self.assertTrue(repo.root_project)
+        self.assertEqual(repo.repo_path, base_repo_path())
+        self.assertEqual(repo.variables_dir, base_repo_path())
+
+    def test_metadata_path(self):
+        repo = RepoConfig(root_project=False)
+        self.assertEqual(
+            repo.metadata_path, os.path.join(base_repo_path(), 'mage_platform/metadata.yaml'),
+        )
+
+        repo = RepoConfig(root_project=True)
+        self.assertEqual(repo.metadata_path, os.path.join(base_repo_path(), 'metadata.yaml'))
+
+    def test_get_repo_config(self):
+        repo = get_repo_config(root_project=False)
+        self.assertFalse(repo.root_project)
+        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
+        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
+
+        repo = get_repo_config(root_project=True)
+        self.assertTrue(repo.root_project)
+        self.assertEqual(repo.repo_path, base_repo_path())
+        self.assertEqual(repo.variables_dir, base_repo_path())
+
+    def test_init_repo(self):
+        path = os.path.join(os.path.dirname(base_repo_path()), 'test2')
+        try:
+            shutil.rmtree(path)
+        except Exception:
+            pass
+        with patch('mage_ai.data_preparation.repo_manager.get_repo_config') as mock_get_repo_config:
+            init_repo(path, root_project=True)
+            mock_get_repo_config.assert_called_once_with(path, root_project=True)
+        try:
+            shutil.rmtree(path)
+        except Exception:
+            pass

mage_ai/tests/data_preparation/test_variable_manager.py

@@ -1,5 +1,3 @@
-from unittest.mock import patch
-
 import pandas as pd
 from pandas.testing import assert_frame_equal
 
@@ -10,12 +8,10 @@ from mage_ai.data_preparation.repo_manager import get_repo_config
 from mage_ai.data_preparation.variable_manager import (
     VariableManager,
     get_global_variable,
-    get_global_variables,
     set_global_variable,
 )
 from mage_ai.settings.repo import set_repo_path
 from mage_ai.tests.base_test import DBTestCase
-from mage_ai.tests.shared.mixins import ProjectPlatformMixin
 
 
 class VariableManagerTest(DBTestCase):
@@ -128,50 +124,3 @@ class VariableManagerTest(DBTestCase):
     #         get_global_variables(None, pipeline=pipeline), pipeline.variables
     #     )
     #     self.assertEqual(get_global_variables(pipeline.uuid), pipeline.variables)
-
-
-class VariableManagerProjectPlatformTests(ProjectPlatformMixin):
-    def test_get_global_variable(self):
-        with patch(
-            "mage_ai.data_preparation.variable_manager.project_platform_activated",
-            lambda: True,
-        ):
-            with patch(
-                "mage_ai.data_preparation.models.pipeline.project_platform_activated",
-                lambda: True,
-            ):
-                for settings in self.repo_paths.values():
-                    pipeline = Pipeline.create(
-                        self.faker.unique.name(),
-                        repo_path=settings["full_path"],
-                    )
-                    value = self.faker.unique.name()
-                    pipeline.variables = dict(mage=value)
-                    pipeline.save()
-
-                    self.assertEqual(get_global_variable(pipeline.uuid, "mage"), value)
-
-    def test_get_global_variables(self):
-        with patch(
-            "mage_ai.data_preparation.variable_manager.project_platform_activated",
-            lambda: True,
-        ):
-            with patch(
-                "mage_ai.data_preparation.models.pipeline.project_platform_activated",
-                lambda: True,
-            ):
-                for settings in self.repo_paths.values():
-                    pipeline = Pipeline.create(
-                        self.faker.unique.name(),
-                        repo_path=settings["full_path"],
-                    )
-                    pipeline.variables = self.faker.unique.name()
-                    pipeline.save()
-
-                    self.assertEqual(
-                        get_global_variables(None, pipeline=pipeline),
-                        pipeline.variables,
-                    )
-                    self.assertEqual(
-                        get_global_variables(pipeline.uuid), pipeline.variables
-                    )

mage_ai/tests/data_preparation/test_variable_manager_project_platform.py

@@ -0,0 +1,41 @@
+from unittest.mock import patch
+
+from mage_ai.data_preparation.models.pipeline import Pipeline
+from mage_ai.data_preparation.variable_manager import (
+    get_global_variable,
+    get_global_variables,
+)
+from mage_ai.tests.shared.mixins import ProjectPlatformMixin
+
+
+@patch('mage_ai.data_preparation.models.pipeline.project_platform_activated', lambda: True)
+@patch('mage_ai.data_preparation.variable_manager.project_platform_activated', lambda: True)
+class VariableManagerProjectPlatformTests(ProjectPlatformMixin):
+    def test_get_global_variable(self):
+        for settings in self.repo_paths.values():
+            pipeline = Pipeline.create(
+                self.faker.unique.name(),
+                repo_path=settings["full_path"],
+            )
+            value = self.faker.unique.name()
+            pipeline.variables = dict(mage=value)
+            pipeline.save()
+
+            self.assertEqual(get_global_variable(pipeline.uuid, "mage"), value)
+
+    def test_get_global_variables(self):
+        for settings in self.repo_paths.values():
+            pipeline = Pipeline.create(
+                self.faker.unique.name(),
+                repo_path=settings["full_path"],
+            )
+            pipeline.variables = self.faker.unique.name()
+            pipeline.save()
+
+            self.assertEqual(
+                get_global_variables(None, pipeline=pipeline),
+                pipeline.variables,
+            )
+            self.assertEqual(
+                get_global_variables(pipeline.uuid), pipeline.variables
+            )

mage_ai/tests/io/create_table/test_postgresql.py

@@ -1,7 +1,9 @@
 import pandas as pd
+import simplejson
 
 from mage_ai.io.export_utils import infer_dtypes
 from mage_ai.io.postgres import Postgres
+from mage_ai.shared.parsers import encode_complex
 from mage_ai.tests.base_test import DBTestCase
 
 
@@ -26,3 +28,28 @@ class TestTablePostgres(DBTestCase):
         )
         self.assertEqual('CREATE TABLE Test.Test ("varchar_time" text,"datetime_time" TIMESTAMP);',
                          query)
+
+    def test_clean_array_value(self):
+        test_cases = [
+            [],
+            [123],
+            [['abc', 'def']],
+            [['08:00', '12:00'], ['15:00', '20:00']],
+            [['08:00', '12:00'], []],
+        ]
+        expected = [
+            '{}',
+            '{123}',
+            '{{"abc", "def"}}',
+            '{{"08:00", "12:00"}, {"15:00", "20:00"}}',
+            '{{"08:00", "12:00"}, {}}',
+        ]
+        for val, expected_val in zip(test_cases, expected):
+            cleaned_array_value = Postgres._clean_array_value(
+                simplejson.dumps(
+                    val,
+                    default=encode_complex,
+                    ignore_nan=True,
+                )
+            )
+            self.assertEqual(cleaned_array_value, expected_val)

mage_ai/tests/orchestration/notification/test_config.py

@@ -2,10 +2,10 @@ from mage_ai.orchestration.notification.config import NotificationConfig
 from mage_ai.tests.base_test import TestCase
 from mage_ai.tests.orchestration.notification.constants import (
     EMAIL_NOTIFICATION_CONFIG,
+    GOOGLE_CHAT_NOTIFICATION_CONFIG,
+    OPSGENIE_NOTIFICATION_CONFIG,
     SLACK_NOTIFICATION_CONFIG,
     TEAMS_NOTIFICATION_CONFIG,
-    GOOGLE_CHAT_NOTIFICATION_CONFIG,
-    OPSGENIE_NOTIFICATION_CONFIG
 )
 
 
@@ -43,7 +43,7 @@ class NotificationConfigTests(TestCase):
         self.assertIsNone(config3.email_config)
         self.assertEqual(config3.slack_config.webhook_url, 'test_webhook_url')
 
-        self.assertEqual(config4.teams_config.webhook_url, 'test_webhook_url')
+        self.assertEqual(config4.teams_config.webhook_url, ['test_webhook_url'])
 
         self.assertEqual(config5.google_chat_config.webhook_url, 'test_webhook_url')
 

mage_ai/tests/orchestration/queue/test_process_queue.py

@@ -33,9 +33,10 @@ class ProcessQueueTests(TestCase):
         self.assertFalse('block_run_3' in self.queue.job_dict)
         self.assertFalse('block_run_4' in self.queue.job_dict)
 
-    @patch('mage_ai.orchestration.queue.process_queue.poll_job_and_execute')
+    @patch.object(ProcessQueue, 'start_worker_pool')
     @patch('mage_ai.orchestration.queue.process_queue.psutil.pid_exists')
-    def test_has_job(self, mock_pid_exists, mock_poll_job_and_execute):
+    def test_has_job(self, mock_pid_exists, mock_start_worker_pool):
+        mock_start_worker_pool.return_value = None
         mock_pid_exists.return_value = True
 
         self.queue.job_dict['block_run_1'] = JobStatus.QUEUED

mage_ai/tests/orchestration/test_pipeline_scheduler.py

@@ -441,6 +441,7 @@ class PipelineSchedulerTests(DBTestCase):
                 execution_partition=f'{pipeline_run.pipeline_schedule_id}',
                 event={},
                 pipeline_run_id=pipeline_run.id,
+                trigger_name='test_pipeline_2_trigger'
             ),
             dict(
                 pipeline_run_id=pipeline_run.id,

mage_ai/tests/settings/test_platform.py

@@ -205,12 +205,12 @@ class PlatformSettingsTest(ProjectPlatformMixin):
         ))
 
     def test_platform_settings_full_path(self):
-        self.assertEquals(
+        self.assertEqual(
             platform_settings_full_path(), os.path.join(base_repo_path(), 'settings.yaml'),
         )
 
     def test_local_platform_settings_full_path(self):
-        self.assertEquals(
+        self.assertEqual(
             local_platform_settings_full_path(), os.path.join(base_repo_path(), '.settings.yaml'),
         )