mirror of
https://github.com/lancedb/lancedb.git
synced 2026-10-26 05:26:14 +00:00
Preston ZhaoandPreston Zhao
13df747f18
LanceDBConnection.session read AsyncConnection.session, but AsyncConnection has no session attribute, so access raised AttributeError. Store the session passed to the constructor and return that object. Assisted by Cursor (Grok). Co-authored-by: Preston Zhao <195942578+canyang25@users.noreply.github.com>
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright The LanceDB Authors
|
|
|
|
import lancedb
|
|
|
|
|
|
def test_session_cache_configuration(tmp_path):
|
|
"""Test Session cache configuration and basic functionality."""
|
|
# Create session with small cache limits for testing
|
|
index_cache_size = 1024 * 1024 # 1MB
|
|
metadata_cache_size = 512 * 1024 # 512KB
|
|
|
|
session = lancedb.Session(
|
|
index_cache_size_bytes=index_cache_size,
|
|
metadata_cache_size_bytes=metadata_cache_size,
|
|
)
|
|
|
|
# Record initial cache state
|
|
initial_cache_size = session.size_bytes
|
|
initial_cache_items = session.approx_num_items
|
|
|
|
# Test session works with database connection
|
|
db = lancedb.connect(tmp_path, session=session)
|
|
|
|
# Create and use a table to exercise the session
|
|
data = [{"id": i, "text": f"item {i}"} for i in range(100)]
|
|
table = db.create_table("test", data)
|
|
results = list(table.to_arrow().to_pylist())
|
|
|
|
assert len(results) == 100
|
|
|
|
# Verify cache usage increased after operations
|
|
final_cache_size = session.size_bytes
|
|
final_cache_items = session.approx_num_items
|
|
|
|
assert final_cache_size > initial_cache_size # Cache should have grown
|
|
assert final_cache_items >= initial_cache_items # Items should not decrease
|
|
assert initial_cache_size < index_cache_size + metadata_cache_size
|
|
|
|
|
|
def test_connection_session_returns_supplied_session(tmp_path):
|
|
"""Sync connections must expose the session they were opened with."""
|
|
db = lancedb.connect(tmp_path)
|
|
assert db.session is None
|
|
|
|
session = lancedb.Session()
|
|
db = lancedb.connect(tmp_path, session=session)
|
|
assert db.session is session
|