update test cases for new growing segment index (#28167)

Signed-off-by: nico <cheng.yuan@zilliz.com>
pull/28189/head
nico 2023-11-06 10:08:16 +08:00 committed by GitHub
parent 90e2c63d9e
commit f40fb01627
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 5 deletions

View File

@ -3033,8 +3033,7 @@ class TestCollectionSearch(TestcaseBase):
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()
@ -3057,8 +3056,7 @@ class TestCollectionSearch(TestcaseBase):
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,
@ -5093,7 +5091,7 @@ class TestSearchString(TestcaseBase):
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()