← Back
Editing: test__procrustes.cpython-311.pyc
� d�cn � �T � d dl Zd dlmZmZmZ d dlmZ d dl m Z G d� d� � Z dS )� N)�assert_allclose� assert_equal�assert_almost_equal)�raises)� procrustesc �8 � e Zd Zd� Zd� Zd� Zd� Zd� Zd� Zd� Z dS ) �TestProcrustesc � � t j ddgddgddgddggd� � | _ t j ddgddgddgddggd� � | _ t j ddgddgddgdd ggd� � | _ t j d z }t j dd gd dgd d gd d ggd� � t j d� � z | _ t j t j |� � t j |� � gt j t j dz |z � � t j t j dz |z � � gt j |� � t j |� � gt j t j dz |z � � t j t j dz |z � � ggd� � t j d� � z | _ d S )zcreates inputs� � � �d� ������i����g @� r ���N) �np�array�data1�data2�data3�pi�sqrt�data4�cos�sin�data5)�self� shiftangles �F/usr/lib/python3/dist-packages/scipy/spatial/tests/test__procrustes.py� setup_methodzTestProcrustes.setup_method s� � � �X��1�v��1�v��1�v��1�v� >��D�D�� � �X��2�w��B��!�R��1�b�'� B�C�H�H�� � �X��1�v��1�v��1�v��3�x� @�#�F�F�� � �U�Q�Y� ��X��1�v��1�v��A�w� �"�g� '�(+�-� -�/1�w�q�z�z�:�� ��X���z� 2� 2�B�F�:�4F�4F� G� !�v�b�e�a�i�*�&<� =� =� !�v�b�e�a�i�*�&<� =� =� ?� "��z� 2� 2� 2� "��z� 2� 2� 2� 4� "��r�u�q�y�:�'=� >� >� >� "��r�u�q�y�:�'=� >� >� >� @� A� "�#� #� &(�W�Q�Z�Z�0�� � � � c �$ � t | j | j � � \ }}}t ||� � t |d� � t | j | j � � \ }}}t || j � � t | j | j � � \ }}} d S )Ng ) r r r r r r r r r ) r �a�b� disparity�m4�m5�disp45�m1�m3�disp13s r! �test_procrustesz TestProcrustes.test_procrustes% s� � � %�T�Z���<�<���1�i���1� � � ��I�r�*�*�*� $�D�J�� �;�;���B���R���$�$�$� $�D�J�� �;�;���B���r# c � � t | j | j � � \ }}}t | j | j � � \ }}}t ||� � t j g d�g d�g d�g d�g d�g d�g d�g d�g� � }t j g d �g d �g d �g d �g d �g d�g d�g d�g� � }t ||� � \ } } }t ||� � \ } } }t ||� � d S )N)gߐ����@g���Ĉ�?g��YA���?)gR;�E�Q�?g��z�O��g/߸� cԿ)g��wKi�?g��c3�O�?g� ٣�ÿ)g�) �A @g���J �?g�:o����?)g�ً �3�?gvO�\���?g���<Р?)g}v#����?g�P#��@g�-�@��)g�N g;��?go{��1��?g�g=����?)g��W�$�g�����*�?gD��W���?)gUϓ|6��?g�|_;��?g�\��X3ƿ)g�8 ���g[����g�7~����)gV�D�?s�?g $�S���g�B9��?)g ��d��?g� q�6�ɿglx����?)g#�)w<��?g�y�w<5�?g�;{���?)g��xi7��?g9��JN�gv[�Sj�)g�YN��g��g�j�$�#��g^p��տ)g��D�Ѝ�g6�t(\ҿg1�K[���?)r r r r r r ) r r+ r, r- �m3_2�m1_2�disp31�rand1�rand3�res1�res3�res3_2�res1_2s r! �test_procrustes2z TestProcrustes.test_procrustes28 sN � �#�D�J�� �;�;���B�� '�� �D�J� ?� ?���d�F��F�F�+�+�+� ��>�>�>�?�?�?�>�>�>�=�=�=�=�=�=�>�>�>�=�=�=�=�=�=�?� @� @�� ��<�<�<�@�@�@�>�>�>�>�>�>�=�=�=�?�?�?�?�?�?�>�>�>�@� A� A�� (��u� 5� 5���d�F�!+�E�5�!9�!9� �����F�F�+�+�+�+�+r# c � � t t t t j ddgddgg� � t j g d�g d�g� � � � d S )Nr r r r )� � � )r � � �� assert_raises� ValueErrorr r r �r s r! � test_procrustes_shape_mismatchz-TestProcrustes.test_procrustes_shape_mismatchT sV � ��j�*��h��A���A�� /�0�0��h� � � �:�:�:� 6�7�7� 9� 9� 9� 9� 9r# c �h � t j g g� � }t t t ||� � d S )N)r r rA rB r )r �emptys r! �"test_procrustes_empty_rows_or_colsz1TestProcrustes.test_procrustes_empty_rows_or_colsY s+ � ���"������j�*�e�U�;�;�;�;�;r# c � � t t t t j ddgddgg� � t j ddgddgg� � � � d S )N�* �- r@ rC s r! � test_procrustes_no_variationz+TestProcrustes.test_procrustes_no_variation] sV � ��j�*��h��R��2�r�(� 3�4�4��h��R��2�r�(� 3�4�4� 6� 6� 6� 6� 6r# c �0 � t t t t j g d�� � t j ddgddgg� � � � t t t t j g d�� � t j g d�� � � � t t t t j d� � t j d� � � � t t t t j dgdggg� � t j dd ggg� � � � d S ) N)r r r r r; r r r r r r= � r; � r@ rC s r! �(test_procrustes_bad_number_of_dimensionsz7TestProcrustes.test_procrustes_bad_number_of_dimensionsb s� � ��j�*��h� 1� 1� 1�2�2��h��A���A�� /�0�0� 2� 2� 2� �j�*��h� 1� 1� 1�2�2��h� 1� 1� 1�2�2� 4� 4� 4� �j�*�b�h�q�k�k�2�8�B�<�<�H�H�H� �j�*��h�"���s� �}�-�-��h�!�R�� �{�+�+� -� -� -� -� -r# N) �__name__� __module__� __qualname__r" r. r9 rD rG rK rO � r# r! r r s} � � � � � �0� 0� 0�8<� <� <�&,� ,� ,�89� 9� 9� <� <� <�6� 6� 6� -� -� -� -� -r# r ) �numpyr � numpy.testingr r r �pytestr rA � scipy.spatialr r rS r# r! �<module>rX s� �� � � � � L� L� L� L� L� L� L� L� L� L� *� *� *� *� *� *� $� $� $� $� $� $�k-� k-� k-� k-� k-� k-� k-� k-� k-� k-r#
Save File
Cancel