Creare fixture con pytest
Quando crei test unitari, a volte devi fare un po' di preparazione prima di iniziare i test. Eseguire questo setup all'interno di un test unitario può rendere i test più difficili da leggere e potrebbe dover essere ripetuto più volte. Per fortuna, pytest offre un modo per risolvere questi problemi: le fixture.
Per questo esercizio, pandas è stato importato come pd ed è disponibile la funzione extract() mostrata qui sotto!
def extract(file_path):
return pd.read_csv(file_path)
Questo esercizio fa parte del corso
ETL and ELT in Python
Istruzioni dell'esercizio
- Importa la libreria
pytest. - Crea una fixture di
pytestchiamataraw_tax_data. - Restituisci il DataFrame
raw_data.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____