capsys for stdout/stderr in PyTest - Build an Automation Script

Capture and verify printed output using capsys in pytest

Preconditions (2)

Step 1: Write a test function that calls the target function which prints to stdout and stderr

Step 2: Use the capsys fixture to capture the output

Step 3: After calling the function, use capsys.readouterr() to get the captured stdout and stderr

Step 4: Assert that the captured stdout contains the expected printed text

Step 5: Assert that the captured stderr contains the expected error text

✅ Expected Result: The test passes if the captured stdout and stderr match the expected printed messages

Automation Requirements - pytest

Assertions Needed:

Assert captured stdout equals expected string

Assert captured stderr equals expected string

Best Practices:

Use the capsys fixture parameter in the test function

Call capsys.readouterr() only once after the function call

Avoid printing inside the test except for debugging

Keep assertions clear and specific

Automated Solution

PyTest

import sys

def greet():
    print("Hello, world!")
    print("Warning: something minor happened", file=sys.stderr)


def test_greet_output(capsys):
    greet()
    captured = capsys.readouterr()
    assert captured.out == "Hello, world!\n"
    assert captured.err == "Warning: something minor happened\n"

The greet function prints a message to standard output and another to standard error.

The test function test_greet_output uses the capsys fixture to capture these outputs.

After calling greet(), capsys.readouterr() returns an object with out and err strings.

Assertions check that the captured output matches exactly what greet printed, including newline characters.

This ensures the printed messages are correctly captured and verified.

Common Mistakes - 3 Pitfalls

Calling capsys.readouterr() multiple times after the function call

Not including newline characters in expected strings

Printing inside the test function for debugging without disabling

Bonus Challenge

Now add tests for a function that prints different messages based on input, capturing and asserting stdout and stderr for three different inputs.

Show Hint

Practice

1. What is the main purpose of using capsys in pytest tests?

easy

A. To capture and check printed output sent to stdout and stderr

B. To speed up test execution by skipping print statements

C. To automatically fix syntax errors in print statements

D. To generate random test data for print output

Solution

1. Step 1: Understand what capsys does

   capsys is a pytest feature that captures output sent to the console, including both stdout and stderr.

2. Step 2: Identify its main use in tests

   It allows tests to check what was printed, so you can verify messages or errors your code produces.

3. Final Answer:

   To capture and check printed output sent to stdout and stderr -> Option A

4. Quick Check:

   capsys captures output = C [OK]

Hint: capsys catches print output for verification [OK]

Common Mistakes:

• Thinking capsys speeds up tests
• Believing capsys fixes code errors
• Confusing capsys with test data generation

2. Which of the following is the correct way to use capsys to capture printed output in a pytest test function?

easy

A. def test_func(capsys): captured = capsys.capture(); print('Hello')

B. def test_func(): capsys = print('Hello'); captured = capsys.readouterr()

C. def test_func(capsys): print('Hello'); captured = capsys.readouterr()

D. def test_func(): captured = capsys.readouterr(); print('Hello')

Solution

1. Step 1: Check how capsys is passed and used

   In pytest, capsys is passed as a function argument and used after printing to capture output.

2. Step 2: Verify correct method to read output

   The method to get output is capsys.readouterr(), which returns captured stdout and stderr.

3. Final Answer:

   def test_func(capsys): print('Hello'); captured = capsys.readouterr() -> Option C

4. Quick Check:

   capsys as argument + readouterr() = A [OK]

Hint: Pass capsys as argument and call readouterr() after print [OK]

Common Mistakes:

• Not passing capsys as a test function argument
• Calling non-existent capture() method
• Calling readouterr() before printing

3. What will be the output of the following pytest test code snippet?

def test_output(capsys):
    print('Test message')
    captured = capsys.readouterr()
    assert captured.out == 'Test message\n'

medium

A. Test passes because printed output matches assertion

B. Test fails because captured.out does not include newline

C. Test fails due to syntax error in print statement

D. Test fails because capsys.readouterr() is called before print

Solution

1. Step 1: Understand output capture timing

   The print statement runs first, sending 'Test message' plus a newline to stdout.

2. Step 2: Check captured output and assertion

   capsys.readouterr() captures the output including the newline. The assertion checks for 'Test message\n', which matches exactly.

3. Final Answer:

   Test passes because printed output matches assertion -> Option A

4. Quick Check:

   Printed output includes newline = B [OK]

Hint: Remember print adds newline; assert with '\n' [OK]

Common Mistakes:

• Forgetting newline in assertion string
• Calling readouterr() before print
• Assuming captured.out excludes newlines

4. Identify the error in this pytest test using capsys:

def test_error(capsys):
    captured = capsys.readouterr()
    print('Hello')
    assert captured.out == 'Hello\n'

medium

A. Missing parentheses in print statement

B. Calling capsys.readouterr() before printing captures nothing

C. Assertion string missing newline character

D. capsys must be imported before use

Solution

1. Step 1: Check order of output capture and print

   capsys.readouterr() is called before the print statement, so it captures output up to that point, which is empty.

2. Step 2: Understand assertion failure reason

   The assertion expects 'Hello\n' but captured.out is empty, causing test failure.

3. Final Answer:

   Calling capsys.readouterr() before printing captures nothing -> Option B

4. Quick Check:

   Read output after print, not before = A [OK]

Hint: Always call readouterr() after print to capture output [OK]

Common Mistakes:

• Calling readouterr() before print
• Forgetting newline in assertion
• Assuming capsys needs import

5. You want to test a function that prints error messages to stderr using pytest and capsys. Which code snippet correctly captures and asserts the stderr output?

hard

A. def test_error_message(capsys):
print('Error occurred', file=sys.stdout)
captured = capsys.readouterr()
assert captured.err == 'Error occurred\n'

B. def test_error_message(capsys):
print('Error occurred')
captured = capsys.readouterr()
assert captured.err == 'Error occurred\n'

C. def test_error_message():
print('Error occurred', file=sys.stderr)
captured = capsys.readouterr()
assert captured.err == 'Error occurred\n'

D. def test_error_message(capsys):
print('Error occurred', file=sys.stderr)
captured = capsys.readouterr()
assert captured.err == 'Error occurred\n'

Solution

1. Step 1: Identify how to print to stderr

   Use print(..., file=sys.stderr) to send output to stderr stream.

2. Step 2: Check correct use of capsys and assertion

   capsys.readouterr() captures both stdout and stderr. To check stderr, assert captured.err matches expected string including newline.

3. Final Answer:

   def test_error_message(capsys): print('Error occurred', file=sys.stderr) captured = capsys.readouterr() assert captured.err == 'Error occurred\n' -> Option D

4. Quick Check:

   Print to stderr + assert captured.err = D [OK]

Hint: Print with file=sys.stderr and assert captured.err [OK]

Common Mistakes:

• Not passing capsys as argument
• Printing to stdout instead of stderr
• Asserting captured.out instead of captured.err