This is an automated email from the ASF dual-hosted git repository.

ash pushed a commit to branch task-sdk-first-code
in repository https://gitbox.apache.org/repos/asf/airflow.git

commit a8457abed40b64e7b461ab172fcb491944290aa5
Author: Ash Berlin-Taylor <[email protected]>
AuthorDate: Wed Oct 30 09:50:25 2024 +0000

    [skip ci]
---
 providers/tests/standard/operators/test_weekday.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/providers/tests/standard/operators/test_weekday.py 
b/providers/tests/standard/operators/test_weekday.py
index c9df8b0124f..127d735c3da 100644
--- a/providers/tests/standard/operators/test_weekday.py
+++ b/providers/tests/standard/operators/test_weekday.py
@@ -22,6 +22,7 @@ import datetime
 import pytest
 import time_machine
 
+from airflow.exceptions import AirflowException
 from airflow.models.dagrun import DagRun
 from airflow.models.taskinstance import TaskInstance as TI
 from airflow.models.xcom import XCom
@@ -207,7 +208,7 @@ class TestBranchDayOfWeekOperator:
 
     def test_branch_with_no_weekday(self, dag_maker):
         """Check if BranchDayOfWeekOperator raises exception on missing 
weekday"""
-        with pytest.raises(TypeError, match="missing keyword argument 
'week_day'"):
+        with pytest.raises((TypeError, AirflowException), match="missing 
keyword argument 'week_day'"):
             with dag_maker(
                 "branch_day_of_week_operator_test",
                 start_date=DEFAULT_DATE,

Reply via email to