Создание фикстур с помощью 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 ____