Modern Testing with pytest & Parametrization

While Python's built-in unittest is solid, modern engineering teams worldwide prefer pytest because:

1. Zero Boilerplate: No need to create classes or remember self.assertEqual().
2. Standard assert Keyword: You write normal, readable Python assertions (assert a == b).
3. Parametrized Testing: Run 20 test variations using one single 3-line function.

---

1. Installation

# Using standard pip
pip install pytest

# Or using high-speed uv
uv add pytest

---

2. Writing Your First Pytest Test

With pytest, you simply create a Python file whose name starts with test_ and write normal functions:

# test_math_operations.py

def add(x, y):
    return x + y

def is_prime(n):
    if n <= 1:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True

# Pytest test functions:
def test_add_numbers():
    assert add(10, 20) == 30
    assert add(-1, 1) == 0

def test_prime_numbers():
    assert is_prime(7) is True
    assert is_prime(4) is False

Running Pytest

Just type pytest in your terminal:

pytest -v

---

3. Parametrized Tests with @pytest.mark.parametrize

Instead of writing 10 separate test functions to test 10 different coupon codes or passwords, pass a table of inputs and expected outputs directly:

import pytest

def validate_age_for_voting(age):
    return age >= 18

# Runs the test 4 separate times with 4 different inputs automatically!
@pytest.mark.parametrize("input_age, expected_result", [
    (21, True),
    (18, True),
    (15, False),
    (0, False),
    (99, True)
])
def test_voting_eligibility(input_age, expected_result):
    assert validate_age_for_voting(input_age) == expected_result

---

4. Pytest Fixtures (@pytest.fixture)

Fixtures provide a clean, dependency-injection style way to supply mock databases, API clients, or test data to your test functions:

import pytest

@pytest.fixture
def sample_user_profile():
    # Returns a fresh user dictionary for tests
    return {
        "user_id": 402,
        "name": "Sai Kumar",
        "role": "admin",
        "is_active": True
    }

def test_user_is_admin(sample_user_profile):
    assert sample_user_profile["role"] == "admin"

def test_user_is_active(sample_user_profile):
    assert sample_user_profile["is_active"] is True

---

Quick Summary

Feature	Built-in unittest	Modern pytest
Requires Classes?	Yes (unittest.TestCase)	No (Plain standalone functions)
Assertions	self.assertEqual(a, b)	Standard Python assert a == b
Parametrization	Complex loops / subtests	Expressive @pytest.mark.parametrize
Fixtures	setUp / tearDown OOP	Dependency injection via @pytest.fixture
Installation	Built-in	pip install pytest

---

What's Next?

🎉 Congratulations! You have completed Part 3: Advanced Python Programming!

You now possess the foundational engineering tools required for modern software development. Next up is Part 4: AI & ML Engineering!

Part 4: AI & ML Engineering

Level up with NumPy, Pandas, Data Visualizations, Machine Learning, and LLM Applications