Créer des fixtures avec pytest
Lorsque vous construisez des tests unitaires, vous devez parfois effectuer une petite préparation avant de pouvoir commencer à tester. Faire cette préparation à l'intérieur d'un test unitaire peut rendre les tests plus difficiles à lire et peut devoir être répétée plusieurs fois. Heureusement, pytest propose une façon de résoudre 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 ____