Creating fixtures with pytest

When building unit tests, you'll sometimes have to do a bit of setup before testing can begin. Doing this setup within a unit test can make the tests more difficult to read, and may have to be repeated several times. Luckily, pytest offers a way to solve these problems, with fixtures.

For this exercise, pandas has been imported as pd, and the extract() function shown below is available for use!

def extract(file_path):
    return pd.read_csv(file_path)

This exercise is part of the course

ETL and ELT in Python

View Course

Exercise instructions

Import the pytest library.
Create a pytest fixture called raw_tax_data.
Return the raw_data DataFrame.

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

# Import pytest
____

# Create a pytest fixture
@pytest.fixture()
def ____():
	raw_data = extract("raw_tax_data.csv")
    
    # Return the raw DataFrame
	return ____

Edit and Run Code