Créer des fixtures avec pytest

Lorsque vous écrivez des tests unitaires, il vous arrive parfois d'effectuer une petite configuration avant de pouvoir commencer les tests. Réaliser 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 propose une solution à 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)

Cet exercice fait partie du cours

<cours>ETL et ELT en Python</cours>

Voir le cours

Instructions de l’exercice

Importez la bibliothèque pytest.
Créez une fixture pytest appelée raw_tax_data.
Renvoyez 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 ____

Modifier et exécuter le code