wehub-resource-sync
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
"""Packaging metadata checks for CPU-only installs."""
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
try:
|
|
import tomllib
|
|
except ModuleNotFoundError: # pragma: no cover - fallback for Python < 3.11
|
|
tomllib = pytest.importorskip("tomli")
|
|
|
|
|
|
def _load_leann_pyproject():
|
|
pyproject_path = Path(__file__).resolve().parents[1] / "packages" / "leann" / "pyproject.toml"
|
|
return tomllib.loads(pyproject_path.read_text())
|
|
|
|
|
|
def _load_leann_core_pyproject():
|
|
pyproject_path = (
|
|
Path(__file__).resolve().parents[1] / "packages" / "leann-core" / "pyproject.toml"
|
|
)
|
|
return tomllib.loads(pyproject_path.read_text())
|
|
|
|
|
|
def test_leann_base_dependencies_include_diskann():
|
|
data = _load_leann_pyproject()
|
|
deps = data["project"].get("dependencies", [])
|
|
|
|
assert "leann-core>=0.1.0" in deps
|
|
assert "leann-backend-hnsw>=0.1.0" in deps
|
|
assert "leann-backend-diskann>=0.1.0" in deps
|
|
|
|
|
|
def test_leann_core_numpy_is_bounded_below_3():
|
|
data = _load_leann_core_pyproject()
|
|
deps = data["project"].get("dependencies", [])
|
|
|
|
assert any(dep.startswith("numpy") and ">=1.20.0" in dep and "<3" in dep for dep in deps)
|
|
|
|
|
|
def test_leann_core_cpu_extra_pins_cpu_torch():
|
|
data = _load_leann_core_pyproject()
|
|
extras = data["project"].get("optional-dependencies", {})
|
|
|
|
cpu_deps = extras.get("cpu", [])
|
|
assert cpu_deps, "cpu extra should be defined"
|
|
assert any(
|
|
dep.startswith("torch")
|
|
and "==2.2.2" in dep
|
|
and "platform_system == 'Linux'" in dep
|
|
and "python_version < '3.13'" in dep
|
|
for dep in cpu_deps
|
|
)
|
|
|
|
|
|
def test_leann_cpu_extra_defined():
|
|
data = _load_leann_pyproject()
|
|
extras = data["project"].get("optional-dependencies", {})
|
|
|
|
assert "cpu" in extras
|
|
assert "leann-core[cpu]>=0.1.0" in extras["cpu"]
|