Single parameter in PyTest - Build an Automation Script

Test function with single parameter using pytest

Preconditions (2)

Step 1: Create a test function named 'test_is_even' that takes one parameter 'number'

Step 2: Use pytest's @pytest.mark.parametrize decorator to pass the values 2, 3, and 4 to 'number'

Step 3: Inside the test, call 'is_even(number)'

Step 4: Assert that the result is True for 2 and 4, and False for 3

✅ Expected Result: The test passes for inputs 2 and 4 (returns True) and fails for input 3 (returns False)

Automation Requirements - pytest

Assertions Needed:

Assert that is_even(2) returns True

Assert that is_even(3) returns False

Assert that is_even(4) returns True

Best Practices:

Use @pytest.mark.parametrize for single parameter testing

Keep test functions simple and focused

Use clear assertion messages if needed

Automated Solution

PyTest

import pytest

def is_even(number: int) -> bool:
    return number % 2 == 0

@pytest.mark.parametrize('number,expected', [
    (2, True),
    (3, False),
    (4, True)
])
def test_is_even(number, expected):
    result = is_even(number)
    assert result == expected, f'Expected is_even({number}) to be {expected} but got {result}'

This code defines a simple function is_even that checks if a number is even.

The test function test_is_even uses @pytest.mark.parametrize to run the test three times with different inputs and expected results.

Each time, it calls is_even with the parameter and asserts the output matches the expected value.

This approach keeps tests clean and easy to extend with more inputs.

Common Mistakes - 3 Pitfalls

Not using @pytest.mark.parametrize and writing separate test functions for each input

Forgetting to include the expected result in the parameter list

Using print statements instead of assertions

Bonus Challenge

Now add data-driven testing with 3 different inputs including negative and zero values

Show Hint

Practice

1. What is the purpose of using a single parameter in a pytest test function?

easy

A. To define multiple test functions at once

B. To make the test function run only once

C. To skip the test function during execution

D. To pass different values to the test function for repeated testing

Solution

1. Step 1: Understand pytest parameterization

   Pytest allows passing parameters to test functions to run the same test with different inputs.

2. Step 2: Role of a single parameter

   A single parameter lets you supply multiple values to the test, so it runs multiple times with each value.

3. Final Answer:

   To pass different values to the test function for repeated testing -> Option D

4. Quick Check:

   Single parameter = multiple test runs [OK]

Hint: Single parameter means testing with different inputs [OK]

Common Mistakes:

• Thinking single parameter defines multiple tests
• Confusing parameter with skipping tests
• Assuming single parameter runs test once only

2. Which of the following is the correct syntax to use a single parameter with pytest's @pytest.mark.parametrize decorator?

easy

A. @pytest.mark.parametrize('num', 1, 2, 3)

B. @pytest.mark.parametrize(num, [1, 2, 3])

C. @pytest.mark.parametrize('num', [1, 2, 3])

D. @pytest.mark.parametrize(['num'], 1, 2, 3)

Solution

1. Step 1: Check parameter name format

   The parameter name must be a string inside quotes, e.g., 'num'.

2. Step 2: Check values format

   The values must be inside a list or iterable, e.g., [1, 2, 3].

3. Final Answer:

   @pytest.mark.parametrize('num', [1, 2, 3]) -> Option C

4. Quick Check:

   Parametrize syntax = string + list [OK]

Hint: Parameter name in quotes, values in list brackets [OK]

Common Mistakes:

• Not quoting the parameter name
• Passing values without list brackets
• Using wrong argument structure

3. What will be the output when running this pytest code?

import pytest

@pytest.mark.parametrize('x', [2, 4])
def test_double(x):
    assert x * 2 == 4

medium

A. First test passes, second test fails

B. Both tests fail

C. First test fails, second test passes

D. Both tests pass

Solution

1. Step 1: Evaluate first test with x=2

   2 * 2 = 4, assertion is true, test passes.

2. Step 2: Evaluate second test with x=4

   4 * 2 = 8, assertion 8 == 4 is false, test fails.

3. Final Answer:

   First test passes, second test fails -> Option A

4. Quick Check:

   2*2=4 pass, 4*2=8 fail [OK]

Hint: Check each input separately for assertion [OK]

Common Mistakes:

• Assuming all tests pass without checking values
• Mixing up which input causes failure
• Ignoring assertion logic

4. Identify the error in this pytest code using a single parameter:

import pytest

@pytest.mark.parametrize('val', 1, 2, 3)
def test_positive(val):
    assert val > 0

medium

A. Parameter name should not be in quotes

B. Missing list brackets around parameter values

C. Test function must have two parameters

D. Assertion should use == instead of >

Solution

1. Step 1: Check parametrize values format

   Values must be inside a list or iterable, e.g., [1, 2, 3].

2. Step 2: Identify missing brackets

   Code passes values as separate arguments instead of a single list, causing syntax error.

3. Final Answer:

   Missing list brackets around parameter values -> Option B

4. Quick Check:

   Values need list brackets [OK]

Hint: Values must be in a list or tuple [OK]

Common Mistakes:

• Passing values without brackets
• Removing quotes from parameter name
• Wrong assertion operator

5. Given this test function, how can you modify it to test multiple string inputs using a single parameter?

def test_is_uppercase(s):
    assert s.isupper()

hard

A. Add @pytest.mark.parametrize('s', ['HELLO', 'WORLD']) above the function

B. Change function to accept two parameters s and t

C. Use a for loop inside the test to iterate over strings

D. Call test_is_uppercase multiple times manually

Solution

1. Step 1: Understand parameterizing tests

   Using @pytest.mark.parametrize allows running the test multiple times with different inputs.

2. Step 2: Apply parametrize decorator with string list

   Adding @pytest.mark.parametrize('s', ['HELLO', 'WORLD']) runs test_is_uppercase twice with each string.

3. Final Answer:

   Add @pytest.mark.parametrize('s', ['HELLO', 'WORLD']) above the function -> Option A

4. Quick Check:

   Parametrize decorator runs test with multiple inputs [OK]

Hint: Use @pytest.mark.parametrize with list of inputs [OK]

Common Mistakes:

• Trying to pass multiple parameters instead of one
• Using loops inside test instead of parametrize
• Calling test manually instead of using pytest features