← Back
Editing: test_trustregion.cpython-311.pyc
� d�c] � �~ � d Z ddlZddlZddlmZmZmZ ddlm Z m Z m Z m Z m Z G d� d� � Z G d� d� � ZdS ) zt Unit tests for trust-region optimization routines. To run it in its simplest form:: nosetests test_optimize.py � N)�assert_� assert_equal�assert_allclose)�minimize�rosen� rosen_der� rosen_hess�rosen_hess_prodc � � e Zd ZdZd� Zd� ZdS )� Accumulatorz This is for testing callbacks.c �" � d| _ d | _ d S )Nr )�count�accum��selfs �G/usr/lib/python3/dist-packages/scipy/optimize/tests/test_trustregion.py�__init__zAccumulator.__init__ s � ��� ��� � � � c � � | xj dz c_ | j �t j |� � | _ d S | xj |z c_ d S )N� )r r �np�array)r �xs r �__call__zAccumulator.__call__ s? � � � � �a�� � � �:� ���!���D�J�J�J� �J�J�!�O�J�J�J�Jr N)�__name__� __module__� __qualname__�__doc__r r � r r r r s8 � � � � � �)�)�� � � � � � � r r c �D � e Zd Zd� Zd� Zd� Zd� Zd� Zd� Zd� Z d� Z d � Z d S ) �TestTrustRegionSolversc �< � ddg| _ ddg| _ ddg| _ d S )Ng �?g @g333333�)�x_opt� easy_guess� hard_guessr s r � setup_methodz#TestTrustRegionSolvers.setup_method s' � ��3�Z�� � ��*��� ��+����r c � � | j }t t |t t ddddi�� � }t ||d d � � t |d |d d � � t |d | j � � d S ) N�:�0�yE>�dogleg� return_allT��jac�hess�tol�method�options�allvecsr r ���)r% r r r r r r# �r �x0�rs r �test_dogleg_accuracyz+TestTrustRegionSolvers.test_dogleg_accuracy$ s� � � �_�� �U�B�I�J�D� $�|�T�.B� E� E� E����A�i�L��O�,�,�,���#���)� �R� 0�1�1�1���#��� �+�+�+�+�+r c � � t � � }d}t t | j t t |dd|d��� � }t |j |� � t t |d � � |dz � � t |d |d d � � t t |d dd � � � |j � � d S ) N� r) T)r* �maxiter)r, r- �callbackr/ r0 r1 r r r2 ) r r r r% r r r r �lenr �sumr )r � accumulatorr9 r5 s r �test_dogleg_callbackz+TestTrustRegionSolvers.test_dogleg_callback- s� � �!�m�m� ��� �U�D�O��� )�(�,0�W� E� E� H� H� H�� �[�&��0�0�0��S��9� �&�&��� �2�2�2���#���)� �R� 0�1�1�1���A�i�L���� ,�-�-�{�/@�A�A�A�A�Ar c �� � t j t d�� � 5 t t | j t t dddd��� � d d d � � d S # 1 swxY w Y d S )Nz Maximum number of iterations)�matchr) Tr )�dispr9 )r, r- r/ r0 )�pytest�warns�RuntimeWarningr r r% r r r s r �test_dogleg_user_warningz/TestTrustRegionSolvers.test_dogleg_user_warning9 s� � � �\�.� ?�A� A� A� =� =� �U�D�O��$�X�&*�q� 9� 9� =� =� =� =� =� =� =� =� =� =� =� =� =� =� =� =���� =� =� =� =� =� =s �-A�A� Ac �� � t }t }t }| j | j fD �];}t ||||ddddi�� � }t ||||ddddi�� � }t ||||ddddi�� � }t ||||ddddi�� � }t ||||dd ddi�� � } t | j |d � � t | j |d � � t | j |d � � t | j |d � � t | j | d � � t t |d � � t |d � � k � � ��=d S ) Nr( r) r* Tr+ � trust-ncg� trust-krylovz newton-cg� trust-exactr r1 ) r r r r$ r% r r r# r r; ) r �f�g�hr4 �r_dogleg� r_trust_ncg�r_trust_krylov�r_ncg� r_iteratives r �test_solver_concordancez.TestTrustRegionSolvers.test_solver_concordance@ s� � � �� �� ���?�D�O�4� F� F�B� ��2�1�1�$�'/�,��9M�O� O� O�H�"�1�b�a�a�T�*5�,8�$�+?�A� A� A�K� &�a�����*8�,8�$�+?� A� A� A�N� �Q����t�$/�,��9M�O� O� O�E�"�1�b�a�a�T�*7�,8�$�+?�A� A� A�K� �D�J��� � 6� 6� 6� �D�J� �C�(8� 9� 9� 9� �D�J��s�(;� <� <� <� �D�J��c� � 3� 3� 3� �D�J� �C�(8� 9� 9� 9� �C���+�,�,�s�5��3C�/D�/D�D� E� E� E� E�' F� Fr c � � | j | j | j fD ]A}t t |t t dd�� � }t | j |d � � �Bd S )Nr( rG )r, �hesspr. r/ r )r$ r% r# r r r r r r3 s r �test_trust_ncg_hesspz+TestTrustRegionSolvers.test_trust_ncg_hessp] sb � ��?�D�O�T�Z�@� 0� 0�B���� � � !�+�7� 7� 7�A� �D�J��#�� /� /� /� /� 0� 0r c � � t t | j t t dd�� � }t | j |d � � d S )Nr( rG �r4 r, r- r. r/ r �r r r# r r r �r r5 s r � test_trust_ncg_start_in_optimumz6TestTrustRegionSolvers.test_trust_ncg_start_in_optimumc sA � � �U�t�z�y�z� �k� 3� 3� 3���� �A�c�F�+�+�+�+�+r c � � t t | j t t dd�� � }t | j |d � � d S )Nr( rH rW r rX rY s r �"test_trust_krylov_start_in_optimumz9TestTrustRegionSolvers.test_trust_krylov_start_in_optimumh sA � � �U�t�z�y�z� �n� 6� 6� 6���� �A�c�F�+�+�+�+�+r c � � t t | j t t dd�� � }t | j |d � � d S )Nr( rI rW r rX rY s r �!test_trust_exact_start_in_optimumz8TestTrustRegionSolvers.test_trust_exact_start_in_optimumm sA � � �U�t�z�y�z� �m� 5� 5� 5���� �A�c�F�+�+�+�+�+r N) r r r r& r6 r> rE rR rU rZ r\ r^ r r r r! r! s� � � � � � �&� &� &� ,� ,� ,� B� B� B�=� =� =�F� F� F�:0� 0� 0� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,� ,r r! )r rB �numpyr � numpy.testingr r r �scipy.optimizer r r r r r r! r r r �<module>rb s� ��� � � � � � � � � � @� @� @� @� @� @� @� @� @� @�-� -� -� -� -� -� -� -� -� -� -� -� -� -� � � � � � � � � S,� S,� S,� S,� S,� S,� S,� S,� S,� S,r
Save File
Cancel