Python：pytestによるテスト

出力 == テスト成功

assert 条件 と書くだけ。条件がFalseならその場でエラーになり、下の行は実行されない。

def add(a, b):
    return a + b


assert add(3, 7) == 10
assert add(-1, 1) == 0

print("テスト成功")

assert はPythonの標準機能で、pytestが無くても使える。「ここは必ずこうなるはず」を書き残しておくのがテストの出発点。

解答 == B

決まった接頭辞で始める必要がある。

B

前に付けるのがルール（add_test では見つけてもらえない）。ファイル名も test_calc.py のように test_ で始める。

出力 == AssertionError が発生しました

try / except AssertionError: で囲む。PY10の例外処理と同じ書き方。

try:
    assert 1 + 1 == 3
except AssertionError:
    print("AssertionError が発生しました")

assert が失敗すると AssertionError という例外が出る。pytestはこれを検知して「失敗したテスト」として集計している。

def double(n):
    return n * 2


____ double(5) == 10
print("OK")

出力 == OK

「〜であることを表明する」という意味の英単語。

assert

assert の後ろにはTrue になるはずの条件を書く。== を = と書き間違えると構文エラーになるので注意。

解答 == B

テストは書いた瞬間より、あとで効いてくる。

B

テストの本当の価値は「安心して直せる」こと。テストが無いコードは、怖くて誰も手を入れられなくなっていく。

pytest 実行で 2 passed

ファイル名を test_judge.py にして、def test_〜(): の中で assert する。実行はターミナルで pytest。

def judge(score):
    if score >= 60:
        return "合格"
    return "不合格"


def test_60点はちょうど合格():
    assert judge(60) == "合格"


def test_59点は不合格():
    assert judge(59) == "不合格"

境界の両側を必ず両方テストする。>= を > と書き間違えるバグは、59点のテストだけでは見つからず60点のテストで初めて捕まる。

import pytest


def divide(a, b):
    if b == 0:
        raise ValueError("0では割れません")
    return a / b


def test_ゼロ除算で例外():
    with pytest.____(ValueError):
        divide(10, 0)

pytest 実行で 1 passed

「（例外が）発生する」という意味のメソッド。

raises

with ブロックの中で例外が出れば成功、出なければ失敗という逆向きの判定になる。異常系のテストはこの形で書く。

解答 == B

parameter（引数）を変えながら、が語源。

B

1件が1テストとして数えられるので、3パターン指定すれば「3 passed」になる。境界値をまとめて並べるのに最適。

pytest 実行で 3 passed

異常系は with pytest.raises(ValueError): の中で呼ぶ。import pytest を忘れずに。

import pytest


def check_age(age):
    if age < 0 or age > 150:
        raise ValueError("年齢が不正です")
    return "OK"


def test_正常な年齢():
    assert check_age(30) == "OK"


def test_マイナスは例外():
    with pytest.raises(ValueError):
        check_age(-1)


def test_上限超えは例外():
    with pytest.raises(ValueError):
        check_age(200)

正常系と異常系をセットで書くのが基本。PY10で書いた raise が、ここで初めて「テストできるもの」として意味を持つ。

pytest 実行で 4 passed

@pytest.mark.parametrize("price, expected", [(0, 0), ...]) を付け、関数の引数に price と expected を受け取る。

import pytest


def with_tax(price, rate=0.1):
    return int(price * (1 + rate))


@pytest.mark.parametrize("price, expected", [(0, 0), (100, 110), (1000, 1100), (999, 1098)])
def test_税込計算(price, expected):
    assert with_tax(price) == expected

同じ assert を4回コピペする代わりに1つで済む。パターンを1行足すだけでテストが増やせるので、あとから気づいた条件も気軽に追加できる。

pytest 実行で 4 passed

テスト名で観点が分かるようにする（test_端数は切り捨て など）。④は with_tax(1000, 0.08) のように第2引数を渡す。

def with_tax(price, rate=0.1):
    return int(price * (1 + rate))


def test_通常の税込計算():
    assert with_tax(1000) == 1100


def test_端数は切り捨て():
    assert with_tax(105) == 115


def test_ゼロ円はゼロ円():
    assert with_tax(0) == 0


def test_税率を指定できる():
    assert with_tax(1000, 0.08) == 1080

105円の税込は115.5円だが int() で115円になる。この「切り捨てる」という仕様をテストに書き残すことで、あとから四捨五入に変えたい人が仕様変更に気づける。

pytest 実行で 4 passed

④が境界値のテスト。8文字ちょうどは通るのが正しい仕様なので、例外にならないことを確かめる。

import pytest


def validate(name, password):
    if not name:
        raise ValueError("名前を入力してください")

    if len(password) < 8:
        raise ValueError("パスワードは8文字以上にしてください")

    return "OK"


def test_正常な入力はOK():
    assert validate("田中", "abcd1234") == "OK"


def test_名前が空なら例外():
    with pytest.raises(ValueError):
        validate("", "abcd1234")


def test_パスワードが短いと例外():
    with pytest.raises(ValueError):
        validate("田中", "abc123")


def test_ちょうど8文字はOK():
    assert validate("田中", "abcd1234") == "OK"

「8文字以上」の境界は7文字と8文字。< を <= と書き間違えると8文字が弾かれるが、そのバグは④のテストでしか捕まらない。

pytest 実行で 3 passed

テストファイルの先頭で from calc import add, divide と書けば、別ファイルの関数を呼べる（PY12のモジュール）。

import pytest

from calc import add, divide


def test_足し算():
    assert add(3, 7) == 10


def test_割り算():
    assert divide(10, 2) == 5


def test_ゼロ除算は例外():
    with pytest.raises(ValueError):
        divide(10, 0)

本体とテストはファイルを分けるのが実務の形。calc.py は納品するコード、test_calc.py は開発用。import で結ぶだけなので、PY12のモジュールがそのまま活きる。

pytest 実行で 6 passed

各しきい値の「ちょうど」と「1つ下」をペアで並べるのがコツ。80と79、60と59が対になっている。

import pytest


def judge(score):
    if score >= 80:
        return "A"
    if score >= 60:
        return "B"
    return "C"


@pytest.mark.parametrize("score, expected", [
    (100, "A"),
    (80, "A"),
    (79, "B"),
    (60, "B"),
    (59, "C"),
    (0, "C"),
])
def test_ランク判定(score, expected):
    assert judge(score) == expected

しきい値ごとに「ちょうど」と「1つ下」を並べる——これだけで境界のバグはほぼ全部捕まる。PY04で学んだ「条件は厳しい順に」が守られているかも、このテストが保証してくれる。

pytest 実行で 4 passed

③がこのバグの再発防止テスト。テスト名にその意図を書き残すと、あとから読む人に理由が伝わる。

def final_price(price, is_member=False, coupon=False):
    if is_member:
        price = int(price * 0.9)

    if coupon:
        price -= 500

    if price < 0:
        price = 0

    return price


def test_会員は10パーセント引き():
    assert final_price(12000, is_member=True) == 10800


def test_クーポンで500円引き():
    assert final_price(1000, coupon=True) == 500


def test_支払額はマイナスにならない():
    assert final_price(400, coupon=True) == 0


def test_割引なしなら定価():
    assert final_price(1000) == 1000

バグを直したら、そのバグを再現するテストを必ず追加する。これを回帰テストと呼ぶ。将来この行を誰かが消してしまっても、テストが即座に知らせてくれる。テストが最も価値を発揮する場面。