mirror of
https://github.com/lllin000/PaperForge.git
synced 2026-07-22 06:50:53 +00:00
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
|
|
class TestLanceBackendCapabilities:
|
|
"""LanceDB backend capability reporting (optional — skips when lancedb missing)."""
|
|
|
|
def test_lance_backend_advertises_file_based_capabilities(self):
|
|
pytest.importorskip("lancedb")
|
|
from paperforge.embedding.backends.lance_backend import LanceBackend
|
|
|
|
backend = LanceBackend("/tmp/lance")
|
|
caps = backend.capabilities()
|
|
assert caps["backend"] == "lancedb"
|
|
assert caps["supports_hybrid"] is True
|
|
assert caps["supports_multimodal"] is True
|
|
|
|
|
|
class TestAvailableBackends:
|
|
"""Status-level backend enumeration (always available, no lancedb dependency)."""
|
|
|
|
def test_get_available_backends_lists_chroma_and_lance(self):
|
|
from paperforge.embedding.status import get_available_backends
|
|
|
|
backends = get_available_backends()
|
|
|
|
assert "chroma" in backends
|
|
assert backends["chroma"]["installed"] is True
|
|
assert backends["chroma"]["selected"] is True
|
|
|
|
assert "lancedb" in backends
|
|
assert backends["lancedb"]["installed"] is False
|
|
assert backends["lancedb"]["selected"] is False
|
|
assert backends["lancedb"]["supports_hybrid"] is True
|
|
assert backends["lancedb"]["supports_multimodal"] is True
|
|
|
|
def test_get_available_backends_returns_dict_with_expected_keys(self):
|
|
from paperforge.embedding.status import get_available_backends
|
|
|
|
backends = get_available_backends()
|
|
for name, info in backends.items():
|
|
for key in ("installed", "selected", "supports_hybrid", "supports_multimodal"):
|
|
assert key in info, f"{name} missing {key}"
|