Get startedGet started for free

Unit testing a task function

The key to unit testing Airflow code is extracting the business logic from @task functions into standalone Python functions that can be tested without running Airflow. Your team has done this with the clean_record function in dags/data_cleaning.py, which strips whitespace from names and lowercases emails.

You'll complete the unit tests for this function.

This exercise is part of the course

Building Data Pipelines with Airflow

View Course

Hands-on interactive exercise

Turn theory into action with one of our interactive exercises

Start Exercise