1. Learn
    2. /
  2. Курси
    3. /
  3. ETL та ELT у Python
Connected

вправа

Юніт-тестування конвеєра даних із фікстурами

У попередньому відео ви дізналися, що юніт-тестування допомагає підвищити довіру до вашого конвеєра даних і навіть виявляти помилки під час розробки. У цій вправі ви попрактикуєтеся писати і фікстури, і юніт-тести, використовуючи бібліотеку pytest та оператор assert.

Нижче наведено функцію transform, навколо якої ви створюватимете юніт-тести. pandas уже імпортовано як pd, а бібліотеку pytest() завантажено й підготовлено до використання.

def transform(raw_data):
    raw_data["tax_rate"] = raw_data["total_taxes_paid"] / raw_data["total_taxable_income"]
    raw_data.set_index("industry_name", inplace=True)
    return raw_data

Інструкції 1/2

undefined XP
    1
    2
  • Створіть фікстуру pytest з назвою clean_tax_data.
  • Застосуйте функцію transform() до набору даних raw_data, збережіть результат у змінній clean_data і поверніть її.