From 8c15784a3574d57d8efd2d36a6ca8d5bfff95984 Mon Sep 17 00:00:00 2001
From: Pratyush Sharma <56130065+pratyush618@users.noreply.github.com>
Date: Thu, 9 Apr 2026 18:51:30 +0530
Subject: [PATCH] fix: increased timeout for assertion test cases

---
 tests/python/execution/test_distributed_executor.py | 2 +-
 tests/python/execution/test_executor.py             | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/python/execution/test_distributed_executor.py b/tests/python/execution/test_distributed_executor.py
index 81b6d36..0f56441 100644
--- a/tests/python/execution/test_distributed_executor.py
+++ b/tests/python/execution/test_distributed_executor.py
@@ -221,7 +221,7 @@ def test_diamond_dag_parallel(self, diamond_dag):
         elapsed = time.monotonic() - start
         assert dist_result.execution_result.succeeded == 4
         # b and c run in parallel, so total should be ~0.2s, not ~0.3s
-        assert elapsed < 0.5
+        assert elapsed < 0.8
 
 
 class TestDistributedExecutionResult:
diff --git a/tests/python/execution/test_executor.py b/tests/python/execution/test_executor.py
index 11f90be..a286d36 100644
--- a/tests/python/execution/test_executor.py
+++ b/tests/python/execution/test_executor.py
@@ -121,7 +121,7 @@ def test_parallel_execution(self, diamond_dag):
         elapsed = time.monotonic() - start
         assert result.succeeded == 4
         # Sequential would be ~0.3s, parallel b+c should be ~0.2s
-        assert elapsed < 0.5
+        assert elapsed < 0.8
 
     def test_with_costs(self, diamond_dag):
         tasks = {