import os import pytest @pytest.fixture(scope="session", autouse=True) def set_test_db_env(): os.environ['DB_ENV'] = 'test'