Skapa fixtures med pytest
När du bygger enhetstester behöver du ibland göra viss förberedelse innan testningen kan börja. Att göra denna förberedelse inuti ett enhetstest kan göra testerna svårare att läsa, och den kan behöva upprepas flera gånger. Lyckligtvis erbjuder pytest ett sätt att lösa dessa problem, med fixtures.
I den här övningen har pandas importerats som pd, och funktionen extract() som visas nedan är tillgänglig att använda!
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 som heterraw_tax_data. - Returnera DataFrame-objektet
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 ____