formatting

Author: kumar-shivam-ranjan

tests/unitary/with_extras/aqua/test_ui_handler.py

@@ -22,7 +22,7 @@ class TestDataset:
     USER_COMPARTMENT_ID = "ocid1.compartment.oc1..<USER_COMPARTMENT_OCID>"
     USER_PROJECT_ID = "ocid1.datascienceproject.oc1.iad.<USER_PROJECT_OCID>"
     DEPLOYMENT_SHAPE_NAME = "VM.GPU.A10.1"
-    LIMIT_NAME="ds-gpu-a10-count"
+    LIMIT_NAME = "ds-gpu-a10-count"
 
 
 class TestAquaUIHandler(unittest.TestCase):
@@ -154,15 +154,18 @@ def test_list_subnets(self, mock_list_subnets):
     def test_get_shape_availability(self, mock_get_shape_availability):
         """Test get shape availability."""
         self.ui_handler.request.path = "aqua/shapes/limit"
-        args = {"instance_shape": TestDataset.DEPLOYMENT_SHAPE_NAME,"limit_name":TestDataset.LIMIT_NAME}
+        args = {
+            "instance_shape": TestDataset.DEPLOYMENT_SHAPE_NAME,
+            "limit_name": TestDataset.LIMIT_NAME,
+        }
         self.ui_handler.get_argument = MagicMock(
             side_effect=lambda arg, default=None: args.get(arg, default)
         )
         self.ui_handler.get()
         mock_get_shape_availability.assert_called_with(
             compartment_id=TestDataset.USER_COMPARTMENT_ID,
             instance_shape=TestDataset.DEPLOYMENT_SHAPE_NAME,
-            limit_name=TestDataset.LIMIT_NAME
+            limit_name=TestDataset.LIMIT_NAME,
         )
 
     @patch("ads.aqua.ui.AquaUIApp.is_bucket_versioned")