[파이썬] pytest 픽스처 스코프(scope) 설정

pytest, a popular testing framework in Python, provides a powerful and flexible mechanism called fixtures for setting up and managing resources needed for tests. Fixtures allow you to define and configure objects, data, or setups that your tests depend on.

One important aspect of fixtures in pytest is the concept of scope. The scope of a fixture determines how long it will be created and available during the test execution. pytest offers different scope options, which can be used to control when fixtures are set up and torn down.

Let’s dive into the different fixture scopes available in pytest:

Function Scope

The default scope for fixtures is function scope, which means that the fixture is created and destroyed for each test function that requires it. This ensures that each test starts with a clean and isolated setup.

import pytest

@pytest.fixture
def my_fixture():
    # set up code
    yield
    # tear down code

def test_my_test1(my_fixture):
    # test code

def test_my_test2(my_fixture):
    # test code

In the above example, the my_fixture fixture is function-scoped. It will be set up and torn down separately for each test function test_my_test1 and test_my_test2.

Class Scope

If you want to share the same fixture instance across all test methods within a test class, you can use the class scope. The fixture will be set up once and torn down after all test methods in the class have executed.

import pytest

@pytest.fixture(scope="class")
def my_fixture():
    # set up code
    yield
    # tear down code

class TestClass:
    def test_method1(self, my_fixture):
        # test code
    
    def test_method2(self, my_fixture):
        # test code

In the above code, the my_fixture fixture is now class-scoped. It will be set up once before any test methods in the class are executed and torn down after all methods have completed.

Module Scope

When you have multiple test modules and want to share fixtures across them, you can use the module scope. The fixture will be set up once at the beginning of the module and torn down at the end when all tests in that module have completed.

import pytest

@pytest.fixture(scope="module")
def my_fixture():
    # set up code
    yield
    # tear down code

def test_my_test1(my_fixture):
    # test code
    
def test_my_test2(my_fixture):
    # test code

In the above code, the my_fixture fixture is module-scoped. It will be set up once before any tests in the module are executed and torn down after all tests have completed.

Session Scope

The widest scope available in pytest is the session scope. A fixture with session scope is set up once before running any tests and torn down after all tests have finished.

import pytest

@pytest.fixture(scope="session")
def my_fixture():
    # set up code
    yield
    # tear down code

def test_my_test1(my_fixture):
    # test code
    
def test_my_test2(my_fixture):
    # test code

In the above code, the my_fixture fixture is session-scoped. It will be set up once before any tests start running and torn down after all tests have completed.

Conclusion

Understanding and leveraging the pytest fixture scope can greatly help in controlling the lifespan and availability of resources for your tests. By using the right scope, you can ensure proper setup and teardown for each test, avoid unnecessary duplication, and optimize test execution.

Make sure to experiment with different fixture scopes to find the appropriate balance between isolation and efficiency for your test suite.