milvus-io / milvus

        status, res = connect.count_entities(ip_collection)

        assert res == nb

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

        '''

        target: test count_entities, after index have been created

        method: add vectors in db, and create index, then calling count_entities with correct params

        expected: count_entities raise exception

        '''

        index_param = get_simple_index["index_param"]

        index_type = get_simple_index["index_type"]

        nb = 100

        vectors = gen_vectors(nb, dim)

        res = connect.insert(collection_name=ip_collection, records=vectors)

        connect.flush([ip_collection])

        connect.create_index(ip_collection, index_type, index_param)

        status, res = connect.count_entities(ip_collection)

        assert res == nb

    # @pytest.mark.level(2)

    # def test_count_without_connection(self, ip_collection, dis_connect):

        status, res = connect.count_entities(collection)

        assert res == nb * 2

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

        '''

        target: test count_entities, after index have been created

        method: add vectors in db, and create index, then calling count_entities with correct params 

        expected: count_entities raise exception

        '''

        index_param = get_simple_index["index_param"]

        index_type = get_simple_index["index_type"]

        nb = 100

        vectors = gen_vectors(nb, dim)

        res = connect.insert(collection_name=collection, records=vectors)

        connect.flush([collection])

        connect.create_index(collection, index_type, index_param)

        status, res = connect.count_entities(collection)

        assert res == nb

    # @pytest.mark.level(2)

    # def test_count_without_connection(self, collection, dis_connect):