Fixtures maken met pytest

Bij het bouwen van unit tests moet je soms eerst wat voorbereiden voordat je kunt testen. Dit binnen een unit test doen kan de tests lastiger leesbaar maken en moet mogelijk meerdere keren herhaald worden. Gelukkig biedt pytest hiervoor een oplossing met fixtures.

Voor deze oefening is pandas al geïmporteerd als pd, en de hieronder getoonde functie extract() 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

Cursus bekijken

Oefeninstructies

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

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