From 6ba9fb2e9a8d1720630c0c535ad8985b58f7fc45 Mon Sep 17 00:00:00 2001 From: break2017 Date: Sat, 26 Jun 2021 11:20:11 +0800 Subject: [PATCH] Disable output vec field in tests (#6133) Signed-off-by: zhenwu Co-authored-by: zhenwu --- tests/python_test/entity/test_query.py | 30 +++++++++++++++++--------- 1 file changed, 20 insertions(+), 10 deletions(-) diff --git a/tests/python_test/entity/test_query.py b/tests/python_test/entity/test_query.py index 7dd99f06be..c82acfec99 100644 --- a/tests/python_test/entity/test_query.py +++ b/tests/python_test/entity/test_query.py @@ -106,7 +106,8 @@ class TestQueryBase: for _id, index in enumerate(ids[:default_pos]): 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[index][ut.default_float_vec_field_name], entities[2]["values"][index]) + # not support + # ut.assert_equal_vector(res[index][ut.default_float_vec_field_name], entities[2]["values"][index]) def test_query_collection_not_existed(self, connect): """ @@ -156,7 +157,8 @@ class TestQueryBase: assert len(res) == default_pos for i in range(default_pos): assert res[i][default_int_field_name] == entities[0]["values"][i] - ut.assert_equal_vector(res[i][ut.default_float_vec_field_name], entities[-1]["values"][i]) + # not support + # ut.assert_equal_vector(res[i][ut.default_float_vec_field_name], entities[-1]["values"][i]) def test_query_after_search(self, connect, collection): """ @@ -180,7 +182,8 @@ class TestQueryBase: for _id, index in enumerate(ids[:default_pos]): 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[index][ut.default_float_vec_field_name], entities[2]["values"][index]) + # not support + # ut.assert_equal_vector(res[index][ut.default_float_vec_field_name], entities[2]["values"][index]) @pytest.mark.xfail(reason="#6053") def test_query_empty_collection(self, connect, collection): @@ -309,7 +312,8 @@ class TestQueryBase: assert len(res) == 1 assert res[0][default_int_field_name] == entities[0]["values"][0] assert res[0][default_float_field_name] == entities[1]["values"][0] - ut.assert_equal_vector(res[0][ut.default_float_vec_field_name], entities[2]["values"][0]) + # not support + # ut.assert_equal_vector(res[0][ut.default_float_vec_field_name], entities[2]["values"][0]) @pytest.mark.xfail(reason="#6072") def test_query_binary_expr_single_term_array(self, connect, binary_collection): @@ -326,7 +330,8 @@ class TestQueryBase: assert len(res) == 1 assert res[0][default_int_field_name] == binary_entities[0]["values"][0] assert res[1][default_float_field_name] == binary_entities[1]["values"][0] - assert res[2][ut.default_float_vec_field_name] == binary_entities[2]["values"][0] + # not support + # assert res[2][ut.default_float_vec_field_name] == binary_entities[2]["values"][0] @pytest.mark.level(2) def test_query_expr_all_term_array(self, connect, collection): @@ -344,7 +349,8 @@ class TestQueryBase: 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[index][ut.default_float_vec_field_name], entities[2]["values"][index]) + # not support + # ut.assert_equal_vector(res[index][ut.default_float_vec_field_name], entities[2]["values"][index]) def test_query_expr_repeated_term_array(self, connect, collection): """ @@ -410,7 +416,8 @@ class TestQueryBase: assert len(ids) == ut.default_nb connect.load_collection(collection) res = connect.query(collection, default_term_expr, output_fields=[]) - fields = [default_int_field_name, default_float_field_name, ut.default_float_vec_field_name] + # not support float_vector + fields = [default_int_field_name, default_float_field_name] for field in fields: assert field in res[0].keys() @@ -436,7 +443,8 @@ class TestQueryBase: entities, ids = init_data(connect, collection) assert len(ids) == ut.default_nb connect.load_collection(collection) - fields = [default_int_field_name, default_float_field_name, ut.default_float_vec_field_name] + # fields = [default_int_field_name, default_float_field_name, ut.default_float_vec_field_name] + fields = [default_int_field_name, default_float_field_name] res = connect.query(collection, default_term_expr, output_fields=fields) for field in fields: assert field in res[0].keys() @@ -496,7 +504,8 @@ class TestQueryPartition: for _id, index in enumerate(ids[:default_pos]): 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[index][ut.default_float_vec_field_name], entities[2]["values"][index]) + # not support + # ut.assert_equal_vector(res[index][ut.default_float_vec_field_name], entities[2]["values"][index]) def test_query_partition_without_loading(self, connect, collection): """ @@ -523,7 +532,8 @@ class TestQueryPartition: for _id, index in enumerate(ids[:default_pos]): 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[index][ut.default_float_vec_field_name], entities[2]["values"][index]) + # not support + # ut.assert_equal_vector(res[index][ut.default_float_vec_field_name], entities[2]["values"][index]) @pytest.mark.xfail(reason="#6075") def test_query_empty_partition(self, connect, collection):