|
1 | | -import os |
2 | 1 | import uuid |
3 | 2 | from datetime import timedelta |
4 | 3 |
|
@@ -622,9 +621,9 @@ def test_query_with_chunk_number_zero(): |
622 | 621 | expected_query_str = ( |
623 | 622 | "((@doc_base_id:{8675309} @file_id:{e9ffbac9ff6f67cc}) @chunk_number:[0 0])" |
624 | 623 | ) |
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 | + ) |
628 | 627 |
|
629 | 628 |
|
630 | 629 | def test_hybrid_policy_batches_mode(index, vector_query): |
@@ -755,37 +754,32 @@ def test_range_query_with_filter_and_hybrid_policy(index): |
755 | 754 |
|
756 | 755 |
|
757 | 756 | def test_query_normalize_cosine_distance(index, normalized_vector_query): |
758 | | - |
759 | 757 | res = index.query(normalized_vector_query) |
760 | 758 |
|
761 | 759 | for r in res: |
762 | 760 | assert 0 <= float(r["vector_distance"]) <= 1 |
763 | 761 |
|
764 | 762 |
|
765 | 763 | def test_query_cosine_distance_un_normalized(index, vector_query): |
766 | | - |
767 | 764 | res = index.query(vector_query) |
768 | 765 |
|
769 | 766 | assert any(float(r["vector_distance"]) > 1 for r in res) |
770 | 767 |
|
771 | 768 |
|
772 | 769 | def test_query_l2_distance_un_normalized(L2_index, vector_query): |
773 | | - |
774 | 770 | res = L2_index.query(vector_query) |
775 | 771 |
|
776 | 772 | assert any(float(r["vector_distance"]) > 1 for r in res) |
777 | 773 |
|
778 | 774 |
|
779 | 775 | def test_query_l2_distance_normalized(L2_index, normalized_vector_query): |
780 | | - |
781 | 776 | res = L2_index.query(normalized_vector_query) |
782 | 777 |
|
783 | 778 | for r in res: |
784 | 779 | assert 0 <= float(r["vector_distance"]) <= 1 |
785 | 780 |
|
786 | 781 |
|
787 | 782 | def test_range_query_normalize_cosine_distance(index, normalized_range_query): |
788 | | - |
789 | 783 | res = index.query(normalized_range_query) |
790 | 784 |
|
791 | 785 | for r in res: |
|
0 commit comments