codestyle test

Author: mohamedzeidan2021

tests/unit/test_utils.py

@@ -1283,85 +1283,79 @@ def test_resolve_value_from_config():
 class TestLogSagemakerConfig(TestCase):
 
     def test_sensitive_info_masking(self):
-        logger = logging.getLogger('sagemaker.config')
+        logger = logging.getLogger("sagemaker.config")
         logger.setLevel(logging.DEBUG)
 
         stream_handler = logging.StreamHandler()
         logger.addHandler(stream_handler)
 
         # source value is None
-        with self.assertLogs(logger, level='DEBUG') as log:
+        with self.assertLogs(logger, level="DEBUG") as log:
             _log_sagemaker_config_single_substitution(
-                None,
-                {"apiKey": "topsecretkey"},
-                "config/path"
+                None, {"apiKey": "topsecretkey"}, "config/path"
             )
 
         self.assertIn("config value that will be used = {'apiKey': '***'}", log.output[0])
 
         # source value is None and config_value == source_value
-        with self.assertLogs(logger, level='DEBUG') as log:
+        with self.assertLogs(logger, level="DEBUG") as log:
             _log_sagemaker_config_single_substitution(
-                {"secretword": "topsecretword"},
-                {"secretword": "topsecretword"},
-                "config/path"
+                {"secretword": "topsecretword"}, {"secretword": "topsecretword"}, "config/path"
             )
 
         self.assertIn("Skipped value", log.output[0])
         self.assertIn("source value that will be used = {'secretword': '***'}", log.output[0])
         self.assertIn("config value = {'secretword': '***'}", log.output[0])
 
         # source value is not None and config_value != source_value
-        with self.assertLogs(logger, level='DEBUG') as log:
+        with self.assertLogs(logger, level="DEBUG") as log:
             _log_sagemaker_config_single_substitution(
-                {"password": "supersecretpassword"},
-                {"apiKey": "topsecretkey"},
-                "config/path"
+                {"password": "supersecretpassword"}, {"apiKey": "topsecretkey"}, "config/path"
             )
 
         self.assertIn("Skipped value", log.output[0])
         self.assertIn("source value that will be used = {'password': '***'}", log.output[0])
         self.assertIn("config value = {'apiKey': '***'}", log.output[0])
 
     def test_non_sensitive_info_masking(self):
-        logger = logging.getLogger('sagemaker.config')
+        logger = logging.getLogger("sagemaker.config")
         logger.setLevel(logging.DEBUG)
 
         stream_handler = logging.StreamHandler()
         logger.addHandler(stream_handler)
 
         # source value is None
-        with self.assertLogs(logger, level='DEBUG') as log:
+        with self.assertLogs(logger, level="DEBUG") as log:
             _log_sagemaker_config_single_substitution(
-                None,
-                {"username": "randomvalue"},
-                "config/path"
+                None, {"username": "randomvalue"}, "config/path"
             )
 
         self.assertIn("config value that will be used = {'username': 'randomvalue'}", log.output[0])
 
         # source value is not None and config_value == source_value
-        with self.assertLogs(logger, level='DEBUG') as log:
+        with self.assertLogs(logger, level="DEBUG") as log:
             _log_sagemaker_config_single_substitution(
                 {"nonsensitivevalue": "randomvalue"},
                 {"nonsensitivevalue": "randomvalue"},
-                "config/path"
+                "config/path",
             )
 
         self.assertIn("Skipped value", log.output[0])
         self.assertIn("source value that will be used = {'nonsensitivevalue': 'randomvalue'}", log.output[0])
         self.assertIn("config value = {'nonsensitivevalue': 'randomvalue'}", log.output[0])
 
         # source value is not None and config_value != source_value
-        with self.assertLogs(logger, level='DEBUG') as log:
+        with self.assertLogs(logger, level="DEBUG") as log:
             _log_sagemaker_config_single_substitution(
                 {"username": "nonsensitiveinfo"},
                 {"configvalue": "nonsensitivevalue"},
-                "config/path/non_sensitive"
+                "config/path/non_sensitive",
             )
 
         self.assertIn("Skipped value", log.output[0])
-        self.assertIn("source value that will be used = {'username': 'nonsensitiveinfo'}", log.output[0])
+        self.assertIn(
+            "source value that will be used = {'username': 'nonsensitiveinfo'}", log.output[0]
+        )
         self.assertIn("config value = {'configvalue': 'nonsensitivevalue'}", log.output[0])