milvus-io / milvus

    return collection_name

@pytest.fixture(scope="function")

def binary_id_collection(request, connect):

    ori_collection_name = getattr(request.module, "collection_id", "test")

    collection_name = gen_unique_str(ori_collection_name)

    try:

        fields = gen_binary_default_fields(auto_id=True)

        connect.create_collection(collection_name, fields)

    except Exception as e:

        pytest.exit(str(e))

    def teardown():

        collection_names = connect.list_collections()

        for collection_name in collection_names:

            connect.drop_collection(collection_name, timeout=delete_timeout)

    request.addfinalizer(teardown)

    assert connect.has_collection(collection_name)

    return collection_name

    return collection_name

@pytest.fixture(scope="function")

def id_collection(request, connect):

    ori_collection_name = getattr(request.module, "collection_id", "test")

    collection_name = gen_unique_str(ori_collection_name)

    try:

        fields = gen_default_fields(auto_id=True)

        connect.create_collection(collection_name, fields)

    except Exception as e:

        pytest.exit(str(e))

    def teardown():

        collection_names = connect.list_collections()

        for collection_name in collection_names:

            connect.drop_collection(collection_name, timeout=delete_timeout)

    request.addfinalizer(teardown)

    assert connect.has_collection(collection_name)

    return collection_name

@pytest.fixture(scope="function")