mirror of
https://github.com/lancedb/lancedb.git
synced 2026-08-18 12:08:35 +00:00
Xuanwo
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from validate_stable_lance import validate
|
|
|
|
|
|
class ValidateStableLanceTest(unittest.TestCase):
|
|
def write_fixture(
|
|
self,
|
|
root: Path,
|
|
*,
|
|
rust: str = "=10.0.0",
|
|
python: str = "10.0.0",
|
|
java: str = "10.0.0",
|
|
) -> None:
|
|
(root / "python").mkdir()
|
|
(root / "java").mkdir()
|
|
(root / "Cargo.toml").write_text(
|
|
f'[workspace.dependencies]\nlance = "{rust}"\nlance-core = "{rust}"\n'
|
|
)
|
|
(root / "python" / "pyproject.toml").write_text(
|
|
'[project.optional-dependencies]\ntests = ["pylance==' + python + '"]\n'
|
|
)
|
|
(root / "java" / "pom.xml").write_text(
|
|
"<project><properties><lance-core.version>"
|
|
+ java
|
|
+ "</lance-core.version></properties></project>"
|
|
)
|
|
|
|
def test_accepts_matching_stable_versions(self) -> None:
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
root = Path(temp_dir)
|
|
self.write_fixture(root)
|
|
self.assertEqual(validate(root), "10.0.0")
|
|
|
|
def test_rejects_prerelease(self) -> None:
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
root = Path(temp_dir)
|
|
self.write_fixture(root, python="10.0.0rc1")
|
|
with self.assertRaisesRegex(ValueError, "not stable"):
|
|
validate(root)
|
|
|
|
def test_rejects_sdk_mismatch(self) -> None:
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
root = Path(temp_dir)
|
|
self.write_fixture(root, java="9.0.0")
|
|
with self.assertRaisesRegex(ValueError, "do not match across SDKs"):
|
|
validate(root)
|
|
|
|
def test_rejects_non_exact_rust_dependency(self) -> None:
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
root = Path(temp_dir)
|
|
self.write_fixture(root, rust="10.0.0")
|
|
with self.assertRaisesRegex(ValueError, "not exact"):
|
|
validate(root)
|
|
|
|
def test_rejects_unpublished_rust_source(self) -> None:
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
root = Path(temp_dir)
|
|
self.write_fixture(root)
|
|
(root / "Cargo.toml").write_text(
|
|
"[workspace.dependencies]\n"
|
|
'lance = { version = "=10.0.0", git = "https://example.com/lance", '
|
|
'tag = "v10.0.0" }\n'
|
|
)
|
|
with self.assertRaisesRegex(ValueError, "unpublished source fields"):
|
|
validate(root)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|