Создание фикстур с помощью pytest
При создании юнит-тестов иногда нужно выполнить небольшую подготовку перед началом тестирования. Если делать это прямо внутри теста, читать его становится сложнее, а саму подготовку порой приходится повторять несколько раз. К счастью, pytest предлагает решение этой проблемы — фикстуры.
Для этого упражнения уже импортирован pandas как pd, а также доступна функция extract(), показанная ниже!
def extract(file_path):
return pd.read_csv(file_path)
Это упражнение является частью курса
ETL и ELT в Python
Инструкции к упражнению
- Импортируйте библиотеку
pytest. - Создайте фикстуру
pytestс именемraw_tax_data. - Верните DataFrame
raw_data.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____