← Back
Editing: test__root.cpython-311.pyc
� d�c5 � �P � d Z ddlmZ ddlmZ ddlZddlm Z G d� d� � Z dS )z5 Unit tests for optimization routines from _root.py. � )�assert_)�raisesN��rootc �& � e Zd Zd� Zd� Zd� Zd� ZdS )�TestRootc � � d� }d� }dD ]�}|dv r�|dv r|}nd }t |ddg|d|�� � }t |ddg|d |�� � }|�d ||j � � �d ||j � � ��}t |j |� � t |j |� � t t ||j � � � � � � � t ||j � � � � � � � k |� � ��d S ) Nc �P � | \ }}t j |dz dz |dz dz g� � S �N� � ��np�array��z�x�ys �A/usr/lib/python3/dist-packages/scipy/optimize/tests/test__root.py�funcz)TestRoot.test_tol_parameter.<locals>.func �0 � ��D�A�q��8�Q��T�A�X�q�!�t�a�x� 0�1�1� 1� c �X � | \ }}t j d|dz z dgdd|dz z gg� � S )Nr � r r r s r �dfuncz*TestRoot.test_tol_parameter.<locals>.dfunc s8 � ��D�A�q��8�a��1��f�a�[�1�a��1��f�+� 6�7�7� 7r )�hybr�lm�broyden1�broyden2�anderson� diagbroyden�krylov)� linearmixing�excitingmixing)r r 皙�����?g-C ��6?)�jac�tol�method� �?z: z vs. )r r r �success�abs�max)�selfr r r( r& �sol1�sol2�msgs r �test_tol_parameterzTestRoot.test_tol_parameter sD � � 2� 2� 2� 8� 8� 8�0� � �F��;�;�;��� �'�'��������s�3�i�S�d�6�J�J�J�D���s�3�i�S�c�&�I�I�I�D�%+�V�V�T�T�$�&�\�\�\�\�4�4���<�<�<�H�C� �D�L�#� &� &� &� �D�L�#� &� &� &� �C���T�V� � �%�%�)�)�+�+�c�$�$�t�v�,�,�.?�.?�.C�.C�.E�.E�E�� � � � �! � r c �x � d� }dD ]3}t t j t j d� � |d|i�� � �4d S )Nc �, � t | d � � S �Nr )r+ )r s r �normz$TestRoot.test_tol_norm.<locals>.norm+ s � ��q��t�9�9� r )r$ r! r# r r r r" r �tol_norm)r( �options)r r � zeros_like�zeros)r- r5 r( s r � test_tol_normzTestRoot.test_tol_norm) sb � � � � �!� ,� ,�F� �� ���� � �F�#�T�*� ,� ,� ,� ,� ,� ,� ,r c �6 � dd�}t |ddgd�� � d S )Nr c �P � | \ }}t j |dz dz |dz |z g� � S r r )r �fr r s r r z=TestRoot.test_minimize_scalar_coerce_args_param.<locals>.func; r r r% g �?)�args)r r )r- r s r �&test_minimize_scalar_coerce_args_paramz/TestRoot.test_minimize_scalar_coerce_args_param9 s7 � � 2� 2� 2� 2� �T�C��:�C�(�(�(�(�(�(r c � � G d� d� � } |� � }t t � � 5 t |ddgd�� � d d d � � d S # 1 swxY w Y d S )Nc � � e Zd Zd� Zd� ZdS )�!TestRoot.test_f_size.<locals>.func � � d| _ d S r4 ��count)r- s r �__init__z*TestRoot.test_f_size.<locals>.fun.__init__E s � � �� � � r c � � | xj dz c_ | j dz s$|d d|d |d z dz z z dz }nC|d d|d |d z dz z z dz d|d |d z dz z |d z g}|S )Nr � r r) r g �?rD )r- r �rets r �__call__z*TestRoot.test_f_size.<locals>.fun.__call__H s� � �� � �a�� � � � �Q�� =��A�$���!��q��t� ��'9�!9�9�C�?�C�C� �a�D�3�!�A�$��1��+�!�);�#;� ;�c� A� �!�A�$��1��+�!�!3�3�a��d�:� <�C� � r N)�__name__� __module__� __qualname__rF rJ � r r �funrB D s2 � � � � � � � � � � � � � r rO g�������?g r )r( )� assert_raises� ValueErrorr )r- rO �Fs r � test_f_sizezTestRoot.test_f_size@ s� � � � � � � � � � � �C�E�E�� �:� &� &� -� -� ��S�#�J�t� ,� ,� ,� ,� -� -� -� -� -� -� -� -� -� -� -� -���� -� -� -� -� -� -s �A�A�AN)rK rL rM r1 r: r? rS rN r r r r sP � � � � � �� � �:,� ,� ,� )� )� )�-� -� -� -� -r r ) �__doc__� numpy.testingr �pytestr rP �numpyr �scipy.optimizer r rN r r �<module>rY s� ��� � "� !� !� !� !� !� *� *� *� *� *� *� � � � � � � � � � �J-� J-� J-� J-� J-� J-� J-� J-� J-� J-r
Save File
Cancel