
    (ph                     f    S SK rS SKJrJr  S SKJr  S SKJs  J	r
  S rS rS rSrSrS	 rS
 rS rg)    N)assert_equalassert_allclose)odeintc                 `    [         R                  " U 5      n[        R                  " XU5        U$ N)np
zeros_like	banded5x5)ytdydts      X/var/www/html/venv/lib/python3.13/site-packages/scipy/integrate/tests/test_odeint_jac.pyrhsr      s%    ==Dd#K    c                 z    [        U 5      n[        R                  " X"4SS9n[        R                  " XSSU5        U$ )NForder   )lenr   zerosr
   banded5x5_jac)r   r   njacs       r   r   r      s6    AA
((A6
%CA!Q,Jr   c                 |    [        U 5      n[        R                  " SU4SS9n[        R                  " XSSU5        U$ )N   r   r   r   )r   r   r   r
   banded5x5_bjac)r   r   r   bjacs       r   r   r      s8    AA88QF#&DQ1a.Kr   r   r   c                    U [         :X  a  S nS n[        nO$U [        :X  a  SnSn[        nO[	        SU < 35      e[
        R                  " SS5      nSnSnSnS	nU[
        R                  " US-   5      -  n	[        [        XIX1UXeS
S9	u  pU
S   nUS   S   nUS   S   nUS   S   nUR                  5       n[        R                  " UXU 5      u  nnn[        UUSS9  [        XU4UUU45        g )N   r   zinvalid jactype: g      ?g      @gdy=gvIh%<=g      ?@   T)Dfunmlmuatolrtolfull_outputnstnfenjeg-q=)r&   )JACTYPE_FULLr   JACTYPE_BANDEDr   
ValueErrorr   aranger   r   copyr
   banded5x5_solver   r   )jactyper#   r$   jacobiany0r&   r%   dtnstepsr   solinfoyfinal
odeint_nst
odeint_nfe
odeint_njey1r)   r*   r+   s                       r   check_odeintr>      s   ,	N	",WK899	3	BDD	BF
RYYvax  AsB$ ?IC WFeRJeRJeRJ	B--b&gFMCc FBU+**5S#Gr   c                  "    [        [        5        g r   )r>   r,    r   r   test_odeint_full_jacrA   E   s
    r   c                  "    [        [        5        g r   )r>   r-   r@   r   r   test_odeint_banded_jacrC   I   s
     r   )numpyr   numpy.testingr   r   scipy.integrater   #scipy.integrate._test_odeint_banded	integrate_test_odeint_bandedr
   r   r   r   r,   r-   r>   rA   rC   r@   r   r   <module>rJ      s@     7 " 7 7 #HL!r   