Add testcases for flush after inserting

Signed-off-by: del-zhenwu <zhenxiang.li@zilliz.com>
pull/4973/head^2
del-zhenwu 2021-03-19 15:36:50 +08:00 committed by yefu.chen
parent c68bcfea16
commit 5a53b118a7
3 changed files with 9 additions and 1 deletions

View File

@ -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)

View File

@ -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):

View File

@ -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")