chore: fix flaky unit tests: tests/unittests/flows/llm_flows/test_functions_simple.py

original tests assert too strict time boundary, now we only assert the parallel execution time should be less than sequential execution time

PiperOrigin-RevId: 800563929
This commit is contained in:
Xiang (Sean) Zhou
2025-08-28 12:17:03 -07:00
committed by Copybara-Service
parent 8d6f138fbe
commit b92b288c97
@@ -760,11 +760,13 @@ async def test_parallel_function_execution_timing():
start_time_diff < 0.01 start_time_diff < 0.01
), f'Functions started too far apart: {start_time_diff}s' ), f'Functions started too far apart: {start_time_diff}s'
# Total execution time should be closer to 0.1s (parallel) than 0.2s (sequential) # Total execution time should be less than the sum of all parallel function delays (0.2s)
# Allow some overhead for task creation and synchronization # This proves parallel execution rather than sequential execution
assert ( sequential_time = 0.2 # 0.1s + 0.1s if functions ran sequentially
total_time < 0.15 assert total_time < sequential_time, (
), f'Execution took too long: {total_time}s, expected < 0.15s' f'Execution took too long: {total_time}s, expected < {sequential_time}s'
' (sequential time)'
)
# Verify the results are correct # Verify the results are correct
assert testing_utils.simplify_events(events) == [ assert testing_utils.simplify_events(events) == [