Files
Auto-GPT/tests/utils.py
Merwane Hamadi b7cd56f72b move decorator higher up
Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com>
2023-04-23 00:45:54 +02:00

19 lines
408 B
Python

import os
import pytest
def requires_api_key(env_var):
def decorator(func):
def wrapper(*args, **kwargs):
if not os.environ.get(env_var):
pytest.skip(
f"Environment variable '{env_var}' is not set, skipping the test."
)
else:
return func(*args, **kwargs)
return wrapper
return decorator