Fix flaky templatized call (apache#20843)

TiDeane · Jan 13, 2022 · ca030b9 · ca030b9
1 parent 2fdc233
commit ca030b9
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/tests/sensors/test_python.py b/tests/sensors/test_python.py
@@ -84,8 +84,6 @@ def test_python_callable_arguments_are_templatized(self):
             task.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE)
 
         ds_templated = DEFAULT_DATE.date().isoformat()
-        # 2 calls: first: at start, second: before timeout
-        assert 2 == len(recorded_calls)
         self._assert_calls_equal(
             recorded_calls[0],
             Call(
@@ -125,10 +123,6 @@ def test_python_callable_keyword_arguments_are_templatized(self):
         with pytest.raises(AirflowSensorTimeout):
             task.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE)
 
-        # 2 calls: first: at start, second: before timeout
-        # But sometimes 1 is also OK if the system is "busy" - we anyhow only check the first one which
-        # will always be there
-        assert len(recorded_calls) >= 1
         self._assert_calls_equal(
             recorded_calls[0],
             Call(