def dag(self):
"""The DAG deserialized from the ``data`` column"""
from airflow.serialization import SerializedDAG # noqa # pylint: disable=redefined-outer-name
if isinstance(self.data, dict):
dag = SerializedDAG.from_dict(self.data) # type: Any
else:
# noinspection PyTypeChecker
dag = SerializedDAG.from_json(self.data)
return dag
def test_deserialization_schedule_interval(self,
serialized_schedule_interval,
expected):
serialized = {
"__version": 1,
"dag": {
"default_args": {
"__type": "dict",
"__var": {}
},
"params": {},
"_dag_id": "simple_dag",
"fileloc": __file__,
"tasks": [],
"timezone": "UTC",
"schedule_interval": serialized_schedule_interval,
},
}
SerializedDAG.validate_schema(serialized)
dag = SerializedDAG.from_dict(serialized)
self.assertEqual(dag.schedule_interval, expected)