Branch coverage in PyTest - Build an Automation Script

Test branch coverage for a function with conditional branches

Preconditions (2)

Step 1: Create a function that returns 'Positive' if input is greater than 0, 'Zero' if input is 0, and 'Negative' if input is less than 0

Step 2: Write test cases to cover all branches of the function

Step 3: Run pytest with coverage to verify branch coverage

✅ Expected Result: All branches of the function are executed and verified by tests, resulting in 100% branch coverage

Automation Requirements - pytest

Assertions Needed:

Assert function returns 'Positive' for input > 0

Assert function returns 'Zero' for input == 0

Assert function returns 'Negative' for input < 0

Best Practices:

Use clear and descriptive test function names

Use pytest parametrize to cover multiple inputs

Use coverage.py with branch coverage enabled

Keep tests independent and simple

Automated Solution

PyTest

import pytest

def check_number_sign(num: int) -> str:
    if num > 0:
        return 'Positive'
    elif num == 0:
        return 'Zero'
    else:
        return 'Negative'

@pytest.mark.parametrize('input_value, expected_output', [
    (10, 'Positive'),
    (0, 'Zero'),
    (-5, 'Negative')
])
def test_check_number_sign(input_value, expected_output):
    assert check_number_sign(input_value) == expected_output

The function check_number_sign has three branches: one for positive numbers, one for zero, and one for negative numbers.

The test uses pytest.mark.parametrize to run the same test logic with three different inputs, covering all branches.

This ensures 100% branch coverage because each conditional path is tested.

Assertions check that the function returns the correct string for each input.

Using parametrize keeps the test code clean and easy to maintain.

Common Mistakes - 3 Pitfalls

Writing only one test case that covers a single branch

Using print statements instead of assertions

Not enabling branch coverage in coverage.py

Bonus Challenge

Now add data-driven testing with 3 different inputs using pytest parametrize

Show Hint

Practice

1.

What does branch coverage measure in pytest testing?

easy

A. It checks if all decision paths in the code are tested.

B. It measures how many lines of code are executed.

C. It counts the number of test cases written.

D. It verifies the syntax correctness of the test code.

Solution

1. Step 1: Understand branch coverage concept

   Branch coverage ensures every possible path in decision points (like if-else) is tested.

2. Step 2: Compare with other coverage types

   Line coverage counts executed lines, but branch coverage focuses on decision paths.

3. Final Answer:

   It checks if all decision paths in the code are tested. -> Option A

4. Quick Check:

   Branch coverage = all decision paths tested [OK]

Hint: Branch coverage tests all if-else paths, not just lines [OK]

Common Mistakes:

• Confusing branch coverage with line coverage
• Thinking it counts test cases
• Assuming it checks syntax errors

2.

Which pytest command option enables branch coverage measurement?

easy

A. pytest --cov --cov-branch

B. pytest --branch-coverage

C. pytest --cov-branch

D. pytest --coverage-branch

Solution

1. Step 1: Recall pytest coverage options

   The option --cov enables coverage measurement, and --cov-branch adds branch coverage.

2. Step 2: Identify correct combined command

   Both options must be used together as pytest --cov --cov-branch to measure branch coverage.

3. Final Answer:

   pytest --cov --cov-branch -> Option A

4. Quick Check:

   Use --cov with --cov-branch for branch coverage [OK]

Hint: Use both --cov and --cov-branch together [OK]

Common Mistakes:

• Using only --cov-branch without --cov
• Typing incorrect option names
• Assuming branch coverage is default

3.

Given this code snippet tested with pytest and branch coverage enabled, what is the branch coverage result?

def check_num(x):
    if x > 0:
        return "Positive"
    else:
        return "Non-positive"

# Test cases:
assert check_num(5) == "Positive"

medium

A. 100% branch coverage

B. 75% branch coverage

C. 0% branch coverage

D. 50% branch coverage

Solution

1. Step 1: Identify branches in the code

   The function has two branches: one for x > 0 and one for else (x ≤ 0).

2. Step 2: Check which branches are tested

   The test calls check_num(5), which covers only the x > 0 branch, missing the else branch.

3. Final Answer:

   50% branch coverage -> Option D

4. Quick Check:

   One branch tested out of two = 50% [OK]

Hint: Test all if-else paths to get 100% branch coverage [OK]

Common Mistakes:

• Assuming one test covers all branches
• Confusing line coverage with branch coverage
• Ignoring else branch testing

4.

Identify the error in this pytest test that aims for full branch coverage:

def classify(n):
    if n > 10:
        return "High"
    elif n > 5:
        return "Medium"
    else:
        return "Low"

def test_classify():
    assert classify(12) == "High"
    assert classify(7) == "Medium"

medium

A. Syntax error in the test function.

B. Incorrect expected values in assertions.

C. Missing test for the else branch (n ≤ 5).

D. The function classify has no return statement.

Solution

1. Step 1: Analyze branches in classify()

   There are three branches: n > 10, n > 5, and else (n ≤ 5).

2. Step 2: Check test coverage of branches

   Tests cover n=12 (High) and n=7 (Medium), but no test covers n ≤ 5 (Low) branch.

3. Final Answer:

   Missing test for the else branch (n ≤ 5). -> Option C

4. Quick Check:

   All branches need tests for full coverage [OK]

Hint: Test all if, elif, else branches to avoid missing coverage [OK]

Common Mistakes:

• Assuming two tests cover all branches
• Looking for syntax errors where none exist
• Ignoring else branch testing

5.

You have this function. Which set of test inputs achieves 100% branch coverage?

def analyze_score(score):
    if score >= 90:
        return "A"
    elif score >= 75:
        return "B"
    else:
        return "C"

hard

A. [90, 85, 76]

B. [95, 80, 70]

C. [91, 74, 60]

D. [89, 84, 73]

Solution

1. Step 1: Identify all branches

   Branches: score ≥ 90 ("A"), score ≥ 75 ("B"), else "C".

2. Step 2: Check which inputs cover all branches

   Input 95 covers "A" branch; 80 covers score ≥ 75 ("B"); 70 covers else "C" branch. This covers all branches.

3. Final Answer:

   [95, 80, 70] -> Option B

4. Quick Check:

   Test inputs cover all if-elif-else paths [OK]

Hint: Pick inputs hitting each if, elif, else branch [OK]

Common Mistakes:

• Missing coverage for one of the branches
• Choosing inputs that skip else branch
• Assuming close values cover all branches