← Back
Editing: test_odeint_jac.cpython-311.pyc
� d�c � �h � d dl Zd dlmZmZ d dlmZ d dlmc m Z d� Z d� Z d� Z dZdZd � Zd � Zd � ZdS ) � N)� assert_equal�assert_allclose)�odeintc �Z � t j | � � }t j || |� � |S �N)�np� zeros_like� banded5x5)�y�t�dydts �G/usr/lib/python3/dist-packages/scipy/integrate/tests/test_odeint_jac.py�rhsr s+ � � �=�� � �D� ���1�d�#�#�#� �K� c � � t | � � }t j ||fd�� � }t j || dd|� � |S )N�F��order� )�lenr �zerosr � banded5x5_jac)r r �n�jacs r r r sC � � �A���A� �(�A�q�6�� %� %� %�C� ��A�q�!�Q��,�,�,� �Jr c � � t | � � }t j d|fd�� � }t j || dd|� � |S )N� r r r )r r r r �banded5x5_bjac)r r r �bjacs r r r sC � � �A���A� �8�Q��F�#� &� &� &�D� � �Q��1�a��.�.�.� �Kr r r c �F � | t k r d }d }t }n)| t k r d}d}t }nt d| ��� � �t j dd� � }d}d}d}d }|t j |dz � � z } t t || |||||d � � � \ } } | d } | d d } | d d }| d d }|� � � }t j |||| � � \ }}}t | |d�� � t | ||f|||f� � d S )N� r zinvalid jactype: g �?g @g�dy���=g�vIh�%<=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> sM � ��,� � � �� ����� �N� "� "� �� ������j�'�'�;�<�<�<� ��3�� � �B� �D� �D� �B� �F� �R�Y�v�a�x� � � �A��s�B�� $��� �t��?� ?� ?�I�C�� ��W�F��e��R� �J��e��R� �J��e��R� �J� �����B� �-�b�&�"�g�F�F�M�C��c� �F�B�U�+�+�+�+��*�j�*�5��S�#��G�G�G�G�Gr c �. � t t � � d S r )r> r, � r r �test_odeint_full_jacrA F s � �� � � � � � r c �. � t t � � d S r )r> r- r@ r r �test_odeint_banded_jacrC J 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� 7� 7� 7� 7� 7� "� "� "� "� "� "� 7� 7� 7� 7� 7� 7� 7� 7� 7�� � � � � �� � � � ���#H� #H� #H�L � � �!� !� !� !� !r
Save File
Cancel