mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-02-04 11:18:44 +08:00
Add test cases of load partition (#24513)
Signed-off-by: nico <cheng.yuan@zilliz.com>
This commit is contained in:
nico
GitHub
parent
283bddec3b
commit
19b826e94a
@ -2238,7 +2238,6 @@ class TestLoadCollection(TestcaseBase):
|
||||
default_limit, partition_names=[partition1])
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.skip(reason="issue #23499")
|
||||
def test_load_collection_after_release_partition(self):
|
||||
"""
|
||||
target: test load collection after load collection and release partition
|
||||
@ -2264,7 +2263,6 @@ class TestLoadCollection(TestcaseBase):
|
||||
collection_w.load()
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
# @pytest.mark.skip(reason="issue #23499")
|
||||
def test_load_partitions_after_release_partition(self):
|
||||
"""
|
||||
target: test load collection after release partition and load partitions
|
||||
@ -2288,7 +2286,6 @@ class TestLoadCollection(TestcaseBase):
|
||||
partition_w2.load()
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
# @pytest.mark.skip(reason="issue #23499")
|
||||
def test_load_collection_after_release_partition_collection(self):
|
||||
"""
|
||||
target: test load collection after release partition and collection
|
||||
@ -2311,7 +2308,6 @@ class TestLoadCollection(TestcaseBase):
|
||||
collection_w.load()
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.skip(reason="issue #23532")
|
||||
def test_load_partitions_after_release_partition_collection(self):
|
||||
"""
|
||||
target: test load partitions after release partition and collection
|
||||
@ -2371,7 +2367,6 @@ class TestLoadCollection(TestcaseBase):
|
||||
partition_w2.load()
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.skip(reason="issue #23499")
|
||||
def test_load_collection_after_drop_partition_and_release_another(self):
|
||||
"""
|
||||
target: test load collection after drop a partition and release another
|
||||
@ -2386,6 +2381,7 @@ class TestLoadCollection(TestcaseBase):
|
||||
partition_w1 = self.init_partition_wrap(collection_w, partition1)
|
||||
partition_w2 = self.init_partition_wrap(collection_w, partition2)
|
||||
collection_w.load()
|
||||
partition_w1.release()
|
||||
partition_w1.drop()
|
||||
partition_w2.release()
|
||||
error = {ct.err_code: 1, ct.err_msg: 'not loaded into memory'}
|
||||
@ -3344,7 +3340,6 @@ class TestLoadPartition(TestcaseBase):
|
||||
collection_w.query(default_term_expr)
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.skip(reason="issue #23534")
|
||||
def test_load_collection_after_load_release_partition(self):
|
||||
"""
|
||||
target: test load collection after load and release partition
|
||||
@ -3384,7 +3379,6 @@ class TestLoadPartition(TestcaseBase):
|
||||
collection_w.query(default_term_expr, partition_names=[partition1, partition2])
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.skip(reason="issue #23534")
|
||||
def test_load_collection_after_load_partition_release_partitions(self):
|
||||
"""
|
||||
target: test load collection after load partition and release partitions
|
||||
@ -3408,7 +3402,6 @@ class TestLoadPartition(TestcaseBase):
|
||||
collection_w.query(default_term_expr, partition_names=[partition1, partition2])
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.skip(reason="issue #23536")
|
||||
def test_load_partition_after_load_drop_partition(self):
|
||||
"""
|
||||
target: test load partition after load and drop partition
|
||||
@ -3428,7 +3421,6 @@ class TestLoadPartition(TestcaseBase):
|
||||
collection_w.query(default_term_expr, partition_names=[partition2])
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.skip(reason="issue #23937")
|
||||
def test_load_collection_after_load_drop_partition(self):
|
||||
"""
|
||||
target: test load collection after load and drop partition
|
||||
@ -3454,7 +3446,6 @@ class TestLoadPartition(TestcaseBase):
|
||||
collection_w.query(default_term_expr)
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.skip(reason="issue #23536")
|
||||
def test_release_load_partition_after_load_drop_partition(self):
|
||||
"""
|
||||
target: test release load partition after load and drop partition
|
||||
@ -3557,7 +3548,6 @@ class TestLoadPartition(TestcaseBase):
|
||||
collection_w.query(default_term_expr)
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.skip(reason="issue #23176")
|
||||
def test_release_unloaded_partition(self):
|
||||
"""
|
||||
target: test load collection after load and drop partition
|
||||
|
||||
@ -6739,8 +6739,7 @@ class TestCollectionRangeSearch(TestcaseBase):
|
||||
class TestCollectionLoadOperation(TestcaseBase):
|
||||
""" Test case of search combining load and other functions """
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24040")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_delete_load_collection_release_partition(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -6764,8 +6763,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
partition_w1.release()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.err_res,
|
||||
@ -6775,8 +6774,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #23532")
|
||||
@pytest.mark.tags(CaseLabel.L1)
|
||||
def test_delete_load_collection_release_collection(self):
|
||||
"""
|
||||
target: test delete load collection release collection
|
||||
@ -6802,8 +6800,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
partition_w2.load()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.err_res,
|
||||
@ -6813,7 +6811,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_delete_load_partition_release_collection(self):
|
||||
"""
|
||||
target: test delete load partition release collection
|
||||
@ -6848,8 +6846,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24074")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_delete_release_collection_load_partition(self):
|
||||
"""
|
||||
target: test delete load collection release collection
|
||||
@ -6875,8 +6872,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
partition_w2.load()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.err_res,
|
||||
@ -6886,7 +6883,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_delete_load_partition_drop_partition(self):
|
||||
"""
|
||||
target: test delete load partition drop partition
|
||||
@ -6922,7 +6919,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not found'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L1)
|
||||
def test_load_collection_delete_release_partition(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -6959,8 +6956,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24040")
|
||||
@pytest.mark.tags(CaseLabel.L1)
|
||||
def test_load_partition_delete_release_collection(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -6986,8 +6982,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
partition_w1.load()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.check_search_results,
|
||||
@ -6997,7 +6993,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_partition_delete_drop_partition(self):
|
||||
"""
|
||||
target: test load partition delete drop partition
|
||||
@ -7033,8 +7029,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not found'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24040")
|
||||
@pytest.mark.tags(CaseLabel.L1)
|
||||
def test_load_collection_release_partition_delete(self):
|
||||
"""
|
||||
target: test load collection release partition delete
|
||||
@ -7058,8 +7053,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
collection_w.delete(f"int64 in {delete_ids}")
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.err_res,
|
||||
@ -7069,8 +7064,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #23989")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_partition_release_collection_delete(self):
|
||||
"""
|
||||
target: test load partition release collection delete
|
||||
@ -7107,8 +7101,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 50})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #23989")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_partition_drop_partition_delete(self):
|
||||
"""
|
||||
target: test load partition drop partition delete
|
||||
@ -7145,7 +7138,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not found'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_compact_load_collection_release_partition(self):
|
||||
"""
|
||||
target: test compact load collection release partition
|
||||
@ -7172,8 +7165,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
partition_w1.release()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 300,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 300,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.err_res,
|
||||
@ -7183,8 +7176,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #23532")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_compact_load_collection_release_collection(self):
|
||||
"""
|
||||
target: test compact load collection release collection
|
||||
@ -7225,8 +7217,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue ")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_compact_load_partition_release_collection(self):
|
||||
"""
|
||||
target: test compact load partition release collection
|
||||
@ -7266,8 +7257,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24048")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_collection_compact_drop_partition(self):
|
||||
"""
|
||||
target: test load collection compact drop partition
|
||||
@ -7307,7 +7297,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not found'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_partition_compact_release_collection(self):
|
||||
"""
|
||||
target: test load partition compact release collection
|
||||
@ -7347,7 +7337,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L1)
|
||||
def test_load_collection_release_partition_compact(self):
|
||||
"""
|
||||
target: test load collection release partition compact
|
||||
@ -7374,8 +7364,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
collection_w.get_compaction_state()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 300,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 300,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.err_res,
|
||||
@ -7385,8 +7375,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24040")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_flush_load_collection_release_partition(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7409,8 +7398,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
partition_w1.release()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.err_res,
|
||||
@ -7420,8 +7409,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #23532 #24040")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_flush_load_collection_release_collection(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7446,8 +7434,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
partition_w2.load()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.err_res,
|
||||
@ -7457,7 +7445,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_flush_load_partition_release_collection(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7491,7 +7479,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_flush_load_partition_drop_partition(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7527,7 +7515,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not found'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_flush_load_collection_drop_partition(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7562,8 +7550,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not found'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24039 #24042")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_collection_flush_release_partition(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7602,7 +7589,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_partition_flush_release_collection(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7638,8 +7625,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24048")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_collection_flush_release_partition(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7674,8 +7660,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not found'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24039")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_collection_release_partition_flush(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7698,8 +7683,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
collection_w.flush()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.check_search_results,
|
||||
@ -7709,8 +7694,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #23532")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_collection_release_collection_flush(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7747,7 +7731,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L1)
|
||||
def test_load_partition_release_collection_flush(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7781,7 +7765,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
@pytest.mark.tags(CaseLabel.L1)
|
||||
def test_load_partition_drop_partition_flush(self):
|
||||
"""
|
||||
target: test delete load collection release partition
|
||||
@ -7816,8 +7800,7 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not found'})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L3)
|
||||
# @pytest.mark.xfail(reason="issue #24040")
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_release_collection_multi_times(self):
|
||||
"""
|
||||
target: test load and release multiple times
|
||||
@ -7836,8 +7819,8 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
partition_w2.load()
|
||||
# search on collection, partition1, partition2
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.err_res,
|
||||
check_items={ct.err_code: 1, ct.err_msg: 'not loaded'})
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
partition_names=[partition_w1.name],
|
||||
check_task=CheckTasks.err_res,
|
||||
@ -7846,3 +7829,68 @@ class TestCollectionLoadOperation(TestcaseBase):
|
||||
partition_names=[partition_w2.name],
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_load_collection_release_all_partitions(self):
|
||||
"""
|
||||
target: test load and release all partitions
|
||||
method: 1. create a collection and 2 partitions
|
||||
2. load collection and release all partitions
|
||||
3. search
|
||||
expected: No exception
|
||||
"""
|
||||
# init the collection
|
||||
collection_w = self.init_collection_general(prefix, True, 200, partition_num=1, is_index=False)[0]
|
||||
partition_w1, partition_w2 = collection_w.partitions
|
||||
collection_w.create_index(default_search_field, default_index_params)
|
||||
# load and release
|
||||
collection_w.load()
|
||||
partition_w1.release()
|
||||
partition_w2.release()
|
||||
# search on collection
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 0})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
@pytest.mark.xfail(reason="issue #24446")
|
||||
def test_search_load_collection_create_partition(self):
|
||||
"""
|
||||
target: test load collection and create partition and search
|
||||
method: 1. create a collection and 2 partitions
|
||||
2. load collection and create a partition
|
||||
3. search
|
||||
expected: No exception
|
||||
"""
|
||||
# init the collection
|
||||
collection_w = self.init_collection_general(prefix, True, 200, partition_num=1, is_index=False)[0]
|
||||
partition_w1, partition_w2 = collection_w.partitions
|
||||
collection_w.create_index(default_search_field, default_index_params)
|
||||
# load and release
|
||||
collection_w.load()
|
||||
partition_w3 = collection_w.create_partition("_default3")[0]
|
||||
# search on collection
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 200})
|
||||
|
||||
@pytest.mark.tags(CaseLabel.L2)
|
||||
def test_search_load_partition_create_partition(self):
|
||||
"""
|
||||
target: test load partition and create partition and search
|
||||
method: 1. create a collection and 2 partitions
|
||||
2. load partition and create a partition
|
||||
3. search
|
||||
expected: No exception
|
||||
"""
|
||||
# init the collection
|
||||
collection_w = self.init_collection_general(prefix, True, 200, partition_num=1, is_index=False)[0]
|
||||
partition_w1, partition_w2 = collection_w.partitions
|
||||
collection_w.create_index(default_search_field, default_index_params)
|
||||
# load and release
|
||||
partition_w1.load()
|
||||
partition_w3 = collection_w.create_partition("_default3")[0]
|
||||
# search on collection
|
||||
collection_w.search(vectors[:1], field_name, default_search_params, 200,
|
||||
check_task=CheckTasks.check_search_results,
|
||||
check_items={"nq": 1, "limit": 100})
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user