Fix test cases (#22423)

Signed-off-by: nico <cheng.yuan@zilliz.com>
pull/22452/head
NicoYuan1986 2023-02-28 10:15:47 +08:00 committed by GitHub
parent 4cc721564f
commit fd4411a8ad
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions

View File

@ -1292,7 +1292,8 @@ class TestCompactionOperation(TestcaseBase):
replica_num = len(replicas.groups)
seg_info = self.utility_wrap.get_query_segment_info(collection_w.name)[0]
self.utility_wrap.wait_index_build_completed(collection_w.name)
assert len(seg_info) == replica_num
if not (len(seg_info) == 1 * replica_num or len(seg_info) == 2 * replica_num):
assert False
@pytest.mark.tags(CaseLabel.L2)
def test_compact_during_search(self):

View File

@ -1081,7 +1081,8 @@ class TestCollectionSearch(TestcaseBase):
# 2. rename collection
new_collection_name = cf.gen_unique_str(prefix + "new")
self.utility_wrap.rename_collection(collection_w.name, new_collection_name)
collection_w = self.init_collection_general(auto_id=auto_id, dim=dim, name=new_collection_name)[0]
collection_w = self.init_collection_general(auto_id=auto_id, is_index=True,
dim=dim, name=new_collection_name)[0]
# 3. search
log.info("test_search_normal_default_params: searching collection %s" % collection_w.name)
vectors = [[random.random() for _ in range(dim)] for _ in range(default_nq)]