test(python): cover vector index prewarming by name

python/python/lancedb/__init__.py

@@ -162,15 +162,6 @@ def connect(
     ...     },
     ... )
 
-    Azure managed identity authentication requires only the storage account name.
-    LanceDB acquires and refreshes managed identity tokens automatically. For a
-    user-assigned identity, also set ``azure_storage_client_id``:
-
-    >>> db = lancedb.connect(  # doctest: +SKIP
-    ...     "az://my-container/lancedb",
-    ...     storage_options={"account_name": "my-storage-account"},
-    ... )
-
     For tests and temporary data, use an in-memory database:
 
     >>> db = lancedb.connect("memory://")
@@ -464,11 +455,6 @@ async def connect_async(
     ...     db = await lancedb.connect_async("s3://my-bucket/lancedb",
     ...                                      storage_options={
     ...                                          "aws_access_key_id": "***"})
-    ...     # Azure managed identity tokens are acquired and refreshed automatically
-    ...     db = await lancedb.connect_async(
-    ...         "az://my-container/lancedb",
-    ...         storage_options={"account_name": "my-storage-account"},
-    ...     )
     ...     # For tests and temporary data, use an in-memory database
     ...     db = await lancedb.connect_async("memory://")
     ...     # Connect to LanceDB cloud

python/python/lancedb/util.py

@@ -102,10 +102,6 @@ def fs_from_uri(uri: str) -> Tuple[pa_fs.FileSystem, str]:
         az_blob_fs = adlfs.AzureBlobFileSystem(
             account_name=os.environ.get("AZURE_STORAGE_ACCOUNT_NAME"),
             account_key=os.environ.get("AZURE_STORAGE_ACCOUNT_KEY"),
-            # Without an explicit key, authenticate with DefaultAzureCredential
-            # instead of attempting anonymous access.  In particular, this enables
-            # managed identity authentication on Azure hosts.
-            anon=False,
         )
 
         fs = pa_fs.PyFileSystem(pa_fs.FSSpecHandler(az_blob_fs))

python/python/tests/test_index.py

@@ -364,6 +364,8 @@ async def test_create_vector_index(some_table: AsyncTable):
     assert indices[0].columns == ["vector"]
     assert indices[0].name == "vector_idx"
 
+    await some_table.prewarm_index(indices[0].name)
+
     stats = await some_table.index_stats("vector_idx")
     assert stats.index_type == "IVF_PQ"
     assert stats.distance_type == "l2"

python/python/tests/test_io.py

@@ -2,7 +2,6 @@
 # SPDX-FileCopyrightText: Copyright The LanceDB Authors
 
 
-import asyncio
 import os
 
 import lancedb
@@ -13,19 +12,11 @@ import pytest
 #    AWS_PROFILE=default TEST_S3_BASE_URL=s3://my_bucket/dataset pytest tests/test_io.py
 #
 # Azure:
-# You need to set up Azure credentials and a base path to run this test. Examples:
-#
-# Account key:
+# You need to setup Azure credentials an a base path to run this test. Example
 #   export AZURE_STORAGE_ACCOUNT_NAME="<account>"
 #   export AZURE_STORAGE_ACCOUNT_KEY="<key>"
 #   export REMOTE_BASE_URL=az://my_blob/dataset
 #   pytest tests/test_io.py
-#
-# Managed identity (system-assigned or user-assigned):
-#   export AZURE_STORAGE_ACCOUNT_NAME="<account>"
-#   export AZURE_STORAGE_CLIENT_ID="<client-id>"  # user-assigned identity only
-#   export REMOTE_BASE_URL=az://my_blob/dataset
-#   pytest tests/test_io.py
 
 
 @pytest.fixture(autouse=True, scope="module")
@@ -67,28 +58,3 @@ def test_remote_io():
     assert len(db) == 1
 
     assert db.open_table("test").name == db["test"].name
-
-
-@pytest.mark.skipif(
-    (os.environ.get("REMOTE_BASE_URL") is None),
-    reason="please setup remote base url",
-)
-def test_remote_io_async():
-    async def run():
-        db = await lancedb.connect_async(os.environ["REMOTE_BASE_URL"])
-        table = await db.create_table(
-            "test_async",
-            data=[
-                {"vector": [3.1, 4.1], "item": "foo"},
-                {"vector": [5.9, 26.5], "item": "bar"},
-            ],
-        )
-
-        assert await table.count_rows() == 2
-        assert (await db.open_table("test_async")).name == "test_async"
-        assert "test_async" in await db.table_names()
-
-        await db.drop_table("test_async")
-        assert "test_async" not in await db.table_names()
-
-    asyncio.run(run())

python/python/tests/test_util.py

@@ -26,14 +26,7 @@ import pandas as pd
 import polars as pl
 import pytest
 import lancedb
-from lancedb import util
-from lancedb.util import (
-    flatten_columns,
-    fs_from_uri,
-    get_uri_scheme,
-    join_uri,
-    value_to_sql,
-)
+from lancedb.util import flatten_columns, get_uri_scheme, join_uri, value_to_sql
 from utils import exception_output
 
 
@@ -85,33 +78,6 @@ def test_normalize_uri():
         assert parsed_scheme == expected_scheme
 
 
-def test_fs_from_uri_azure_uses_default_credential(monkeypatch):
-    azure_options = {}
-    filesystem = object()
-
-    class MockAdlfs:
-        @staticmethod
-        def AzureBlobFileSystem(**kwargs):
-            azure_options.update(kwargs)
-            return object()
-
-    monkeypatch.setenv("AZURE_STORAGE_ACCOUNT_NAME", "account")
-    monkeypatch.delenv("AZURE_STORAGE_ACCOUNT_KEY", raising=False)
-    monkeypatch.setattr(util, "adlfs", MockAdlfs)
-    monkeypatch.setattr(util.pa_fs, "FSSpecHandler", lambda _: object())
-    monkeypatch.setattr(util.pa_fs, "PyFileSystem", lambda _: filesystem)
-
-    actual_filesystem, path = fs_from_uri("az://container/database")
-
-    assert actual_filesystem is filesystem
-    assert path == "container/database"
-    assert azure_options == {
-        "account_name": "account",
-        "account_key": None,
-        "anon": False,
-    }
-
-
 def test_join_uri_remote():
     schemes = ["s3", "az", "gs"]
     for scheme in schemes: