Fixtures maken met pytest

Wanneer je unittests bouwt, moet je soms eerst wat voorbereiden voordat je kunt testen. Als je deze setup binnen een unittest doet, worden tests moeilijker leesbaar en moet je het misschien meerdere keren herhalen. Gelukkig biedt pytest een manier om dit op te lossen: fixtures.

Voor deze oefening is pandas geïmporteerd als pd, en de functie extract() hieronder is beschikbaar!

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

Deze oefening maakt deel uit van de cursus

ETL en ELT in Python

Bekijk cursus

Oefeninstructies

Importeer de bibliotheek pytest.
Maak een pytest-fixture met de naam raw_tax_data.
Retourneer de DataFrame raw_data.

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan 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