!z)assert_task_scheduled..zsW  Zqz"YY!&YYQXYYquYYYr z not found with z Scheduled tasks: ) snapshotlistfuturerrrrrAssertionErrorr) rrr r r r, all_tasksrcriteriar found_tasks found_strs rassert_task_scheduledr4Ns22__&&&&&&&&HX_%%I Y$ D D  FF  $<t |t r|n|jtfd|D}d}||krtd|d|d|dS)aAssert the number of scheduled tasks matches the expected count. Args: docket: The Docket instance to check function: Optional task function or name to count (if None, counts all tasks) count: Expected number of tasks Raises: AssertionError: If the count doesn't match Example: await assert_task_count(docket, count=5) # All tasks await assert_task_count(docket, my_task, count=2) # Specific function Nz all tasksc3<K|]}|jjkdVdS)N)rr)r(r)rs r z$assert_task_count..s2XX1:3F-3W3W13W3W3W3WXXrz tasks for z Expected  z, found ) r,r-r.lenrrrsumr/)rrr7r,r0 actual_count function_descrs @rassert_task_countrAs(__&&&&&&&&HX_%%I9~~ # $.x$=$=T8CT XXXXiXXXXX 4]44 uVVVVV VVWWWrcK|d{V}t|j}|rCd|D}d|}t dt |d|dS)zAssert that no tasks are scheduled on the docket. Args: docket: The Docket instance to check Raises: AssertionError: If any tasks are scheduled Example: await assert_no_tasks(docket) Nc ^g|]*}d|jjd|jd|jd|jd +Sr!r&r's rr*z#assert_no_tasks..sY    ^1:& ] ]af ] ] ] ]STSX ] ] ]   rr+zExpected no tasks, found z task(s) scheduled: )r,r-r.rr/r=)rr,r0r2r3s rassert_no_tasksrDs__&&&&&&&&HX_%%I      IIk**  XI X XY X X     r)N)__doc__collections.abcrtypingr docket.docketrdocket.executionrrtupledictboolrrr4r6intrArDrrrOs$%$$$$$ &&&&&&HS#X&& S/D  cNT !  t   <HS#X&& S/D  cNT ! t   4$($( 555 5HS#X&&5 S/D 5 cNT ! 5 t 5 5555x$($( $$$ $HS#X&&$ S/D $ cNT ! $ t $ $$$$R15"X"X "XHS#X&&-"X  "X  "X"X"X"XJ & T      r