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

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