Apply pre-commit formatting to vector index tests

Author: Pushpendra-Garg

tests/vector_index/test_5200_async_setindex_attributes.py

@@ -155,9 +155,7 @@ def get_cred_param(cls, cred_name=None) -> dict:
         )
 
     @classmethod
-    async def create_credential(
-        cls, genai_cred=None, objstore_cred=None
-    ):
+    async def create_credential(cls, genai_cred=None, objstore_cred=None):
         genai_cred = genai_cred or cls.genai_cred
         objstore_cred = objstore_cred or cls.objstore_cred
         logger.info("Creating credentials: %s, %s", genai_cred, objstore_cred)

tests/vector_index/test_5200_setindex_attributes.py

@@ -152,9 +152,7 @@ def get_cred_param(cls, cred_name=None) -> dict:
         )
 
     @classmethod
-    def create_credential(
-        cls, genai_cred=None, objstore_cred=None
-    ):
+    def create_credential(cls, genai_cred=None, objstore_cred=None):
         genai_cred = genai_cred or cls.genai_cred
         objstore_cred = objstore_cred or cls.objstore_cred
         logger.info(f"Creating credentials: {genai_cred}, {objstore_cred}")

tests/vector_index/test_5300_getindex_attributes.py

@@ -140,9 +140,7 @@ def get_cred_param(cls, cred_name=None):
         )
 
     @classmethod
-    def create_credential(
-        cls, genai_cred=None, objstore_cred=None
-    ):
+    def create_credential(cls, genai_cred=None, objstore_cred=None):
         genai_cred = genai_cred or cls.genai_cred
         objstore_cred = objstore_cred or cls.objstore_cred
         logger.info("Creating credentials: GENAI_CRED, OBJSTORE_CRED")

tests/vector_index/test_5400_async_list_index.py

@@ -282,7 +282,9 @@ async def test_5406(self):
         logger.info(
             "Verifying multiple matches for pattern '^test_vector_index'..."
         )
-        actual_indexes = await self.collect_indexes(f"^{self.list_vec_params["list_index_prefix"]}")
+        actual_indexes = await self.collect_indexes(
+            f"^{self.list_vec_params["list_index_prefix"]}"
+        )
         actual_index_names = [index.index_name for index in actual_indexes]
         expected_index_names = [name.upper() for name in self.indexes[:5]]
         matched_names = [
@@ -300,7 +302,9 @@ async def test_5407(self):
             "Testing case-sensitive regex pattern for listing indexes..."
         )
         indexes = await self.collect_indexes("^TEST_VECTOR_INDEX?")
-        assert any(idx.index_name == self.indexes[1].upper() for idx in indexes)
+        assert any(
+            idx.index_name == self.indexes[1].upper() for idx in indexes
+        )
         logger.info("Case-sensitive pattern matched correctly.")
 
     async def test_5408(self):
@@ -310,7 +314,8 @@ async def test_5408(self):
         )
         indexes = await self.collect_indexes("^TEST")
         assert any(
-            idx.index_name.upper() == self.indexes[0].upper() for idx in indexes
+            idx.index_name.upper() == self.indexes[0].upper()
+            for idx in indexes
         )
         logger.info("Case-insensitive pattern matched correctly.")
 

tests/vector_index/test_5400_list_index.py

@@ -144,9 +144,7 @@ def get_cred_param(cls, cred_name=None) -> dict:
         )
 
     @classmethod
-    def create_credential(
-        cls, genai_cred=None, objstore_cred=None
-    ):
+    def create_credential(cls, genai_cred=None, objstore_cred=None):
         genai_cred = genai_cred or cls.genai_cred
         objstore_cred = objstore_cred or cls.objstore_cred
         logger.info("Creating credentials: %s, %s", genai_cred, objstore_cred)
@@ -274,7 +272,9 @@ def test_5406(self):
             "Verifying multiple matches for pattern '^test_vector_index'..."
         )
         actual_indexes = list(
-            self.vector_index.list(index_name_pattern=f"^{self.list_vec_params["list_index_prefix"]}")
+            self.vector_index.list(
+                index_name_pattern=f"^{self.list_vec_params["list_index_prefix"]}"
+            )
         )
         expected_count = 5
         assert (
@@ -293,7 +293,9 @@ def test_5407(self):
             "Testing case-sensitive regex pattern for listing indexes..."
         )
         indexes = self.vector_index.list("^TEST_VECTOR_INDEX?")
-        assert any(idx.index_name == self.indexes[1].upper() for idx in indexes)
+        assert any(
+            idx.index_name == self.indexes[1].upper() for idx in indexes
+        )
         logger.info("Case-sensitive pattern matched correctly.")
 
     def test_5408(self):
@@ -303,14 +305,17 @@ def test_5408(self):
         )
         indexes = self.vector_index.list("^TEST")
         assert any(
-            idx.index_name.upper() == self.indexes[0].upper() for idx in indexes
+            idx.index_name.upper() == self.indexes[0].upper()
+            for idx in indexes
         )
         logger.info("Case-insensitive pattern matched correctly.")
 
     def test_5409(self):
         """Test complex regex pattern with OR operator."""
         logger.info("Testing complex regex pattern with OR operator...")
-        indexes = self.vector_index.list(f"^({self.list_vec_params["list_index_prefix"]}|{self.list_vec_params["list_vecidx_prefix"]})")
+        indexes = self.vector_index.list(
+            f"^({self.list_vec_params["list_index_prefix"]}|{self.list_vec_params["list_vecidx_prefix"]})"
+        )
         names = [idx.index_name for idx in indexes]
         assert self.indexes[0].upper() in names
         assert self.indexes[5].upper() in names