Створення фікстур у 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 ____

Редагувати та запускати код