Fixture & Parametrize

เมื่อเทสต์หลายตัวต้องการข้อมูลเตรียมเหมือนกัน หรืออยากทดสอบฟังก์ชันเดียวด้วยหลายชุด input pytest มีเครื่องมือ fixture และ parametrize ที่ลดโค้ดซ้ำได้มาก

fixture — เตรียมของให้เทสต์

fixture คือฟังก์ชันที่เตรียมข้อมูล/สถานะให้เทสต์ใช้ร่วมกัน เทสต์ไหนต้องการก็รับชื่อ fixture เป็นพารามิเตอร์ pytest จะเรียกให้เอง

import pytest

@pytest.fixture
def sample_users():
    return [
        {"name": "Aph", "age": 25},
        {"name": "Bee", "age": 30},
    ]

def test_count(sample_users):       # รับชื่อ fixture
    assert len(sample_users) == 2

def test_first_name(sample_users):  # ใช้ข้อมูลชุดเดียวกัน
    assert sample_users[0]["name"] == "Aph"

parametrize — เทสต์หลายเคสด้วยโค้ดเดียว

แทนที่จะเขียนเทสต์ซ้ำหลายตัวสำหรับหลาย input ใช้ parametrize ใส่ชุด (input, expected) แล้ว pytest รันให้ครบทุกชุด

import pytest

def is_even(n):
    return n % 2 == 0

@pytest.mark.parametrize("number, expected", [
    (2, True),
    (3, False),
    (0, True),
    (-4, True),
    (7, False),
])
def test_is_even(number, expected):
    assert is_even(number) == expected
# รัน 5 เคส จากเทสต์เดียว

pytest.raises — ทดสอบว่า raise error

บางครั้งพฤติกรรมที่ถูกต้องคือ "ต้อง raise exception" ใช้ pytest.raises ตรวจว่าโค้ด raise ตามคาด

import pytest

def divide(a, b):
    return a / b

def test_divide_by_zero():
    with pytest.raises(ZeroDivisionError):
        divide(10, 0)        # ถ้าไม่ raise = เทสต์พัง

สรุปหัวข้อนี้

• fixture เตรียมข้อมูล/สถานะให้เทสต์ใช้ร่วม — รับเป็นพารามิเตอร์
• parametrize รันเทสต์เดียวด้วยหลายชุด (input, expected)
• pytest.raises ตรวจว่าโค้ด raise exception ตามคาด
• ทั้งหมดช่วยลดโค้ดซ้ำและครอบคลุมเคสมากขึ้น