Створення фікстур у pytest
Під час написання модульних тестів інколи потрібно виконати певну підготовку перед початком тестування. Якщо робити цю підготовку всередині модульного тесту, тести стають важчими для читання, і підготовчі кроки можуть повторюватися багато разів. На щастя, pytest пропонує розвʼязання цих проблем — фікстури.
У цій вправі pandas уже імпортовано як pd, і доступна функція extract() нижче!
def extract(file_path):
return pd.read_csv(file_path)
Ця вправа є частиною курсу
ETL та ELT у Python
Інструкції до вправи
- Імпортуйте бібліотеку
pytest. - Створіть фікстуру
pytestпід назвоюraw_tax_data. - Поверніть датафрейм
raw_data.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____