Vytváření fixtures s pytest
Při vytváření unit testů je někdy potřeba provést určitou přípravu ještě před samotným testováním. Psát tuto přípravu přímo do unit testů jejich čitelnost zhoršuje a navíc ji může být nutné několikrát opakovat. Naštěstí pytest nabízí elegantní řešení — fixtures.
Pro toto cvičení je pandas importován jako pd a níže uvedená funkce extract() je připravena k použití!
def extract(file_path):
return pd.read_csv(file_path)
Toto cvičení je součástí kurzu
ETL a ELT v Pythonu
Pokyny k cvičení
- Importuj knihovnu
pytest. - Vytvoř
pytestfixture s názvemraw_tax_data. - Vrať DataFrame
raw_data.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____