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)

Deze oefening maakt deel uit van de cursus

ETL and ELT in Python

Cursus bekijken

Oefeninstructies

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

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# Import pytest
____

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

Code bewerken en uitvoeren