From 57e9f9ad3e690374ba30a93b00d9d2f9ea73e879 Mon Sep 17 00:00:00 2001
From: jingkl <34296482+jingkl@users.noreply.github.com>
Date: Sat, 7 May 2022 15:47:51 +0800
Subject: [PATCH] [test]Modify the common func (#16828)
Signed-off-by: jingkl <jingjing.jia@zilliz.com>
---
tests/python_client/common/common_func.py | 4 ++--
tests/python_client/testcases/test_delete.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/tests/python_client/common/common_func.py b/tests/python_client/common/common_func.py
index 7573ecf4a3..39b56d2b35 100644
--- a/tests/python_client/common/common_func.py
+++ b/tests/python_client/common/common_func.py
@@ -254,9 +254,9 @@ def gen_dataframe_all_data_type(nb=ct.default_nb, dim=ct.default_dim, start=0):
int16_values = pd.Series(data=[np.int16(i) for i in range(start, start + nb)], dtype="int16")
int8_values = pd.Series(data=[np.int8(i) for i in range(start, start + nb)], dtype="int8")
bool_values = pd.Series(data=[np.bool(i) for i in range(start, start + nb)], dtype="bool")
- string_values = pd.Series(data=[str(i) for i in range(start, start + nb)], dtype="string")
float_values = pd.Series(data=[np.float32(i) for i in range(start, start + nb)], dtype="float32")
double_values = pd.Series(data=[np.double(i) for i in range(start, start + nb)], dtype="double")
+ string_values = pd.Series(data=[str(i) for i in range(start, start + nb)], dtype="string")
float_vec_values = gen_vectors(nb, dim)
df = pd.DataFrame({
ct.default_int64_field_name: int64_values,
@@ -265,8 +265,8 @@ def gen_dataframe_all_data_type(nb=ct.default_nb, dim=ct.default_dim, start=0):
ct.default_int8_field_name: int8_values,
ct.default_bool_field_name: bool_values,
ct.default_float_field_name: float_values,
- ct.default_string_field_name: string_values,
ct.default_double_field_name: double_values,
+ ct.default_string_field_name: string_values,
ct.default_float_vec_field_name: float_vec_values
})
return df
diff --git a/tests/python_client/testcases/test_delete.py b/tests/python_client/testcases/test_delete.py
index 6da834220f..4a65570021 100644
--- a/tests/python_client/testcases/test_delete.py
+++ b/tests/python_client/testcases/test_delete.py
@@ -895,7 +895,7 @@ class TestDeleteOperation(TestcaseBase):
log.debug(collection_w.num_entities)
collection_w.query(tmp_expr, output_fields=[ct.default_float_vec_field_name],
check_task=CheckTasks.check_query_results,
- check_items={'exp_res': df_new.iloc[[0], [0, 2]].to_dict('records'), 'with_vec': True})
+ check_items={'exp_res': df_new.iloc[[0], [0, 3]].to_dict('records'), 'with_vec': True})
collection_w.delete(tmp_expr)
if to_flush_delete: