Skip to content

Commit 876cafc

Browse files
authored
Fix search query with params tests (#2076)
1 parent c7685a0 commit 876cafc

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/test_search.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1542,7 +1542,8 @@ def test_vector_field(modclient):
15421542
modclient.hset("b", "v", "aaaabaaa")
15431543
modclient.hset("c", "v", "aaaaabaa")
15441544

1545-
q = Query("*=>[KNN 2 @v $vec]").return_field("__v_score").sort_by("__v_score", True)
1545+
query = "*=>[KNN 2 @v $vec]"
1546+
q = Query(query).return_field("__v_score").sort_by("__v_score", True).dialect(2)
15461547
res = modclient.ft().search(q, query_params={"vec": "aaaaaaaa"})
15471548

15481549
assert "a" == res.docs[0].id
@@ -1572,7 +1573,7 @@ def test_text_params(modclient):
15721573
modclient.ft().add_document("doc3", name="Carol")
15731574

15741575
params_dict = {"name1": "Alice", "name2": "Bob"}
1575-
q = Query("@name:($name1 | $name2 )")
1576+
q = Query("@name:($name1 | $name2 )").dialect(2)
15761577
res = modclient.ft().search(q, query_params=params_dict)
15771578
assert 2 == res.total
15781579
assert "doc1" == res.docs[0].id
@@ -1589,7 +1590,7 @@ def test_numeric_params(modclient):
15891590
modclient.ft().add_document("doc3", numval=103)
15901591

15911592
params_dict = {"min": 101, "max": 102}
1592-
q = Query("@numval:[$min $max]")
1593+
q = Query("@numval:[$min $max]").dialect(2)
15931594
res = modclient.ft().search(q, query_params=params_dict)
15941595

15951596
assert 2 == res.total
@@ -1607,7 +1608,7 @@ def test_geo_params(modclient):
16071608
modclient.ft().add_document("doc3", g="29.68746, 34.94882")
16081609

16091610
params_dict = {"lat": "34.95126", "lon": "29.69465", "radius": 1000, "units": "km"}
1610-
q = Query("@g:[$lon $lat $radius $units]")
1611+
q = Query("@g:[$lon $lat $radius $units]").dialect(2)
16111612
res = modclient.ft().search(q, query_params=params_dict)
16121613
assert 3 == res.total
16131614
assert "doc1" == res.docs[0].id

0 commit comments

Comments
 (0)