[AIRFLOW-6490] Improve time delta comparison in local task job tests

apache · Jan 9, 2020 · 7e277c7 · 7e277c7
1 parent 6cbedf8
commit 7e277c7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/jobs/test_local_task_job.py b/tests/jobs/test_local_task_job.py
@@ -159,7 +159,10 @@ def test_heartbeat_failed_fast(self, mock_getpid):
             for i in range(1, len(heartbeat_records)):
                 time1 = heartbeat_records[i - 1]
                 time2 = heartbeat_records[i]
-                self.assertGreaterEqual((time2 - time1).total_seconds(), job.heartrate)
+                # Assert that difference small enough to avoid:
+                # AssertionError: 1.996401 not greater than or equal to 2
+                delta = (time2 - time1).total_seconds()
+                self.assertAlmostEqual(delta, job.heartrate, delta=0.006)
 
     @unittest.skipIf('mysql' in conf.get('core', 'sql_alchemy_conn'),
                      "flaky when run on mysql")