rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-08-18 12:08:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(python): expect float32 distance arithmetic

Browse Source
This commit is contained in:
Gatefixer
2026-08-08 20:37:46 +00:00
parent 5b5238d4ac
commit 11160d1641
1 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/python/tests/test_query.py
+3 -9
View File
@@ -675,14 +675,8 @@ def test_distance_range(table: lancedb.table.Table):
assert res["_distance"].to_pylist() == [min_dist, max_dist]
@pytest.mark.parametrize(
("expression", "expected_type"),
[
("1 - _distance", pa.float32()),
("1.0 - _distance", pa.float64()),
],
)
def test_select_arithmetic_with_distance(table, expression, expected_type):
@pytest.mark.parametrize("expression", ["1 - _distance", "1.0 - _distance"])
def test_select_arithmetic_with_distance(table, expression):
result = (
table.search([10, 10])
.select({"similarity": expression})
@@ -690,7 +684,7 @@ def test_select_arithmetic_with_distance(table, expression, expected_type):
.to_arrow()
)
assert result.schema.field("similarity").type == expected_type
assert result.schema.field("similarity").type == pa.float32()
assert result["similarity"].to_pylist() == pytest.approx(
[1 - distance for distance in result["_distance"].to_pylist()]
)