From a5f74c4a6c1541e4e75d77f3430391aebc81defb Mon Sep 17 00:00:00 2001 From: del-zhenwu <56623710+del-zhenwu@users.noreply.github.com> Date: Mon, 28 Jun 2021 13:08:08 +0800 Subject: [PATCH] update query case (#6156) Signed-off-by: del-zhenwu --- tests/python_test/entity/test_query.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/python_test/entity/test_query.py b/tests/python_test/entity/test_query.py index 21e820769e..d313f5d553 100644 --- a/tests/python_test/entity/test_query.py +++ b/tests/python_test/entity/test_query.py @@ -154,10 +154,10 @@ class TestQueryBase: res = connect.query(collection, term_expr) logging.getLogger().info(res) assert len(res) == default_pos - for i in range(default_pos): - assert res[i][default_int_field_name] == entities[0]["values"][i] - # not support - # ut.assert_equal_vector(res[i][ut.default_float_vec_field_name], entities[-1]["values"][i]) + for _id, index in enumerate(ids): + if res[index][default_int_field_name] == entities[0]["values"][index]: + assert res[index][default_float_field_name] == entities[1]["values"][index] + # # ut.assert_equal_vector(res[i][ut.default_float_vec_field_name], entities[-1]["values"][i]) def test_query_after_search(self, connect, collection): """