Skip to content

Commit f10dd34

Browse files
committed
Fix various commands to work with Redis Cluster
1 parent 22a363d commit f10dd34

File tree

3 files changed

+5
-12
lines changed

3 files changed

+5
-12
lines changed

tests/integration/test_aggregation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from redisvl.index import SearchIndex
44
from redisvl.query import HybridQuery
5-
from redisvl.query.filter import FilterExpression, Geo, GeoRadius, Num, Tag, Text
5+
from redisvl.query.filter import Geo, GeoRadius, Num, Tag, Text
66
from redisvl.redis.utils import array_to_buffer
77
from tests.conftest import skip_if_redis_version_below
88

@@ -275,7 +275,7 @@ def test_aggregate_query_with_text_filter(index):
275275
assert "medical" in result[text_field].lower()
276276

277277
filter_expression = (Text(text_field) == ("medical")) & (
278-
(Text(text_field) != ("research"))
278+
Text(text_field) != ("research")
279279
)
280280
hybrid_query = HybridQuery(
281281
text=text,

tests/integration/test_message_history.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
import pytest
44
from redis.exceptions import ConnectionError
55

6-
from redisvl.exceptions import RedisModuleVersionError
76
from redisvl.extensions.constants import ID_FIELD_NAME
87
from redisvl.extensions.message_history import MessageHistory, SemanticMessageHistory
98
from tests.conftest import skip_if_module_version_error

tests/integration/test_query.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import os
21
import uuid
32
from datetime import timedelta
43

@@ -622,9 +621,9 @@ def test_query_with_chunk_number_zero():
622621
expected_query_str = (
623622
"((@doc_base_id:{8675309} @file_id:{e9ffbac9ff6f67cc}) @chunk_number:[0 0])"
624623
)
625-
assert (
626-
str(filter_conditions) == expected_query_str
627-
), "Query with chunk_number zero is incorrect"
624+
assert str(filter_conditions) == expected_query_str, (
625+
"Query with chunk_number zero is incorrect"
626+
)
628627

629628

630629
def test_hybrid_policy_batches_mode(index, vector_query):
@@ -755,37 +754,32 @@ def test_range_query_with_filter_and_hybrid_policy(index):
755754

756755

757756
def test_query_normalize_cosine_distance(index, normalized_vector_query):
758-
759757
res = index.query(normalized_vector_query)
760758

761759
for r in res:
762760
assert 0 <= float(r["vector_distance"]) <= 1
763761

764762

765763
def test_query_cosine_distance_un_normalized(index, vector_query):
766-
767764
res = index.query(vector_query)
768765

769766
assert any(float(r["vector_distance"]) > 1 for r in res)
770767

771768

772769
def test_query_l2_distance_un_normalized(L2_index, vector_query):
773-
774770
res = L2_index.query(vector_query)
775771

776772
assert any(float(r["vector_distance"]) > 1 for r in res)
777773

778774

779775
def test_query_l2_distance_normalized(L2_index, normalized_vector_query):
780-
781776
res = L2_index.query(normalized_vector_query)
782777

783778
for r in res:
784779
assert 0 <= float(r["vector_distance"]) <= 1
785780

786781

787782
def test_range_query_normalize_cosine_distance(index, normalized_range_query):
788-
789783
res = index.query(normalized_range_query)
790784

791785
for r in res:

0 commit comments

Comments
 (0)