Prueba de la unidad de flujo de air Python?

Me gustaría agregar algunas pruebas unitarias para nuestros DAG, pero no pude encontrar ninguna. ¿Hay un marco para la prueba de unidad para DAGs? Existe un marco de pruebas de extremo a extremo, pero supongo que está muerto: https://issues.apache.org/jira/browse/AIRFLOW-79 . Por favor sugerir, gracias!

Prueba tus operadores de esta manera:

 class TestMyOperator(TestCase): def test_execute(self): dag = DAG(dag_id='foo', start_date=datetime.now()) task = MyOperator(dag=dag, task_id='foo') ti = TaskInstance(task=task, execution_date=datetime.now()) result = task.execute(ti.get_template_context()) self.assertEqual(result, 'foo') 

Fuente

Actualmente no pude encontrar nada mejor que simplemente usar BashOperator :

 with DAG('platform-test', start_date=datetime(2017, 8, 29)) as dag: test_command = "python3 -m unittest --verbose {}".format(platform_test_fname) op = BashOperator( task_id="platform-test", bash_command=test_command, )