Why Conditional parametrize in PyTest? - Purpose & Use Cases

The Big Idea

What if your tests could smartly pick only the data they really need to check, all by themselves?

The Scenario

Imagine you have a test that needs to run with different sets of data, but only some data should be tested under certain conditions. Manually writing separate tests or adding many if-else checks inside tests can get messy fast.

The Problem

Manually managing which data to test wastes time and causes mistakes. You might forget to test some cases or run unnecessary ones. It's hard to keep track, and your test code becomes cluttered and confusing.

The Solution

Conditional parametrize lets you tell pytest exactly which data to use based on conditions. This keeps your tests clean, runs only relevant cases, and saves you from writing repetitive code or complex logic inside tests.

Before vs After

✗ Before

def test_example(data):
    if condition:
        assert data in allowed_values
    else:
        assert data in other_values

✓ After

@pytest.mark.parametrize('data', [1, 2, 3] if condition else [4, 5])
def test_example(data):
    assert data in expected_values

What It Enables

It enables running just the right tests automatically, making your testing faster, clearer, and less error-prone.

Real Life Example

For example, testing a login feature with different user roles: only admin users get tested with admin-only data, while regular users get tested with their own data sets, all controlled neatly by conditional parametrize.

Key Takeaways

Manual test data handling is slow and error-prone.

Conditional parametrize runs tests only with relevant data.

This keeps tests clean, efficient, and easy to maintain.

Practice

1. What is the main purpose of using pytest.param with skipif inside @pytest.mark.parametrize?

easy

A. To mark tests as expected failures

B. To run all test cases regardless of conditions

C. To conditionally skip specific test cases based on environment or conditions

D. To parametrize tests without any conditions

Solution

1. Step 1: Understand pytest.param with skipif

   This syntax allows marking individual parameters to be skipped if a condition is true.

2. Step 2: Purpose of conditional skipping

   This helps run only relevant tests in certain environments, skipping others automatically.

3. Final Answer:

   To conditionally skip specific test cases based on environment or conditions -> Option C

4. Quick Check:

   Conditional skipping = skipif with pytest.param [OK]

Hint: Use pytest.param with skipif to skip tests conditionally [OK]

Common Mistakes:

• Thinking skipif skips all tests, not just specific params
• Confusing skipif with xfail
• Not using pytest.param inside parametrize

2. Which of the following is the correct syntax to skip a parameter when Python version is less than 3.8 using pytest.mark.parametrize?

easy

A. @pytest.mark.parametrize('x', [pytest.param(1, marks=pytest.mark.skipif(sys.version_info < (3,8), reason='Old Python'))])

B. @pytest.mark.parametrize('x', [pytest.param(1, skipif=sys.version_info < (3,8))])

C. @pytest.mark.parametrize('x', [pytest.param(1, marks=skipif(sys.version_info < (3,8)))])

D. @pytest.mark.parametrize('x', [pytest.param(1, marks=pytest.skipif(sys.version_info < (3,8)))])

Solution

1. Step 1: Identify correct usage of skipif mark

   The skipif mark must be used inside pytest.mark.skipif with a reason string.

2. Step 2: Check syntax correctness

   @pytest.mark.parametrize('x', [pytest.param(1, marks=pytest.mark.skipif(sys.version_info < (3,8), reason='Old Python'))]) correctly uses pytest.mark.skipif with condition and reason inside marks parameter.

3. Final Answer:

   @pytest.mark.parametrize('x', [pytest.param(1, marks=pytest.mark.skipif(sys.version_info < (3,8), reason='Old Python'))]) -> Option A

4. Quick Check:

   Use pytest.mark.skipif with marks and reason [OK]

Hint: Use marks=pytest.mark.skipif(condition, reason='') inside pytest.param [OK]

Common Mistakes:

• Using skipif without pytest.mark
• Omitting the reason argument
• Passing skipif as a direct argument instead of marks

3. Given the code below, what will be the output when running pytest on Python 3.7?

import sys
import pytest

@pytest.mark.parametrize('num', [
    pytest.param(1, marks=pytest.mark.skipif(sys.version_info < (3,8), reason='Skip on old Python')),
    2,
    3
])
def test_numbers(num):
    assert num > 0

