From 11160d1641824940910ec11753ace835c4e30d4b Mon Sep 17 00:00:00 2001 From: Gatefixer <313497061+lancedb-gatefixer[bot]@users.noreply.github.com> Date: Sat, 8 Aug 2026 20:37:46 +0000 Subject: [PATCH] test(python): expect float32 distance arithmetic --- python/python/tests/test_query.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/python/python/tests/test_query.py b/python/python/tests/test_query.py index 8e35fe4ba..8b190c07e 100644 --- a/python/python/tests/test_query.py +++ b/python/python/tests/test_query.py @@ -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()] )