def METHOD_NAME(
    mock_timeout_dag, task_id, expected_result
):
    task = mock_timeout_dag.get_task(task_id)
    actual_result = PostgresHook.get_execution_timeout(task)
    assert actual_result == expected_result