Créer des fixtures avec pytest
Lorsque vous créez des tests unitaires, vous devrez parfois faire un peu de préparation avant de commencer les tests. Faire cette préparation directement dans un test unitaire peut rendre les tests plus difficiles à lire et vous obliger à la répéter plusieurs fois. Heureusement, pytest offre une façon de régler ces problèmes grâce aux fixtures.
Pour cet exercice, pandas a été importé sous le nom pd, et la fonction extract() ci-dessous est disponible !
def extract(file_path):
return pd.read_csv(file_path)
Cette activité fait partie du cours
ETL et ELT en Python
Instructions de l’exercice
- Importez la bibliothèque
pytest. - Créez une fixture
pytestappeléeraw_tax_data. - Retournez le DataFrame
raw_data.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____