test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-03 20:39:36 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity
fbb9a5a790
milvus/tests-deprecating/python_test/test_compact.py
binbin eff75c7701
Replace sdk source and merge tests and tests20 (#7182) 
Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
2021-08-20 11:00:56 +08:00

723 lines
32 KiB
Python
Raw Blame History

import time
import pdb
import threading
import logging
from multiprocessing import Pool, Process
import pytest
from utils import *
from constants import *
# COMPACT_TIMEOUT = 180
# field_name = default_float_vec_field_name
# binary_field_name = default_binary_vec_field_name
# default_single_query = {
# "bool": {
# "must": [
# {"vector": {field_name: {"topk": 10, "query": gen_vectors(1, default_dim), "metric_type":"L2",
# "params": {"nprobe": 10}}}}
# ]
# }
# }
# default_binary_single_query = {
# "bool": {
# "must": [
# {"vector": {binary_field_name: {"topk": 10, "query": gen_binary_vectors(1, default_dim),
# "metric_type":"JACCARD", "params": {"nprobe": 10}}}}
# ]
# }
# }
# default_query, default_query_vecs = gen_query_vectors(binary_field_name, default_binary_entities, 1, 2)
#
#
# def ip_query():
# query = copy.deepcopy(default_single_query)
# query["bool"]["must"][0]["vector"][field_name].update({"metric_type": "IP"})
# return query
#
#
# class TestCompactBase:
# """
# ******************************************************************
# The following cases are used to test `compact` function
# ******************************************************************
# """
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_compact_collection_name_None(self, connect, collection):
# '''
# target: compact collection where collection name is None
# method: compact with the collection_name: None
# expected: exception raised
# '''
# collection_name = None
# with pytest.raises(Exception) as e:
# status = connect.compact(collection_name)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_compact_collection_name_not_existed(self, connect, collection):
# '''
# target: compact collection not existed
# method: compact with a random collection_name, which is not in db
# expected: exception raised
# '''
# collection_name = gen_unique_str("not_existed")
# with pytest.raises(Exception) as e:
# status = connect.compact(collection_name)
#
# @pytest.fixture(
# scope="function",
# params=gen_invalid_strs()
# )
# def get_collection_name(self, request):
# yield request.param
#
# @pytest.fixture(
# scope="function",
# params=gen_invalid_ints()
# )
# def get_threshold(self, request):
# yield request.param
#
# @pytest.mark.tags(CaseLabel.L2)
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_compact_collection_name_invalid(self, connect, get_collection_name):
# '''
# target: compact collection with invalid name
# method: compact with invalid collection_name
# expected: exception raised
# '''
# collection_name = get_collection_name
# with pytest.raises(Exception) as e:
# status = connect.compact(collection_name)
# # assert not status.OK()
#
# @pytest.mark.tags(CaseLabel.L2)
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_compact_threshold_invalid(self, connect, collection, get_threshold):
# '''
# target: compact collection with invalid name
# method: compact with invalid threshold
# expected: exception raised
# '''
# threshold = get_threshold
# if threshold != None:
# with pytest.raises(Exception) as e:
# status = connect.compact(collection, threshold)
#
# @pytest.mark.tags(CaseLabel.L2)
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_add_entity_and_compact(self, connect, collection):
# '''
# target: test add entity and compact
# method: add entity and compact collection
# expected: data_size before and after Compact
# '''
# # vector = gen_single_vector(dim)
# ids = connect.bulk_insert(collection, default_entity)
# assert len(ids) == 1
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# logging.getLogger().info(info)
# size_before = info["partitions"][0]["segments"][0]["data_size"]
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(collection)
# size_after = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_before == size_after)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_and_compact(self, connect, collection):
# '''
# target: test add entities and compact
# method: add entities and compact collection
# expected: data_size before and after Compact
# '''
# ids = connect.bulk_insert(collection, default_entities)
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# # assert status.OK()
# size_before = info["partitions"][0]["segments"][0]["data_size"]
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(collection)
# # assert status.OK()
# size_after = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_before == size_after)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_delete_part_and_compact(self, connect, collection):
# '''
# target: test add entities, delete part of them and compact
# method: add entities, delete a few and compact collection
# expected: status ok, data size maybe is smaller after compact
# '''
# ids = connect.bulk_insert(collection, default_entities)
# assert len(ids) == default_nb
# connect.flush([collection])
# delete_ids = [ids[0], ids[-1]]
# status = connect.delete_entity_by_id(collection, delete_ids)
# assert status.OK()
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# logging.getLogger().info(info["partitions"])
# size_before = info["partitions"][0]["data_size"]
# logging.getLogger().info(size_before)
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(collection)
# logging.getLogger().info(info["partitions"])
# size_after = info["partitions"][0]["data_size"]
# logging.getLogger().info(size_after)
# assert(size_before >= size_after)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_delete_part_and_compact_threshold(self, connect, collection):
# '''
# target: test add entities, delete part of them and compact
# method: add entities, delete a few and compact collection
# expected: status ok, data size maybe is smaller after compact
# '''
# ids = connect.bulk_insert(collection, default_entities)
# assert len(ids) == default_nb
# connect.flush([collection])
# delete_ids = [ids[0], ids[-1]]
# status = connect.delete_entity_by_id(collection, delete_ids)
# assert status.OK()
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# logging.getLogger().info(info["partitions"])
# size_before = info["partitions"][0]["data_size"]
# logging.getLogger().info(size_before)
# status = connect.compact(collection, 0.1)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(collection)
# logging.getLogger().info(info["partitions"])
# size_after = info["partitions"][0]["data_size"]
# logging.getLogger().info(size_after)
# assert(size_before >= size_after)
#
# @pytest.mark.tags(CaseLabel.L2)
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_delete_all_and_compact(self, connect, collection):
# '''
# target: test add entities, delete them and compact
# method: add entities, delete all and compact collection
# expected: status ok, no data size in collection info because collection is empty
# '''
# ids = connect.bulk_insert(collection, default_entities)
# assert len(ids) == default_nb
# connect.flush([collection])
# status = connect.delete_entity_by_id(collection, ids)
# assert status.OK()
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(collection)
# logging.getLogger().info(info["partitions"])
# assert not info["partitions"][0]["segments"]
#
# # TODO: enable
# @pytest.mark.tags(CaseLabel.L2)
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_partition_delete_half_and_compact(self, connect, collection):
# '''
# target: test add entities into partition, delete them and compact
# method: add entities, delete half of entities in partition and compact collection
# expected: status ok, data_size less than the older version
# '''
# connect.create_partition(collection, default_tag)
# assert connect.has_partition(collection, default_tag)
# ids = connect.bulk_insert(collection, default_entities, partition_name=default_tag)
# connect.flush([collection])
# info = connect.get_collection_stats(collection)
# logging.getLogger().info(info["partitions"])
# delete_ids = ids[:default_nb//2]
# status = connect.delete_entity_by_id(collection, delete_ids)
# assert status.OK()
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# logging.getLogger().info(info["partitions"])
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact
# info_after = connect.get_collection_stats(collection)
# logging.getLogger().info(info_after["partitions"])
# assert info["partitions"][1]["segments"][0]["data_size"] >= info_after["partitions"][1]["segments"][0]["data_size"]
#
# @pytest.fixture(
# scope="function",
# params=gen_simple_index()
# )
# def get_simple_index(self, request, connect):
# if str(connect._cmd("mode")) == "GPU":
# if not request.param["index_type"] not in ivf():
# pytest.skip("Only support index_type: idmap/ivf")
# if str(connect._cmd("mode")) == "CPU":
# if request.param["index_type"] in index_cpu_not_support():
# pytest.skip("CPU not support index_type: ivf_sq8h")
# return request.param
#
# @pytest.mark.tags(CaseLabel.L2)
# def test_compact_after_index_created(self, connect, collection, get_simple_index):
# '''
# target: test compact collection after index created
# method: add entities, create index, delete part of entities and compact
# expected: status ok, index description no change, data size smaller after compact
# '''
# count = 10
# ids = connect.bulk_insert(collection, default_entities)
# connect.flush([collection])
# connect.create_index(collection, field_name, get_simple_index)
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# size_before = info["partitions"][0]["segments"][0]["data_size"]
# delete_ids = ids[:default_nb//2]
# status = connect.delete_entity_by_id(collection, delete_ids)
# assert status.OK()
# connect.flush([collection])
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(collection)
# size_after = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_before >= size_after)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_add_entity_and_compact_twice(self, connect, collection):
# '''
# target: test add entity and compact twice
# method: add entity and compact collection twice
# expected: status ok, data size no change
# '''
# ids = connect.bulk_insert(collection, default_entity)
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# size_before = info["partitions"][0]["segments"][0]["data_size"]
# status = connect.compact(collection)
# assert status.OK()
# connect.flush([collection])
# # get collection info after compact
# info = connect.get_collection_stats(collection)
# size_after = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_before == size_after)
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact twice
# info = connect.get_collection_stats(collection)
# size_after_twice = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_after == size_after_twice)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_delete_part_and_compact_twice(self, connect, collection):
# '''
# target: test add entities, delete part of them and compact twice
# method: add entities, delete part and compact collection twice
# expected: status ok, data size smaller after first compact, no change after second
# '''
# ids = connect.bulk_insert(collection, default_entities)
# connect.flush([collection])
# delete_ids = [ids[0], ids[-1]]
# status = connect.delete_entity_by_id(collection, delete_ids)
# assert status.OK()
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# size_before = info["partitions"][0]["data_size"]
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(collection)
# size_after = info["partitions"][0]["data_size"]
# assert(size_before >= size_after)
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact twice
# info = connect.get_collection_stats(collection)
# size_after_twice = info["partitions"][0]["data_size"]
# assert(size_after == size_after_twice)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_compact_multi_collections(self, connect):
# '''
# target: test compact works or not with multiple collections
# method: create 50 collections, add entities into them and compact in turn
# expected: status ok
# '''
# nb = 100
# num_collections = 20
# entities = gen_entities(nb)
# collection_list = []
# for i in range(num_collections):
# collection_name = gen_unique_str("test_compact_multi_collection_%d" % i)
# collection_list.append(collection_name)
# connect.create_collection(collection_name, default_fields)
# for i in range(num_collections):
# ids = connect.bulk_insert(collection_list[i], entities)
# connect.delete_entity_by_id(collection_list[i], ids[:nb//2])
# status = connect.compact(collection_list[i])
# assert status.OK()
# connect.drop_collection(collection_list[i])
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_add_entity_after_compact(self, connect, collection):
# '''
# target: test add entity after compact
# method: after compact operation, add entity
# expected: status ok, entity added
# '''
# ids = connect.bulk_insert(collection, default_entities)
# assert len(ids) == default_nb
# connect.flush([collection])
# # get collection info before compact
# info = connect.get_collection_stats(collection)
# size_before = info["partitions"][0]["segments"][0]["data_size"]
# status = connect.compact(collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(collection)
# size_after = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_before == size_after)
# ids = connect.bulk_insert(collection, default_entity)
# connect.flush([collection])
# res = connect.count_entities(collection)
# assert res == default_nb+1
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_index_creation_after_compact(self, connect, collection, get_simple_index):
# '''
# target: test index creation after compact
# method: after compact operation, create index
# expected: status ok, index description no change
# '''
# ids = connect.bulk_insert(collection, default_entities)
# connect.flush([collection])
# status = connect.delete_entity_by_id(collection, ids[:10])
# assert status.OK()
# connect.flush([collection])
# status = connect.compact(collection)
# assert status.OK()
# status = connect.create_index(collection, field_name, get_simple_index)
# assert status.OK()
# # status, result = connect.get_index_info(collection)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_delete_entities_after_compact(self, connect, collection):
# '''
# target: test delete entities after compact
# method: after compact operation, delete entities
# expected: status ok, entities deleted
# '''
# ids = connect.bulk_insert(collection, default_entities)
# assert len(ids) == default_nb
# connect.flush([collection])
# status = connect.compact(collection)
# assert status.OK()
# connect.flush([collection])
# status = connect.delete_entity_by_id(collection, ids)
# assert status.OK()
# connect.flush([collection])
# assert connect.count_entities(collection) == 0
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_search_after_compact(self, connect, collection):
# '''
# target: test search after compact
# method: after compact operation, search vector
# expected: status ok
# '''
# ids = connect.bulk_insert(collection, default_entities)
# assert len(ids) == default_nb
# connect.flush([collection])
# status = connect.compact(collection)
# assert status.OK()
# query = copy.deepcopy(default_single_query)
# query["bool"]["must"][0]["vector"][field_name]["query"] = [default_entity[-1]["values"][0],
# default_entities[-1]["values"][0],
# default_entities[-1]["values"][-1]]
# res = connect.search(collection, query)
# logging.getLogger().debug(res)
# assert len(res) == len(query["bool"]["must"][0]["vector"][field_name]["query"])
# assert res[0]._distances[0] > epsilon
# assert res[1]._distances[0] < epsilon
# assert res[2]._distances[0] < epsilon
#
#
# class TestCompactBinary:
# """
# ******************************************************************
# The following cases are used to test `compact` function
# ******************************************************************
# """
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_add_entity_and_compact(self, connect, binary_collection):
# '''
# target: test add binary vector and compact
# method: add vector and compact collection
# expected: status ok, vector added
# '''
# ids = connect.bulk_insert(binary_collection, default_binary_entity)
# assert len(ids) == 1
# connect.flush([binary_collection])
# # get collection info before compact
# info = connect.get_collection_stats(binary_collection)
# size_before = info["partitions"][0]["segments"][0]["data_size"]
# status = connect.compact(binary_collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(binary_collection)
# size_after = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_before == size_after)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_and_compact(self, connect, binary_collection):
# '''
# target: test add entities with binary vector and compact
# method: add entities and compact collection
# expected: status ok, entities added
# '''
# ids = connect.bulk_insert(binary_collection, default_binary_entities)
# assert len(ids) == default_nb
# connect.flush([binary_collection])
# # get collection info before compact
# info = connect.get_collection_stats(binary_collection)
# size_before = info["partitions"][0]["segments"][0]["data_size"]
# status = connect.compact(binary_collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(binary_collection)
# size_after = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_before == size_after)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_delete_part_and_compact(self, connect, binary_collection):
# '''
# target: test add entities, delete part of them and compact
# method: add entities, delete a few and compact collection
# expected: status ok, data size is smaller after compact
# '''
# ids = connect.bulk_insert(binary_collection, default_binary_entities)
# assert len(ids) == default_nb
# connect.flush([binary_collection])
# delete_ids = [ids[0], ids[-1]]
# status = connect.delete_entity_by_id(binary_collection, delete_ids)
# assert status.OK()
# connect.flush([binary_collection])
# # get collection info before compact
# info = connect.get_collection_stats(binary_collection)
# logging.getLogger().info(info["partitions"])
# size_before = info["partitions"][0]["data_size"]
# logging.getLogger().info(size_before)
# status = connect.compact(binary_collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(binary_collection)
# logging.getLogger().info(info["partitions"])
# size_after = info["partitions"][0]["data_size"]
# logging.getLogger().info(size_after)
# assert(size_before >= size_after)
#
# @pytest.mark.tags(CaseLabel.L2)
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_delete_all_and_compact(self, connect, binary_collection):
# '''
# target: test add entities, delete them and compact
# method: add entities, delete all and compact collection
# expected: status ok, no data size in collection info because collection is empty
# '''
# ids = connect.bulk_insert(binary_collection, default_binary_entities)
# assert len(ids) == default_nb
# connect.flush([binary_collection])
# status = connect.delete_entity_by_id(binary_collection, ids)
# assert status.OK()
# connect.flush([binary_collection])
# # get collection info before compact
# info = connect.get_collection_stats(binary_collection)
# status = connect.compact(binary_collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(binary_collection)
# assert status.OK()
# logging.getLogger().info(info["partitions"])
# assert not info["partitions"][0]["segments"]
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_add_entity_and_compact_twice(self, connect, binary_collection):
# '''
# target: test add entity and compact twice
# method: add entity and compact collection twice
# expected: status ok
# '''
# ids = connect.bulk_insert(binary_collection, default_binary_entity)
# assert len(ids) == 1
# connect.flush([binary_collection])
# # get collection info before compact
# info = connect.get_collection_stats(binary_collection)
# size_before = info["partitions"][0]["segments"][0]["data_size"]
# status = connect.compact(binary_collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(binary_collection)
# size_after = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_before == size_after)
# status = connect.compact(binary_collection)
# assert status.OK()
# # get collection info after compact twice
# info = connect.get_collection_stats(binary_collection)
# size_after_twice = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_after == size_after_twice)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_insert_delete_part_and_compact_twice(self, connect, binary_collection):
# '''
# target: test add entities, delete part of them and compact twice
# method: add entities, delete part and compact collection twice
# expected: status ok, data size smaller after first compact, no change after second
# '''
# ids = connect.bulk_insert(binary_collection, default_binary_entities)
# assert len(ids) == default_nb
# connect.flush([binary_collection])
# delete_ids = [ids[0], ids[-1]]
# status = connect.delete_entity_by_id(binary_collection, delete_ids)
# assert status.OK()
# connect.flush([binary_collection])
# # get collection info before compact
# info = connect.get_collection_stats(binary_collection)
# size_before = info["partitions"][0]["data_size"]
# status = connect.compact(binary_collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(binary_collection)
# size_after = info["partitions"][0]["data_size"]
# assert(size_before >= size_after)
# status = connect.compact(binary_collection)
# assert status.OK()
# # get collection info after compact twice
# info = connect.get_collection_stats(binary_collection)
# size_after_twice = info["partitions"][0]["data_size"]
# assert(size_after == size_after_twice)
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_compact_multi_collections(self, connect):
# '''
# target: test compact works or not with multiple collections
# method: create 10 collections, add entities into them and compact in turn
# expected: status ok
# '''
# nq = 100
# num_collections = 10
# tmp, entities = gen_binary_entities(nq)
# collection_list = []
# for i in range(num_collections):
# collection_name = gen_unique_str("test_compact_multi_collection_%d" % i)
# collection_list.append(collection_name)
# connect.create_collection(collection_name, default_binary_fields)
# for i in range(num_collections):
# ids = connect.bulk_insert(collection_list[i], entities)
# assert len(ids) == nq
# status = connect.delete_entity_by_id(collection_list[i], [ids[0], ids[-1]])
# assert status.OK()
# connect.flush([collection_list[i]])
# status = connect.compact(collection_list[i])
# assert status.OK()
# status = connect.drop_collection(collection_list[i])
# assert status.OK()
#
# @pytest.mark.tags(CaseLabel.L2)
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_add_entity_after_compact(self, connect, binary_collection):
# '''
# target: test add entity after compact
# method: after compact operation, add entity
# expected: status ok, entity added
# '''
# ids = connect.bulk_insert(binary_collection, default_binary_entities)
# connect.flush([binary_collection])
# # get collection info before compact
# info = connect.get_collection_stats(binary_collection)
# size_before = info["partitions"][0]["segments"][0]["data_size"]
# status = connect.compact(binary_collection)
# assert status.OK()
# # get collection info after compact
# info = connect.get_collection_stats(binary_collection)
# size_after = info["partitions"][0]["segments"][0]["data_size"]
# assert(size_before == size_after)
# ids = connect.bulk_insert(binary_collection, default_binary_entity)
# connect.flush([binary_collection])
# res = connect.count_entities(binary_collection)
# assert res == default_nb + 1
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_delete_entities_after_compact(self, connect, binary_collection):
# '''
# target: test delete entities after compact
# method: after compact operation, delete entities
# expected: status ok, entities deleted
# '''
# ids = connect.bulk_insert(binary_collection, default_binary_entities)
# connect.flush([binary_collection])
# status = connect.compact(binary_collection)
# assert status.OK()
# connect.flush([binary_collection])
# status = connect.delete_entity_by_id(binary_collection, ids)
# assert status.OK()
# connect.flush([binary_collection])
# res = connect.count_entities(binary_collection)
# assert res == 0
#
# @pytest.mark.tags(CaseLabel.L2)
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_search_after_compact(self, connect, binary_collection):
# '''
# target: test search after compact
# method: after compact operation, search vector
# expected: status ok
# '''
# ids = connect.bulk_insert(binary_collection, default_binary_entities)
# assert len(ids) == default_nb
# connect.flush([binary_collection])
# status = connect.compact(binary_collection)
# assert status.OK()
# query_vecs = [default_raw_binary_vectors[0]]
# distance = jaccard(query_vecs[0], default_raw_binary_vectors[0])
# query = copy.deepcopy(default_binary_single_query)
# query["bool"]["must"][0]["vector"][binary_field_name]["query"] = [default_binary_entities[-1]["values"][0],
# default_binary_entities[-1]["values"][-1]]
#
# res = connect.search(binary_collection, query)
# assert abs(res[0]._distances[0]-distance) <= epsilon
#
# @pytest.mark.timeout(COMPACT_TIMEOUT)
# def test_search_after_compact_ip(self, connect, collection):
# '''
# target: test search after compact
# method: after compact operation, search vector
# expected: status ok
# '''
# ids = connect.bulk_insert(collection, default_entities)
# assert len(ids) == default_nb
# connect.flush([collection])
# status = connect.compact(collection)
# query = ip_query()
# query["bool"]["must"][0]["vector"][field_name]["query"] = [default_entity[-1]["values"][0],
# default_entities[-1]["values"][0],
# default_entities[-1]["values"][-1]]
# res = connect.search(collection, query)
# logging.getLogger().info(res)
# assert len(res) == len(query["bool"]["must"][0]["vector"][field_name]["query"])
# assert res[0]._distances[0] < 1 - epsilon
# assert res[1]._distances[0] > 1 - epsilon
# assert res[2]._distances[0] > 1 - epsilon
Reference in New Issue View Git Blame Copy Permalink