mirror of https://github.com/milvus-io/milvus.git
Fix insert and add index
Signed-off-by: ThreadDao <yufen.zong@zilliz.com>pull/4973/head^2
parent
4b0210d723
commit
ab859effad
|
@ -746,7 +746,6 @@ class TestInsertAsync:
|
|||
stats = connect.get_collection_stats(collection)
|
||||
assert stats[row_count] == 0
|
||||
|
||||
# #1339
|
||||
@pytest.mark.tags("0331")
|
||||
def test_insert_async_invalid_params(self, connect):
|
||||
'''
|
||||
|
@ -770,6 +769,7 @@ class TestInsertAsync:
|
|||
'''
|
||||
entities = []
|
||||
future = connect.insert(collection, entities, _async=True)
|
||||
future.done()
|
||||
with pytest.raises(Exception) as e:
|
||||
future.result()
|
||||
|
||||
|
@ -1128,12 +1128,11 @@ class TestInsertInvalidBinary(object):
|
|||
connect.insert(binary_collection, tmp_entity)
|
||||
|
||||
@pytest.mark.level(2)
|
||||
# #1352
|
||||
@pytest.mark.tags("fail")
|
||||
@pytest.mark.tags("0331")
|
||||
def test_insert_with_invalid_field_entity_value(self, connect, binary_collection, get_field_vectors_value):
|
||||
tmp_entity = copy.deepcopy(default_binary_entity)
|
||||
src_vectors = tmp_entity[-1]["values"]
|
||||
src_vectors[0][1] = get_field_vectors_value
|
||||
src_vectors[0] = get_field_vectors_value
|
||||
with pytest.raises(Exception):
|
||||
connect.insert(binary_collection, tmp_entity)
|
||||
|
||||
|
@ -1160,7 +1159,6 @@ class TestInsertInvalidBinary(object):
|
|||
|
||||
@pytest.mark.level(2)
|
||||
@pytest.mark.tags("0331")
|
||||
# TODO 'bytes' object does not support item assignment
|
||||
def test_insert_with_invalid_field_entities_value(self, connect, binary_collection, get_field_vectors_value):
|
||||
tmp_entities = copy.deepcopy(default_binary_entities)
|
||||
src_vector = tmp_entities[-1]["values"]
|
||||
|
|
|
@ -557,6 +557,13 @@ class TestIndexBase:
|
|||
connect.create_index(collection, field_name, get_simple_index)
|
||||
connect.drop_index(collection, field_name)
|
||||
|
||||
def test_create_PQ_without_nbits(self, connect, collection):
|
||||
PQ_index = {"index_type": "IVF_PQ", "params": {"nlist": 128, "m": 16}, "metric_type": "L2"}
|
||||
ids = connect.insert(collection, default_entities)
|
||||
connect.create_index(collection, field_name, PQ_index)
|
||||
index = connect.describe_index(collection, field_name)
|
||||
assert index == PQ_index
|
||||
|
||||
|
||||
class TestIndexBinary:
|
||||
@pytest.fixture(
|
||||
|
|
Loading…
Reference in New Issue