mirror of https://github.com/milvus-io/milvus.git
Add testcases for flush after inserting
Signed-off-by: del-zhenwu <zhenxiang.li@zilliz.com>pull/4973/head^2
parent
c68bcfea16
commit
5a53b118a7
|
@ -854,6 +854,7 @@ class TestInsertMultiCollections:
|
|||
collection_name = gen_unique_str(uid)
|
||||
connect.create_collection(collection_name, default_fields)
|
||||
ids = connect.insert(collection, default_entity)
|
||||
connect.flush([collection])
|
||||
connect.create_index(collection_name, field_name, get_simple_index)
|
||||
index = connect.describe_index(collection_name, field_name)
|
||||
assert index == get_simple_index
|
||||
|
@ -1164,4 +1165,4 @@ class TestInsertInvalidBinary(object):
|
|||
src_vector = tmp_entities[-1]["values"]
|
||||
src_vector[1] = get_field_vectors_value
|
||||
with pytest.raises(Exception):
|
||||
connect.insert(binary_collection, tmp_entities)
|
||||
connect.insert(binary_collection, tmp_entities)
|
||||
|
|
|
@ -201,6 +201,8 @@ class TestFlushBase:
|
|||
'''
|
||||
ids = [i for i in range(default_nb)]
|
||||
ids = connect.insert(id_collection, default_entities, ids)
|
||||
# add flush
|
||||
connect.flush([id_collection])
|
||||
timeout = 20
|
||||
start_time = time.time()
|
||||
while (time.time() - start_time < timeout):
|
||||
|
|
|
@ -144,6 +144,7 @@ class TestIndexBase:
|
|||
expected: return search success
|
||||
'''
|
||||
ids = connect.insert(collection, default_entities)
|
||||
connect.flush([collection])
|
||||
connect.create_index(collection, field_name, get_simple_index)
|
||||
logging.getLogger().info(connect.describe_index(collection, field_name))
|
||||
nq = get_nq
|
||||
|
@ -234,6 +235,7 @@ class TestIndexBase:
|
|||
expected: return code 0, and describe index result equals with the second index params
|
||||
'''
|
||||
ids = connect.insert(collection, default_entities)
|
||||
connect.flush([collection])
|
||||
indexs = [default_index, {"metric_type":"L2", "index_type": "FLAT", "params":{"nlist": 1024}}]
|
||||
for index in indexs:
|
||||
connect.create_index(collection, field_name, index)
|
||||
|
@ -310,6 +312,7 @@ class TestIndexBase:
|
|||
'''
|
||||
metric_type = "IP"
|
||||
ids = connect.insert(collection, default_entities)
|
||||
connect.flush([collection])
|
||||
get_simple_index["metric_type"] = metric_type
|
||||
connect.create_index(collection, field_name, get_simple_index)
|
||||
connect.load_collection(collection)
|
||||
|
@ -404,6 +407,7 @@ class TestIndexBase:
|
|||
expected: return code 0, and describe index result equals with the second index params
|
||||
'''
|
||||
ids = connect.insert(collection, default_entities)
|
||||
connect.flush([collection])
|
||||
connect.load_collection(collection)
|
||||
stats = connect.get_collection_stats(collection)
|
||||
assert stats["row_count"] == default_nb
|
||||
|
@ -649,6 +653,7 @@ class TestIndexBinary:
|
|||
'''
|
||||
nq = get_nq
|
||||
ids = connect.insert(binary_collection, default_binary_entities)
|
||||
connect.flush([binary_collection])
|
||||
connect.create_index(binary_collection, binary_field_name, get_jaccard_index)
|
||||
connect.load_collection(binary_collection)
|
||||
query, vecs = gen_query_vectors(binary_field_name, default_binary_entities, default_top_k, nq, metric_type="JACCARD")
|
||||
|
|
Loading…
Reference in New Issue