Fixture teardown (yield) in PyTest - Build an Automation Script

Test resource setup and teardown using pytest fixture with yield

Preconditions (2)

Step 1: Create a fixture named 'resource' that prints 'Setup resource' before yield and 'Teardown resource' after yield

Step 2: Write a test function 'test_example' that uses the 'resource' fixture and asserts True is True

Step 3: Run the test using pytest

✅ Expected Result: The test passes, and the console output shows 'Setup resource' before the test and 'Teardown resource' after the test

Automation Requirements - pytest

Assertions Needed:

Assert True is True in the test function

Verify that the fixture setup and teardown messages are printed in correct order

Best Practices:

Use yield in fixture to separate setup and teardown

Keep fixture scope appropriate (function scope by default)

Use print statements or logging to confirm teardown execution

Automated Solution

PyTest

import pytest

@pytest.fixture
def resource():
    print("Setup resource")
    yield
    print("Teardown resource")

def test_example(resource):
    assert True is True

The fixture resource uses yield to separate setup and teardown steps. The code before yield runs before the test, and the code after yield runs after the test finishes.

The test function test_example uses the fixture by naming it as a parameter. It asserts a simple condition True is True to pass.

When running pytest, you will see the print statements in the console showing the setup message before the test and the teardown message after the test, confirming the fixture lifecycle.

Common Mistakes - 3 Pitfalls

Not using yield and putting all code before yield

Using print statements without running pytest in verbose mode

{'mistake': 'Not including the fixture as a parameter in the test function', 'why_bad': "The fixture will not be invoked, so setup and teardown won't run", 'correct_approach': 'Add the fixture name as a parameter to the test function to use it'}

Bonus Challenge

Now add a fixture that yields a resource value (like a string) and verify the test uses that value

Show Hint

Practice

1. What is the main purpose of using yield in a pytest fixture?

easy

A. To separate setup and teardown code in the fixture

B. To skip the test execution

C. To run the test multiple times

D. To mark the test as expected to fail

Solution

1. Step 1: Understand fixture setup and teardown

   In pytest, setup code prepares resources before a test, and teardown code cleans up after the test.

2. Step 2: Role of yield in fixture

   The yield statement splits the fixture into two parts: code before yield runs before the test (setup), and code after yield runs after the test (teardown).

3. Final Answer:

   To separate setup and teardown code in the fixture -> Option A

4. Quick Check:

   Yield separates setup and teardown [OK]

Hint: Remember: yield splits setup and cleanup in fixtures [OK]

Common Mistakes:

• Thinking yield skips tests
• Confusing yield with test repetition
• Using yield to mark test outcomes

2. Which of the following is the correct syntax for a pytest fixture using yield for teardown?

easy

A. def my_fixture():
setup()
yield
teardown()

B. def my_fixture():
yield
setup()
teardown()

C. def my_fixture():
setup()
teardown()
yield

D. def my_fixture():
teardown()
yield
setup()

Solution

1. Step 1: Identify setup and teardown order

   Setup code must run before yield, teardown code after yield.

2. Step 2: Check each option's order

   def my_fixture(): setup() yield teardown() correctly places setup before yield and teardown after. Others have incorrect order.

3. Final Answer:

   def my_fixture(): setup() yield teardown() -> Option A

4. Quick Check:

   Setup before yield, teardown after [OK]

Hint: Setup before yield, teardown after yield in fixture [OK]

Common Mistakes:

• Placing teardown before yield
• Calling setup after yield
• Yield placed at the end after teardown

3. What will be the output when running this pytest fixture and test?

import pytest

@pytest.fixture
def resource():
    print('Setup')
    yield
    print('Teardown')

def test_example(resource):
    print('Test running')

medium

A. Test running\nSetup\nTeardown

B. Setup\nTeardown\nTest running

C. Setup\nTest running\nTeardown

D. Teardown\nSetup\nTest running

Solution

1. Step 1: Trace fixture execution order

   Fixture prints 'Setup' before yield, then test runs printing 'Test running', then fixture prints 'Teardown' after yield.

2. Step 2: Combine output sequence

   Output order is 'Setup', 'Test running', then 'Teardown'.

3. Final Answer:

   Setup\nTest running\nTeardown -> Option C

4. Quick Check:

   Setup -> Test -> Teardown order [OK]

Hint: Yield splits setup (before) and teardown (after) print order [OK]

Common Mistakes:

• Assuming teardown runs before test
• Mixing print order
• Ignoring yield effect on execution flow

4. Identify the error in this pytest fixture using yield:

@pytest.fixture
def faulty_fixture():
    yield
    setup()
    teardown()

medium

A. Teardown code must be before yield

B. Yield cannot be used in fixtures

C. Fixture must return a value instead of yield

D. Setup code is after yield, so it runs after the test instead of before

Solution

1. Step 1: Analyze code order around yield

   Setup code should run before yield to prepare resources; here it is after yield, so it runs after the test.

2. Step 2: Understand consequences

   This means setup runs too late, breaking test assumptions. Teardown after setup is also misplaced.

3. Final Answer:

   Setup code is after yield, so it runs after the test instead of before -> Option D

4. Quick Check:

   Setup must be before yield [OK]

Hint: Setup must be before yield, teardown after [OK]

Common Mistakes:

• Placing setup after yield
• Thinking yield is not allowed
• Confusing teardown position

5. You want to create a pytest fixture that opens a file before a test and closes it after. Which fixture code correctly uses yield for this purpose?

hard

A. def file_fixture():
yield open('data.txt')
f.close()

B. def file_fixture():
f = open('data.txt')
yield f
f.close()

C. def file_fixture():
f = open('data.txt')
f.close()
yield f

D. def file_fixture():
f = open('data.txt')
yield
f.close()

Solution

1. Step 1: Open file before yield and yield file object

   def file_fixture(): f = open('data.txt') yield f f.close() opens the file, yields the file object to the test, allowing test to use it.

2. Step 2: Close file after yield

   After test finishes, code after yield closes the file, ensuring cleanup.

3. Final Answer:

   def file_fixture(): f = open('data.txt') yield f f.close() -> Option B

4. Quick Check:

   Open before yield, yield resource, close after yield [OK]

Hint: Yield resource to test, cleanup after yield [OK]

Common Mistakes:

• Closing file before yield
• Yielding without resource variable
• Not closing file after test