Créer des fixtures avec pytest
Lorsque vous élaborez des tests unitaires, vous devez parfois procéder à une certaine configuration avant de pouvoir commencer les tests. La mise en place de cette configuration dans un test unitaire peut rendre les tests plus difficiles à lire et peut devoir être répétée plusieurs fois. Heureusement, pytest offre un moyen de résoudre ces problèmes, avec des luminaires.
Pour cet exercice, pandas a été importé en tant que pd, et la fonction extract() présentée ci-dessous peut être utilisée !
def extract(file_path):
return pd.read_csv(file_path)
Cet exercice fait partie du cours
ETL et ELT en Python
Instructions
- Importez la bibliothèque
pytest. - Créez une fixation
pytestappeléeraw_tax_data. - Renvoyez le DataFrame
raw_data.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____