未验证 提交 d70190cf 编写于 作者: B binbin 提交者: GitHub

Modify test case for interface change (#15881)

Signed-off-by: NBinbin Lv <binbin.lv@zilliz.com>
上级 8a708112
...@@ -161,7 +161,7 @@ class TestInsertParams(TestcaseBase): ...@@ -161,7 +161,7 @@ class TestInsertParams(TestcaseBase):
assert collection_w.is_empty assert collection_w.is_empty
assert collection_w.num_entities == 0 assert collection_w.num_entities == 0
@pytest.mark.tags(CaseLabel.L2) @pytest.mark.tags(CaseLabel.L1)
def test_insert_numpy_data(self): def test_insert_numpy_data(self):
""" """
target: test insert numpy.ndarray data target: test insert numpy.ndarray data
...@@ -171,8 +171,7 @@ class TestInsertParams(TestcaseBase): ...@@ -171,8 +171,7 @@ class TestInsertParams(TestcaseBase):
c_name = cf.gen_unique_str(prefix) c_name = cf.gen_unique_str(prefix)
collection_w = self.init_collection_wrap(name=c_name) collection_w = self.init_collection_wrap(name=c_name)
data = cf.gen_numpy_data(nb=10) 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)
collection_w.insert(data=data, check_task=CheckTasks.err_res, check_items=error)
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_insert_binary_dataframe(self): def test_insert_binary_dataframe(self):
......
...@@ -18,6 +18,7 @@ default_dim = ct.default_dim ...@@ -18,6 +18,7 @@ default_dim = ct.default_dim
default_nb = ct.default_nb default_nb = ct.default_nb
num_loaded_entities = "num_loaded_entities" num_loaded_entities = "num_loaded_entities"
num_total_entities = "num_total_entities" num_total_entities = "num_total_entities"
loading_progress = "loading_progress"
class TestUtilityParams(TestcaseBase): class TestUtilityParams(TestcaseBase):
...@@ -868,7 +869,7 @@ class TestUtilityBase(TestcaseBase): ...@@ -868,7 +869,7 @@ class TestUtilityBase(TestcaseBase):
assert res[num_total_entities] == half assert res[num_total_entities] == half
assert res[num_loaded_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): def test_wait_loading_collection_empty(self):
""" """
target: test wait_for_loading target: test wait_for_loading
...@@ -880,7 +881,7 @@ class TestUtilityBase(TestcaseBase): ...@@ -880,7 +881,7 @@ class TestUtilityBase(TestcaseBase):
cw.load() cw.load()
self.utility_wrap.wait_for_loading_complete(cw.name) self.utility_wrap.wait_for_loading_complete(cw.name)
res, _ = self.utility_wrap.loading_progress(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 assert res == exp_res
@pytest.mark.tag(CaseLabel.L1) @pytest.mark.tag(CaseLabel.L1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册