Skip tests failing with vector memory

Author: fileames

integrations/oracle/tests/test_oracle_features_integration.py

@@ -70,6 +70,11 @@ def _drop_sql_index_if_exists(store: OracleDocumentStore, index_name: str) -> No
         raise
 
 
+def _skip_if_vector_memory_exhausted(exc: oracledb.DatabaseError) -> None:
+    if "ORA-51962" in str(exc):
+        pytest.skip("Oracle vector memory area is exhausted; live vector index creation is unavailable")
+
+
 @contextmanager
 def _temporary_store(
     connection_config, embedding_dim: int = 4, *, prefix: str = "HS_IT"
@@ -141,43 +146,53 @@ def test_contains_and_not_contains_filters_live(connection_config) -> None:
         assert [doc.content for doc in not_contains_results] == ["Haystack pipelines"]
 
 
-def test_hnsw_and_ivf_vector_index_creation_live(connection_config) -> None:
+def test_hnsw_vector_index_creation_live(connection_config) -> None:
     with _temporary_store(connection_config, prefix="HS_HNSW") as hnsw_store:
         hnsw_index_name = f"{hnsw_store.table_name}_HNSW"
         hnsw_store.write_documents(
             [Document(content="hnsw", embedding=[1.0, 0.0, 0.0, 0.0])],
             policy=DuplicatePolicy.NONE,
         )
         try:
-            hnsw_store.create_vector_index(
-                index_type="HNSW",
-                params={
-                    "idx_name": hnsw_index_name,
-                    "neighbors": 2,
-                    "efConstruction": 16,
-                    "accuracy": 80,
-                    "parallel": 1,
-                },
-            )
+            try:
+                hnsw_store.create_vector_index(
+                    index_type="HNSW",
+                    params={
+                        "idx_name": hnsw_index_name,
+                        "neighbors": 2,
+                        "efConstruction": 16,
+                        "accuracy": 80,
+                        "parallel": 1,
+                    },
+                )
+            except oracledb.DatabaseError as exc:
+                _skip_if_vector_memory_exhausted(exc)
+                raise
         finally:
             _drop_sql_index_if_exists(hnsw_store, hnsw_index_name)
 
+
+def test_ivf_vector_index_creation_live(connection_config) -> None:
     with _temporary_store(connection_config, prefix="HS_IVF") as ivf_store:
         ivf_index_name = f"{ivf_store.table_name}_IVF"
         ivf_store.write_documents(
             [Document(content="ivf", embedding=[1.0, 0.0, 0.0, 0.0])],
             policy=DuplicatePolicy.NONE,
         )
         try:
-            ivf_store.create_vector_index(
-                index_type="IVF",
-                params={
-                    "idx_name": ivf_index_name,
-                    "neighbor_partitions": 1,
-                    "accuracy": 90,
-                    "parallel": 1,
-                },
-            )
+            try:
+                ivf_store.create_vector_index(
+                    index_type="IVF",
+                    params={
+                        "idx_name": ivf_index_name,
+                        "neighbor_partitions": 1,
+                        "accuracy": 90,
+                        "parallel": 1,
+                    },
+                )
+            except oracledb.DatabaseError as exc:
+                _skip_if_vector_memory_exhausted(exc)
+                raise
         finally:
             _drop_sql_index_if_exists(ivf_store, ivf_index_name)
 
@@ -191,15 +206,19 @@ async def test_async_ivf_vector_index_creation_live(connection_config) -> None:
             policy=DuplicatePolicy.NONE,
         )
         try:
-            await store.create_vector_index_async(
-                index_type="IVF",
-                params={
-                    "idx_name": index_name,
-                    "neighbor_partitions": 1,
-                    "accuracy": 90,
-                    "parallel": 1,
-                },
-            )
+            try:
+                await store.create_vector_index_async(
+                    index_type="IVF",
+                    params={
+                        "idx_name": index_name,
+                        "neighbor_partitions": 1,
+                        "accuracy": 90,
+                        "parallel": 1,
+                    },
+                )
+            except oracledb.DatabaseError as exc:
+                _skip_if_vector_memory_exhausted(exc)
+                raise
         finally:
             _drop_sql_index_if_exists(store, index_name)