Python Unit Test Automation

Listing 5-6.  test_module05.py
import pytest
@pytest.fixture()
def fixture01():
    print("\nIn fixture01()...")
@pytest.mark.usefixtures('fixture01')
def test_case01(fixture01):
    print("\nIn test_case01()...")
The output of Listing 
5-6
 is exactly the same as the code in Listing 
5-5
.
You can use the @pytest.mark.usefixtures() decorator for a class, as shown in 
Listing 
5-7
.
Listing 5-7.  test_module06.py
import pytest
@pytest.fixture()
def fixture01():
    print("\nIn fixture01()...")
@pytest.mark.usefixtures('fixture01')
class TestClass03:
    def test_case01(self):
        print("I'm the test_case01")
    def test_case02(self):
        print("I'm the test_case02")
Here is the output:
============================ test session starts =======================
platform linux -- Python 3.4.2, pytest-3.0.4, py-1.4.31, pluggy-0.4.0 --  
/usr/bin/python3
cachedir: .cache
rootdir: /home/pi/book/code/chapter05/test, inifile:
collected 2 items
test_module06.py::TestClass03::test_case01
In fixture01()...

---

Chapter 5 
■
 pytest
95
I'm the test_case01
PASSED
test_module06.py::TestClass03::test_case02
In fixture01()...
I'm the test_case02
PASSED
========================== 2 passed in 0.08 seconds ====================
If you want to run a block of code after the test with a fixture has run, you have to add 
a finalizer function to the fixture. Listing 
5-8
 demonstrates this idea.