[Python] pytestで環境変数を設定するときはfixtureでpatchするのがいい

#Python

#pytest

武田隆志

2025.02.27

こんにちは。サービス開発室の武田です。

pytestを使用してユニットテストをする上で、環境変数が必要になる場合があります。しかし間違った方法では正しくテストができません。

正しくない方法

os.environを使用して環境変数に値を設定しています。一見シンプルな方法ですが、これ以降のテストケースにもこの環境変数が引き継がれます。つまりユニットテストの 順序に依存 してしまいます。

import os
import pytest


@pytest.fixture(autouse=True)
def setup():
    os.environ["TEST_CODE"] = "PY_TEST"


def test_main():
    assert os.environ["TEST_CODE"] == "PY_TEST"

正しい方法

テストが実行順序に依存しないよう、各テスト終了後は設定済みの環境変数をクリアする必要があります。これによってテストの独立性が高くなり、他のテストに影響を受けなくなります。これには次のようにpatchとyieldを組み合わせれば実現できます。

import os
from unittest.mock import patch
import pytest


@pytest.fixture(autouse=True)
def setup():
    with patch.dict("os.environ", {"TEST_CODE": "PY_TEST"}):
        yield


def test_main():
    assert os.environ["TEST_CODE"] == "PY_TEST"