o
    g                     @   sr   d dl Zd dlmZmZ d dlmZ d dlm  m	Z
 dd Zdd Zdd	 Zd
ZdZdd Zdd Zdd ZdS )    N)assert_equalassert_allclose)odeintc                 C   s   t | }t|| | |S N)np
zeros_like	banded5x5)ytdydt r   c/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/scipy/integrate/tests/test_odeint_jac.pyrhs   s   
r   c                 C   s0   t | }tj||fdd}t|| dd| |S )NForder   )lenr   zerosr   banded5x5_jac)r	   r
   njacr   r   r   r         r   c                 C   s0   t | }tjd|fdd}t|| dd| |S )N   r   r   r   )r   r   r   r   banded5x5_bjac)r	   r
   r   bjacr   r   r   r      r   r   r   r   c                 C   s   | t krd }d }t}n| tkrd}d}t}ntd| f tdd}d}d}d}d	}|t|d  }	tt||	|||||d
d	\}
}|
d }|d d }|d d }|d d }|	 }t
|||| \}}}t||dd t|||f|||f d S )N   r   zinvalid jactype: %rg      ?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'   r   r   r   check_odeint    s6   
r:   c                   C      t t d S r   )r:   r(   r   r   r   r   test_odeint_full_jacF      r<   c                   C   r;   r   )r:   r)   r   r   r   r   test_odeint_banded_jacJ   r=   r>   )numpyr   numpy.testingr   r   scipy.integrater   #scipy.integrate._test_odeint_banded	integrate_test_odeint_bandedr   r   r   r   r(   r)   r:   r<   r>   r   r   r   r   <module>   s   &