test: modify cases for the update of IVF_FLAT (#29069)

pr: #27179 Signed-off-by: nico <[email protected]>
milvus-io · Dec 12, 2023 · 30ec598 · 30ec598
1 parent d36bf21
commit 30ec598
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/python_client/testcases/test_search.py b/tests/python_client/testcases/test_search.py
@@ -3027,8 +3027,7 @@ def test_search_with_expression_bool(self, dim, auto_id, _async, bool_type, enab
                                                                              enable_dynamic_field=enable_dynamic_field)[0:4]
 
         # 2. create index
-        index_param = {"index_type": "IVF_FLAT",
-                       "metric_type": "COSINE", "params": {"nlist": 100}}
+        index_param = {"index_type": "FLAT", "metric_type": "COSINE", "params": {"nlist": 100}}
         collection_w.create_index("float_vector", index_param)
         collection_w.load()
 
@@ -3051,8 +3050,7 @@ def test_search_with_expression_bool(self, dim, auto_id, _async, bool_type, enab
         expression = f"{default_bool_field_name} == {bool_type}"
         log.info(
             "test_search_with_expression_bool: searching with bool expression: %s" % expression)
-        vectors = [[random.random() for _ in range(dim)]
-                   for _ in range(default_nq)]
+        vectors = [[random.random() for _ in range(dim)] for _ in range(default_nq)]
 
         search_res, _ = collection_w.search(vectors[:default_nq], default_search_field,
                                             default_search_params, nb, expression,
@@ -5087,7 +5085,7 @@ def test_search_with_different_string_expr(self, dim, expression, _async, enable
                 filter_ids.append(_id)
 
         # 2. create index
-        index_param = {"index_type": "IVF_FLAT", "metric_type": "COSINE", "params": {"nlist": 100}}
+        index_param = {"index_type": "FLAT", "metric_type": "COSINE", "params": {"nlist": 100}}
         collection_w.create_index("float_vector", index_param)
         collection_w.load()