mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-03 04:19:18 +08:00
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
import datetime
|
|
import pytest
|
|
|
|
from base.client_base import TestcaseBase
|
|
from common import common_func as cf
|
|
from common import common_type as ct
|
|
from common.common_type import CaseLabel
|
|
|
|
prefix = "e2e_"
|
|
|
|
|
|
class TestE2e(TestcaseBase):
|
|
""" Test case of end to end"""
|
|
@pytest.mark.tags(CaseLabel.L2)
|
|
# @pytest.mark.parametrize("name", [(cf.gen_unique_str(prefix))])
|
|
def test_milvus_default(self):
|
|
from utils.util_log import test_log as log
|
|
# create
|
|
name = cf.gen_unique_str(prefix)
|
|
t0 = datetime.datetime.now()
|
|
collection_w = self.init_collection_wrap(name=name)
|
|
tt = datetime.datetime.now() - t0
|
|
log.debug(f"assert create: {tt}")
|
|
assert collection_w.name == name
|
|
|
|
# insert
|
|
data = cf.gen_default_list_data()
|
|
t0 = datetime.datetime.now()
|
|
_, res = collection_w.insert(data)
|
|
tt = datetime.datetime.now() - t0
|
|
log.debug(f"assert insert: {tt}")
|
|
assert res
|
|
|
|
# flush
|
|
t0 = datetime.datetime.now()
|
|
assert collection_w.num_entities == len(data[0])
|
|
tt = datetime.datetime.now() - t0
|
|
log.debug(f"assert flush: {tt}")
|
|
|
|
# search
|
|
collection_w.load()
|
|
search_vectors = cf.gen_vectors(1, ct.default_dim)
|
|
t0 = datetime.datetime.now()
|
|
res_1, _ = collection_w.search(data=search_vectors,
|
|
anns_field=ct.default_float_vec_field_name,
|
|
param={"nprobe": 16}, limit=1)
|
|
tt = datetime.datetime.now() - t0
|
|
log.debug(f"assert search: {tt}")
|
|
assert len(res_1) == 1
|
|
# collection_w.release()
|
|
|
|
# index
|
|
collection_w.insert(cf.gen_default_dataframe_data(nb=5000))
|
|
assert collection_w.num_entities == len(data[0]) + 5000
|
|
_index_params = {"index_type": "IVF_SQ8", "metric_type": "L2", "params": {"nlist": 64}}
|
|
t0 = datetime.datetime.now()
|
|
index, _ = collection_w.create_index(field_name=ct.default_float_vec_field_name,
|
|
index_params=_index_params,
|
|
name=cf.gen_unique_str())
|
|
tt = datetime.datetime.now() - t0
|
|
log.debug(f"assert index: {tt}")
|
|
assert len(collection_w.indexes) == 1
|
|
|
|
# query
|
|
term_expr = f'{ct.default_int64_field_name} in [3001,4001,4999,2999]'
|
|
t0 = datetime.datetime.now()
|
|
res, _ = collection_w.query(term_expr)
|
|
tt = datetime.datetime.now() - t0
|
|
log.debug(f"assert query: {tt}")
|
|
assert len(res) == 4
|