test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-01 19:39:21 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

[skip e2e] Modify L2 test cases (#19444)

Browse Source 
Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>

Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
This commit is contained in:
binbin 2022-09-26 18:04:57 +08:00 committed by GitHub
parent 52c6a2706e
commit 0f1c2ea26a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/python_client/testcases/test_collection.py
View File

@ -2248,6 +2248,8 @@ class TestLoadCollection(TestcaseBase):
2.load with a new replica number
3.release collection
4.load with a new replica
5.create index is a must because get_query_segment_info could
only return indexed and loaded segment
expected: The second time successfully loaded with a new replica number
"""
# create, insert
@ -2255,6 +2257,7 @@ class TestLoadCollection(TestcaseBase):
df = cf.gen_default_dataframe_data()
insert_res, _ = collection_w.insert(df)
assert collection_w.num_entities == ct.default_nb
collection_w.create_index(ct.default_float_vec_field_name, ct.default_index)
collection_w.load(replica_number=1)
for seg in self.utility_wrap.get_query_segment_info(collection_w.name)[0]:
assert len(seg.nodeIds) == 1

9
tests/python_client/testcases/test_partition.py
View File

@ -360,6 +360,8 @@ class TestPartitionParams(TestcaseBase):
2.load with a new replica number
3.release partition
4.load with a new replica
5.create index is a must because get_query_segment_info could
only return indexed and loaded segment
expected: The second time successfully loaded with a new replica number
"""
# create, insert
@ -368,6 +370,7 @@ class TestPartitionParams(TestcaseBase):
partition_w = self.init_partition_wrap(collection_w)
partition_w.insert(cf.gen_default_list_data())
assert partition_w.num_entities == ct.default_nb
collection_w.create_index(ct.default_float_vec_field_name, ct.default_index)
partition_w.load(replica_number=1)
collection_w.query(expr=f"{ct.default_int64_field_name} in [0]", check_task=CheckTasks.check_query_results,
@ -392,12 +395,13 @@ class TestPartitionParams(TestcaseBase):
assert num_entities == ct.default_nb
@pytest.mark.tags(CaseLabel.ClusterOnly)
@pytest.mark.skip(reason="wait to solve")
def test_partition_replicas_change_cross_partitions(self):
"""
target: test load with different replicas between partitions
method: 1.Create two partitions and insert data
2.Load two partitions with different replicas
2.Create index is a must because get_query_segment_info could
only return indexed and loaded segment
3.Load two partitions with different replicas
expected: Raise an exception
"""
# Create two partitions and insert data
@ -407,6 +411,7 @@ class TestPartitionParams(TestcaseBase):
partition_w1.insert(cf.gen_default_dataframe_data())
partition_w2.insert(cf.gen_default_dataframe_data(start=ct.default_nb))
assert collection_w.num_entities == ct.default_nb * 2
collection_w.create_index(ct.default_float_vec_field_name, ct.default_index)
# load with different replicas
partition_w1.load(replica_number=1)