a ztest_levin_0..r levin_variant)rdpsmpfeps extraprecprecroneappend update_psum RuntimeErrorexplogabspinsuminf) rLSsrveerrwrrr test_levin_0 s$  0  r/c Csdt_ttj}tdtjrtjddd}gd}}t|d}|d7}||||\}}||krtq|dkr.r)rrrrrrrrupdater!r"r#r$zetar&r') rr(Arr*r+r,r-r.rrr test_levin_1"s$   0  r4c s$dt_tdttj}tdtjptjddd}d}d|t|| }||\}}|d 7}||kr|q|d kr@t d q@Wdn1s0Yt d t |}t fd ddtj g}t||}||ksJtjfdddtj gddd}||ks JdS)Nr rsiditrrr r r rcst| d|Srrr"xzrrrGrztest_levin_2..csd|t|| S)Nr8rfacrr<rrrLrr)rrrrrrrr?stepr!r"r#quadr'r$r& rr(rr*r+r,exactr-r.rr<r test_levin_27s$  0   rDc sdt_tdttj}tdtjtjddd}d\}}| |td|t|td|d|7}|d 7}||\}}||krq|d krDt d qDWdn1s0Yt d t |}t fd ddtj gdtdtj}t||}||ks,Jtjfdddtj gdddtjdgddtd Dd}t||}||ksJdS)Nrrrr7r)rrr r r 皙?cs t| |d|dS)NrrFr9r:r<rrr^rztest_levin_3..rcs8 |td|t|td|d|S)NrFrr>rr<rrrcrcSsg|]}dqS)r r.0r;rrr crz test_levin_3..)r rZworkprecsteps)rrrrrrrr?Z step_psumr!r"r#rAr'sqrtr%r$r&xrangerBrr<r test_levin_3Os&  <0. > rOcsDdt_ttjZtddtjfdddtjgddd}t|tjd ks^JWdn1sr0Yt d t tj }tjd ddtjgd d}t|t d |ksJdfdd}tj|dtjgdddt dDd}t d tddtddtd}t|||ks@JdS)Nrr5ics|d Srrrr<rrrlrz!test_levin_nsum..r l)r g|=rGcSsd|d|S)Nr8r rrrrrrqrr6ry@?csTtdtd|tdtd||tdtd|t|S)NrrFr )rrfrr?rr<rrrurrrcSsg|]}dqS)r5rrIrrrrKvrz#test_levin_nsum..r )r rLrQrF)rrrrrr&r'r$Zeulerr"r#rrNZhyp2f1)arfr+rCrr<rtest_levin_nsumgs&4 &.rUcCsdt_t}gdd}}}|d| tj||7}|d7}||||\}}|tjkrfqx|dkrtdqtdt tj}t |tj dd }||ksJdS) Nrrr r8r r rr ) rr cohen_altrrr rr!r"r#r$r%)ACr)r*rr+r,rr-rrrtest_cohen_alt_0zs  rYcCsdt_g}t}d}|tdtjd|d|tdtjd| |d7}||\}}|tjkrzq|dkrtdqt |}t |d}|dksJdS)Nrr rr r g?g-q=) rrrWrZloggammarr1rr!r"r$)r3rXrr+r,r-rrrtest_cohen_alt_1s"    rZN) ZmpmathrrbackendrNr/r4rDrOrUrYrZrrrrs