mock_object()
        with dag:
            check_task()
        dag.test()
        mock_object.assert_called_once()
    def test_dag_test_with_dependencies(self):
        dag = DAG(dag_id="test_local_testing_conn_file", start_date=DEFAULT_DATE)
        mock_object = mock.MagicMock()
        @task_decorator
        def check_task():
            return "output of first task"
        @task_decorator
        def check_task_2(my_input):
            
            mock_object(my_input)
        with dag:
            check_task_2(check_task())
        dag.test()
        mock_object.assert_called_with("output of first task")
    def test_dag_test_with_task_mapping(self):
        dag = DAG(dag_id="test_local_testing_conn_file", start_date=DEFAULT_DATE)
        mock_object = mock.MagicMock()
        @task_decorator()
        def get_index(current_val, ti=None):
            return ti.map_index
        @task_decorator
        def check_task(my_input):