データ駆動テスト入門: pytestのparametrizeで複数パターンを一括検証

# file: calc.py
def is_even(n: int) -> bool:
    """偶数ならTrueを返します。"""
    return n % 2 == 0

# file: tests/test_even_parametrize.py
import pytest
from calc import is_even

# nに複数の値を渡し、同じアサーションを繰り返す
@pytest.mark.parametrize("n", [0, 2, 10, 42])  # 必ず偶数だけを並べています
def test_is_even_returns_true_for_even_numbers(n):
    # 単一引数の例では、1つの性質(偶数→True)だけに集中します
    assert is_even(n) is True

# file: calc.py
def add(a: float, b: float) -> float:
    """2数の和を返します。"""
    return a + b


# file: tests/test_add_parametrize.py
import pytest
from calc import add

# "a,b,expected"の順で引数名を定義し、各データをタプルで与えます
@pytest.mark.parametrize(
    "a,b,expected",
    [
        (1, 2, 3),
        (-1, 5, 4),
        (0, 0, 0),
        (2.5, 0.5, 3.0),
    ],
)
def test_add_returns_expected(a, b, expected):
    # Arrange(準備)はparametrizeで済み、Act(実行)とAssert(検証)に集中できます
    assert add(a, b) == expected

# file: tests/test_add_with_ids.py
import pytest
from calc import add

@pytest.mark.parametrize(
    "a,b,expected",
    [
        (1, 2, 3),
        (-1, 5, 4),
        (0, 0, 0),
    ],
    ids=["small-positives", "with-negative", "all-zero"],
)
def test_add_readable_ids(a, b, expected):
    assert add(a, b) == expected

# file: tests/test_divide_param_marks.py
import pytest

def divide(a: float, b: float) -> float:
    """単純な割り算(ゼロ除算は現状未対応の想定)。"""
    return a / b

@pytest.mark.parametrize(
    "a,b,expected",
    [
        (6, 3, 2),
        (5, 2, 2.5),
        # 未対応の0除算は、意図的にxfailで印を付けておきます
        pytest.param(1, 0, None, marks=pytest.mark.xfail(reason="未対応: 0除算"), id="zero-division"),
        # デモ目的で一時的にスキップ
        pytest.param(10, 5, 2, marks=pytest.mark.skip(reason="デモ: スキップ"), id="skip-demo"),
    ],
)
def test_divide(a, b, expected):
    # 0除算ケースは例外で失敗しますが、xfail指定により「期待された失敗」として扱われます
    assert divide(a, b) == expected

# file: text_utils.py
def is_blank(s: str) -> bool:
    """空文字または空白のみをTrueにします。"""
    return s.strip() == ""


# file: tests/test_edge_cases.py
import pytest
from calc import is_even
from text_utils import is_blank

# 数値のエッジケース: 0は偶数
@pytest.mark.parametrize("n", [0, 2, 100])
def test_is_even_including_zero(n):
    assert is_even(n) is True

# 文字列のエッジケース: 空文字と空白のみ
@pytest.mark.parametrize(
    "s,expected",
    [
        ("", True),
        ("   ", True),
        ("abc", False),
    ],
    ids=["empty", "spaces-only", "normal"],
)
def test_is_blank(s, expected):
    assert is_blank(s) is expected

# 悪い例: 並びが不一致
@pytest.mark.parametrize("a,b,expected", [
    (1, 2, 3),
    # 本来は (a, b, expected) だが (expected, a, b) になっている
    (3, 1, 2),
])
def test_add_mismatch(a, b, expected):
    ...

# file: tests/test_bad_arity.py
import pytest

@pytest.mark.parametrize("a,b,expected", [
    (1, 2),  # expectedが欠けている
])
def test_arity(a, b, expected):
    ...

# file: tests/test_dict_param.py
import pytest

@pytest.mark.parametrize(
    "user",
    [
        {"name": "Alice", "age": 20},
        {"name": "Bob", "age": 0},
    ],
    ids=["alice", "bob"],
)
def test_user_age_nonnegative(user):
    assert user["age"] >= 0

# file: tests/test_strings_param.py
import pytest

def greet(name: str, suffix: str) -> str:
    return f"Hello, {name}{suffix}"

@pytest.mark.parametrize(
    "name,suffix,expected",
    [
        ("World", "!", "Hello, World!"),
        ("Python", " 3.12", "Hello, Python 3.12"),
    ],
)
def test_greet(name, suffix, expected):
    assert greet(name, suffix) == expected