medium

A. All three tests run and pass

B. Test with num=1 is skipped; tests with num=2 and num=3 run and pass

C. All tests are skipped

D. Test with num=1 fails due to skipif syntax error

Solution

1. Step 1: Analyze skipif condition with Python 3.7

   sys.version_info < (3,8) is True for Python 3.7, so param with num=1 is skipped.

2. Step 2: Check other parameters

   Parameters 2 and 3 have no skip marks, so their tests run and pass the assertion.

3. Final Answer:

   Test with num=1 is skipped; tests with num=2 and num=3 run and pass -> Option B

4. Quick Check:

   skipif True skips param; others run [OK]

Hint: Check skipif condition against environment to predict skipped tests [OK]

Common Mistakes:

• Assuming all tests run regardless of skipif
• Confusing skipif with xfail
• Thinking skipif causes syntax error

4. Identify the error in the following code snippet that uses conditional parametrize:

import pytest

@pytest.mark.parametrize('val', [
    pytest.param(10, marks=pytest.mark.skipif('sys.version_info < (3,8)', reason='Old Python')),
    20
])
def test_val(val):
    assert val > 0

medium

A. The skipif condition is a string instead of a boolean expression

B. Missing import of sys module

C. pytest.param cannot be used inside parametrize

D. The reason argument is missing in skipif

Solution

1. Step 1: Check skipif condition type

   The condition is passed as a string, but skipif expects a boolean expression, not a string.

2. Step 2: Confirm other parts

   sys module is not imported, but the main error is the string condition. pytest.param usage and reason argument are correct.

3. Final Answer:

   The skipif condition is a string instead of a boolean expression -> Option A

4. Quick Check:

   skipif condition must be boolean, not string [OK]

Hint: skipif condition must be a boolean expression, not a string [OK]

Common Mistakes:

• Passing condition as string instead of boolean
• Ignoring missing sys import (not main error here)
• Misusing pytest.param outside parametrize

5. You want to run a test with parameters mode set to 'fast', 'slow', and 'medium'. However, you want to skip the 'slow' mode test only if an environment variable SKIP_SLOW is set to '1'. Which is the correct way to write this conditional parametrize in pytest?

hard

A. import os
@pytest.mark.parametrize('mode', [
'fast',
pytest.param('slow', marks=pytest.mark.skipif('os.getenv("SKIP_SLOW") == "1"')),
'medium'
])

B. import os
@pytest.mark.parametrize('mode', [
'fast',
pytest.param('slow', skipif=os.getenv('SKIP_SLOW') == '1'),
'medium'
])

C. import os
@pytest.mark.parametrize('mode', [
'fast',
pytest.param('slow', marks=pytest.skipif(os.getenv('SKIP_SLOW') == '1')),
'medium'
])

D. import os
@pytest.mark.parametrize('mode', [
'fast',
pytest.param('slow', marks=pytest.mark.skipif(os.getenv('SKIP_SLOW') == '1', reason='Skip slow mode')),
'medium'
])

Solution

1. Step 1: Use os.getenv to check environment variable

   os.getenv('SKIP_SLOW') returns the value of the environment variable as a string.

2. Step 2: Apply skipif mark correctly

   Use pytest.mark.skipif with a boolean condition and a reason inside marks parameter of pytest.param.

3. Step 3: Verify syntax correctness

   import os @pytest.mark.parametrize('mode', [ 'fast', pytest.param('slow', marks=pytest.mark.skipif(os.getenv('SKIP_SLOW') == '1', reason='Skip slow mode')), 'medium' ]) correctly uses marks=pytest.mark.skipif(condition, reason='...') and imports os.

4. Final Answer:

   import os @pytest.mark.parametrize('mode', [ 'fast', pytest.param('slow', marks=pytest.mark.skipif(os.getenv('SKIP_SLOW') == '1', reason='Skip slow mode')), 'medium' ]) -> Option D

5. Quick Check:

   Use marks=pytest.mark.skipif with env check and reason [OK]

Hint: Use os.getenv and marks=pytest.mark.skipif with reason to skip params [OK]

Common Mistakes:

• Passing skipif as direct argument instead of marks
• Using string condition instead of boolean
• Omitting reason in skipif mark