code: def setup_tests(self):
      clear_db_runs()
      drs = _create_dagruns(
          self.dag1,
          [
              _DagRunInfo(d, (d, d + datetime.timedelta(days=1)))
              for d in self.execution_dates
          ],
          state=State.RUNNING,
          run_type=DagRunType.SCHEDULED,
      )
      for dr in drs:
          dr.dag = self.dag1
  
      drs = _create_dagruns(
          self.dag2,
          [
              _DagRunInfo(
                  self.dag2.start_date,
                  (
                      self.dag2.start_date,
                      self.dag2.start_date + datetime.timedelta(days=1),
                  ),
              ),
          ],
          state=State.RUNNING,
          run_type=DagRunType.SCHEDULED,
      )
  
      for dr in drs:
          dr.dag = self.dag2
  
      drs = _create_dagruns(
          self.dag3,
          [_DagRunInfo(d, (d, d)) for d in self.dag3_execution_dates],
          state=State.SUCCESS,
          run_type=DagRunType.MANUAL,
      )
      for dr in drs:
          dr.dag = self.dag3
  
      drs = _create_dagruns(
          self.dag4,
          [
              _DagRunInfo(
                  self.dag4.start_date,
                  (
                      self.dag4.start_date,
                      self.dag4.start_date + datetime.timedelta(days=1),
                  ),
              )
          ],
          state=State.SUCCESS,
          run_type=DagRunType.MANUAL,
      )
      for dr in drs:
          dr.dag = self.dag4
  
      yield
  
      clear_db_runs()
  
imports: from __future__ import annotations
  import datetime
  from typing import Callable
  import pytest
  from sqlalchemy.orm import eagerload
  from airflow import models
  from airflow.api.common.mark_tasks import _create_dagruns, _DagRunInfo, set_dag_run_state_to_failed, set_dag_run_state_to_queued, set_dag_run_state_to_running, set_dag_run_state_to_success, set_state
  from airflow.models import DagRun
  from airflow.utils import timezone
  from airflow.utils.session import create_session, provide_session
  from airflow.utils.state import State
  from airflow.utils.types import DagRunType
  from tests.test_utils.db import clear_db_runs
  from tests.test_utils.mapping import expand_mapped_task
label_desc: Use image node_id 2 for calling the TestMarkTasks obj's underlying member method code with example usage: obj.setup_tests() without return types