Skapa fixtures med pytest
När du skriver enhetstester behöver du ibland göra vissa förberedelser innan testningen kan börja. Att lägga in dessa förberedelser direkt i testerna gör dem svårare att läsa, och samma kod kan behöva upprepas flera gånger. Lyckligtvis erbjuder pytest en lösning på detta problem – fixtures.
I den här övningen har pandas importerats som pd, och funktionen extract() nedan är tillgänglig för användning!
def extract(file_path):
return pd.read_csv(file_path)
Den här övningen är en del av kursen
ETL och ELT i Python
Övningsinstruktioner
- Importera biblioteket
pytest. - Skapa en
pytest-fixture med namnetraw_tax_data. - Returnera DataFrame:n
raw_data.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____