Creare fixture con pytest
Quando scrivi test unitari, a volte dovrai fare un po' di setup prima di iniziare a testare. Eseguire questo setup dentro un test unitario può rendere i test più difficili da leggere e potrebbe dover essere ripetuto più volte. Per fortuna, pytest offre una soluzione a questi problemi con 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 interattivo pratico
Prova questo esercizio completando questo codice di esempio.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____