@@ -96,7 +96,7 @@ async def test_snapshot_with_mixed_tasks(docket: Docket):
9696 future = datetime .now (timezone .utc ) + timedelta (seconds = 5 )
9797 await docket .add (tasks .trace , when = future )("hi!" )
9898 for _ in range (5 ): # more than the concurrency allows
99- await docket .add (tasks .sleep )(4 )
99+ await docket .add (tasks .sleep )(2 )
100100
101101 async with Worker (docket , name = "test-worker" , concurrency = 2 ) as worker :
102102 worker_running = asyncio .create_task (worker .run_until_finished ())
@@ -187,10 +187,10 @@ async def test_snapshot_with_stats_flag_mixed_tasks(docket: Docket):
187187
188188 # Add multiple tasks of different types
189189 future = datetime .now (timezone .utc ) + timedelta (seconds = 5 )
190- await docket .add (tasks .trace , when = future , key = "trace-1" )("hi!" )
191- await docket .add (tasks .trace , when = future , key = "trace-2" )("hello!" )
190+ await docket .add (tasks .trace , when = future )("hi!" )
191+ await docket .add (tasks .trace , when = future )("hello!" )
192192 for _ in range (3 ):
193- await docket .add (tasks .sleep , key = f"sleep- { _ } " )( 4 )
193+ await docket .add (tasks .sleep )( 2 )
194194
195195 async with Worker (docket , name = "test-worker" , concurrency = 2 ) as worker :
196196 worker_running = asyncio .create_task (worker .run_until_finished ())
@@ -206,7 +206,7 @@ async def test_snapshot_with_stats_flag_mixed_tasks(docket: Docket):
206206 assert result .exit_code == 0 , result .output
207207
208208 # Should show the normal summary
209- assert "1 workers, 2/5 running" in result .output , result . output
209+ assert "1 workers, 2/5 running" in result .output
210210
211211 # Should show task statistics table with enhanced columns
212212 assert "Task Count Statistics by Function" in result .output
0 commit comments