milvus-io / milvus

            assert False, str(e)

        # TODO: 

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

        '''

        target: get vector ids when there is index and with partition

        method: create partition, add vectors to it and call list_id_in_segment, check if the segment contains vectors

        expected: status ok

        '''

        connect.create_partition(collection, tag)

        ids = connect.insert(collection, entities, partition_tag=tag)

        connect.flush([collection])

        stats = connect.get_collection_stats(collection)

        assert stats["partitions"][1]["tag"] == tag

        try:

            connect.list_id_in_segment(collection, stats["partitions"][1]["segments"][0]["id"])

        except Exception as e:

            assert False, str(e)

        # vector_ids should match ids

        # TODO

        vector_ids = connect.list_id_in_segment(jac_collection, seg_id)

        # TODO: 

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

        '''

        target: get vector ids when there is index and with partition

        method: create partition, add vectors to it and call list_id_in_segment, check if the segment contains vectors

        expected: status ok

        '''

        connect.create_partition(jac_collection, tag)

        ids = connect.insert(jac_collection, entities, partition_tag=tag)

        connect.flush([jac_collection])

        stats = connect.get_collection_stats(jac_collection)

        assert stats["partitions"][1]["tag"] == tag

        vector_ids = connect.list_id_in_segment(jac_collection, stats["partitions"][1]["segments"][0]["id"])

        # vector_ids should match ids

        # TODO

        vector_ids = connect.list_id_in_segment(ip_collection, seg_id)

        # TODO: 

    def test_list_id_in_segment_with_index_B(self, connect, ip_collection, get_simple_index):

        '''

        target: get vector ids when there is index and with partition

        method: create partition, add vectors to it and call list_id_in_segment, check if the segment contains vectors

        expected: status ok

        '''

        connect.create_partition(ip_collection, tag)

        ids = connect.insert(ip_collection, entities, partition_tag=tag)

        connect.flush([ip_collection])

        stats = connect.get_collection_stats(ip_collection)

        assert stats["partitions"][1]["tag"] == tag

        vector_ids = connect.list_id_in_segment(ip_collection, stats["partitions"][1]["segments"][0]["id"])

        # vector_ids should match ids

        # TODO