def test_collection_binary_created_by_data_list(self):
"""
target: test create collection by data list
method: data type is list-like
expected: raise exception
"""
self._connect()
c_name = cf.gen_unique_str(prefix)
data, _ = cf.gen_default_binary_list_data(nb=100)
error = {ct.err_code: 0, ct.err_msg: "Data of not pandas.DataFrame type should bepassed into the schema"}
self.collection_wrap.init_collection(name=c_name, schema=None, data=data,
check_task=CheckTasks.err_res, check_items=error)
def test_insert_binary_data(self):
"""
target: test insert list-like binary data
method: 1. create by schema 2. insert data
expected: assert num_entities
"""
c_name = cf.gen_unique_str(prefix)
collection_w = self.init_collection_wrap(name=c_name, schema=default_binary_schema)
data, _ = cf.gen_default_binary_list_data(ct.default_nb)
mutation_res, _ = collection_w.insert(data=data)
assert mutation_res.insert_count == ct.default_nb
assert mutation_res.primary_keys == data[0]
assert collection_w.num_entities == ct.default_nb
def test_insert_binary_data(self):
"""
target: test insert list-like binary data
method: 1. create by schema 2. insert data
expected: assert num_entities
"""
c_name = cf.gen_unique_str(prefix)
collection_w = self.init_collection_wrap(name=c_name, schema=default_binary_schema)
data, _ = cf.gen_default_binary_list_data(ct.default_nb)
collection_w.insert(data=data)
conn, _ = self.connection_wrap.get_connection()
conn.flush([c_name])
assert collection_w.num_entities == ct.default_nb
def test_collection_binary_with_data_list(self):
"""
target: test collection with data (list-like)
method: create binary collection with data (list-like)
expected: collection num entities equal to nb
"""
conn = self._connect()
c_name = cf.gen_unique_str(prefix)
data, _ = cf.gen_default_binary_list_data(ct.default_nb)
self.collection_wrap.init_collection(c_name, schema=default_binary_schema, data=data,
check_task=CheckTasks.check_collection_property,
check_items={exp_name: c_name, exp_schema: default_binary_schema})
conn.flush([c_name])
assert self.collection_wrap.num_entities == ct.default_nb
