milvus-io / milvus

        assert status.OK()

        assert len(result) == len(query_vecs)

    @pytest.mark.timeout(BUILD_TIMEOUT)

    def test_create_index_search_with_query_vectors_superstructure(self, connect, superstructure_collection, get_superstructure_index):

        '''

        target: test create index interface, search with more query vectors

        method: create collection and add vectors in it, create index

        expected: return code equals to 0, and search success

        '''

        index_param = get_superstructure_index["index_param"]

        index_type = get_superstructure_index["index_type"]

        logging.getLogger().info(get_superstructure_index)

        status, ids = connect.insert(superstructure_collection, self.vectors)

        status = connect.create_index(superstructure_collection, index_type, index_param)

        logging.getLogger().info(connect.get_index_info(superstructure_collection))

        query_vecs = [self.vectors[0], self.vectors[1], self.vectors[2]]

        top_k = 5

        search_param = get_search_param(index_type)

        status, result = connect.search(superstructure_collection, top_k, query_vecs, params=search_param)

        logging.getLogger().info(result)

        assert status.OK()

        assert len(result) == len(query_vecs)

    """

    ******************************************************************

    #     with pytest.raises(Exception) as e:

    #         status = dis_connect.create_index(ham_collection, IndexType.IVF_SQ8, index_param)

    @pytest.mark.timeout(BUILD_TIMEOUT)

    def test_create_index_search_with_query_vectors(self, connect, ham_collection, get_hamming_index):

        '''

        target: test create index interface, search with more query vectors

        method: create collection and add vectors in it, create index

        expected: return code equals to 0, and search success

        '''

        index_param = get_hamming_index["index_param"]

        index_type = get_hamming_index["index_type"]

        logging.getLogger().info(get_hamming_index)

        status, ids = connect.insert(ham_collection, self.vectors)

        status = connect.create_index(ham_collection,  index_type, index_param)

        logging.getLogger().info(connect.get_index_info(ham_collection))

        query_vecs = [self.vectors[0], self.vectors[1], self.vectors[2]]

        top_k = 5

        search_param = get_search_param(index_type)

        status, result = connect.search(ham_collection, top_k, query_vecs, params=search_param)

        logging.getLogger().info(result)

        assert status.OK()

        assert len(result) == len(query_vecs)

    @pytest.mark.timeout(BUILD_TIMEOUT)

    def test_create_index_search_with_query_vectors_superstructure(self, connect, superstructure_collection, get_superstructure_index):

    #     with pytest.raises(Exception) as e:

    #         status = dis_connect.create_index(jac_collection, IndexType.IVF_SQ8, index_param)

    @pytest.mark.timeout(BUILD_TIMEOUT)

    def test_create_index_search_with_query_vectors(self, connect, jac_collection, get_jaccard_index):

        '''

        target: test create index interface, search with more query vectors

        method: create collection and add vectors in it, create index

        expected: return code equals to 0, and search success

        '''

        index_param = get_jaccard_index["index_param"]

        index_type = get_jaccard_index["index_type"]

        logging.getLogger().info(get_jaccard_index)

        status, ids = connect.insert(jac_collection, self.vectors)

        status = connect.create_index(jac_collection, index_type, index_param)

        logging.getLogger().info(connect.get_index_info(jac_collection))

        query_vecs = [self.vectors[0], self.vectors[1], self.vectors[2]]

        top_k = 5

        search_param = get_search_param(index_type)

        status, result = connect.search(jac_collection, top_k, query_vecs, params=search_param)

        logging.getLogger().info(result)

        assert status.OK()

        assert len(result) == len(query_vecs)

    """

    ******************************************************************

    #     with pytest.raises(Exception) as e:

    #         status = dis_connect.create_index(collection, index_type, index_param)

    @pytest.mark.timeout(BUILD_TIMEOUT)

    def test_create_index_search_with_query_vectors(self, connect, collection, get_simple_index):

        '''

        target: test create index interface, search with more query vectors

        method: create collection and add vectors in it, create index

        expected: return code equals to 0, and search success

        '''

        index_param = get_simple_index["index_param"]

        index_type = get_simple_index["index_type"]

        logging.getLogger().info(get_simple_index)

        status, ids = connect.insert(collection, vectors)

        status = connect.create_index(collection, index_type, index_param)

        logging.getLogger().info(connect.get_index_info(collection))

        query_vecs = [vectors[0], vectors[1], vectors[2]]

        top_k = 5

        search_param = get_search_param(index_type)

        status, result = connect.search(collection, top_k, query_vecs, params=search_param)

        assert status.OK()

        assert len(result) == len(query_vecs)

        logging.getLogger().info(result)

    @pytest.mark.timeout(BUILD_TIMEOUT)

    @pytest.mark.level(2)