Modify test case for interface change (#15881)

Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
pull/15894/head
binbin 2022-03-05 14:23:57 +08:00 committed by GitHub
parent 8a708112eb
commit d70190cf2f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 5 deletions

View File

@ -161,7 +161,7 @@ class TestInsertParams(TestcaseBase):
assert collection_w.is_empty
assert collection_w.num_entities == 0
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.tags(CaseLabel.L1)
def test_insert_numpy_data(self):
"""
target: test insert numpy.ndarray data
@ -171,8 +171,7 @@ class TestInsertParams(TestcaseBase):
c_name = cf.gen_unique_str(prefix)
collection_w = self.init_collection_wrap(name=c_name)
data = cf.gen_numpy_data(nb=10)
error = {ct.err_code: 0, ct.err_msg: "Data type not support numpy.ndarray"}
collection_w.insert(data=data, check_task=CheckTasks.err_res, check_items=error)
collection_w.insert(data=data)
@pytest.mark.tags(CaseLabel.L1)
def test_insert_binary_dataframe(self):

View File

@ -18,6 +18,7 @@ default_dim = ct.default_dim
default_nb = ct.default_nb
num_loaded_entities = "num_loaded_entities"
num_total_entities = "num_total_entities"
loading_progress = "loading_progress"
class TestUtilityParams(TestcaseBase):
@ -868,7 +869,7 @@ class TestUtilityBase(TestcaseBase):
assert res[num_total_entities] == half
assert res[num_loaded_entities] == half
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.tags(CaseLabel.L1)
def test_wait_loading_collection_empty(self):
"""
target: test wait_for_loading
@ -880,7 +881,7 @@ class TestUtilityBase(TestcaseBase):
cw.load()
self.utility_wrap.wait_for_loading_complete(cw.name)
res, _ = self.utility_wrap.loading_progress(cw.name)
exp_res = {num_total_entities: 0, num_loaded_entities: 0}
exp_res = {loading_progress: "100%"}
assert res == exp_res
@pytest.mark.tag(CaseLabel.L1)