← Back
Editing: test_c_api.cpython-311.pyc
� d�c� � � � d dl Zd dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d� d� d � d � gZ d � d � d � d� gZ d� d� d� d� gZ d� Zd� Zd� ZdS )� N)�assert_allclose)�ndimage)�_ctest)�_cytest)�LowLevelCallablec �* � t j | � � S �N)r �filter1d�� filter_sizes �@/usr/lib/python3/dist-packages/scipy/ndimage/tests/test_c_api.py�<lambda>r s � ��� �4�4� � c �. � t j | d�� � S �NF��with_signature)r r r s r r r s � ��(��U�K�K�K� r c �H � t t j | d�� � � � S �NTr )r r r r s r r r s � �(��)9�+�VZ�)[�)[�)[�\�\� r c �\ � t j t dt j | � � � � S )N� _filter1d)r � from_cythonr �filter1d_capsuler s r r r s( � �(�4�W�k�5<�5M�k�5Z�5Z�\� \� r c �* � t j | � � S r )r �filter2d��weightss r r r s � �F�O�G�,�,� r c �. � t j | d�� � S r )r r r s r r r s � �G�$�W�U�C�C�C� r c �H � t t j | d�� � � � S r )r r r r s r r r s � �$�W�%5�g�d�%S�%S�%S�T�T� r c �\ � t j t dt j | � � � � S )N� _filter2d)r r r �filter2d_capsuler s r r r s" � �$�0��+�w�G_�`g�Gh�Gh�i�i� r c �* � t j | � � S r )r � transform��shifts r r r s � �&�"�5�)�)� r c �. � t j | d�� � S r )r r$ r% s r r r s � �'�#�E�%�@�@�@� r c �H � t t j | d�� � � � S r )r r r$ r% s r r r s � �"�7�#4�U�4�#P�#P�#P�Q�Q� r c �\ � t j t dt j | � � � � S )N� _transform)r r r �transform_capsuler% s r r r s" � �"�.�w� �g�F_�`e�Ff�Ff�g�g� r c �b �� d� ��fd�} t t � � D ]\ }} | |� � �d S )Nc �0 � || z � � � S r )�sum)�footprint_elementsr s r r z%test_generic_filter.<locals>.filter2d! s � ��*�*�/�/�1�1�1r c � �� t | }t j d� � }d|d d�d d�f< t j g d�g d�g d�g� � }t j |� � }t j |� � |z }t j | ||� � |�� � }t j |�||f�� � }t ||d� | � � � � � d S ) N)� r1 r � )r � r )r3 r3 r3 )� footprint)r4 �extra_arguments� #{} failed��err_msg) �FILTER2D_FUNCTIONS�np�ones�array� count_nonzeror �generic_filterr �format) �j�func�imr4 �footprint_sizer �res�stdr s �r �checkz"test_generic_filter.<locals>.check$ s� �� �!�!�$�� �W�X� � ����3�B�3�s��s�7� ��H�i�i�i����I�I�I� >�?�?� ��)�)�4�4���'�.�)�)�.�8���$�R���g� � �/8�:� :� :���$�R��Y�6=�Z�A� A� A����S�,�*=�*=�a�*@�*@�A�A�A�A�A�Ar )� enumerater9 )rF r@ rA r s @r �test_generic_filterrH sg �� �2� 2� 2� B� B� B� B� B� � /�0�0� � ���4� ��a������ r c �b �� d� ��fd�} t t � � D ]\ }} | |� � �d S )Nc � � t |j � � D ]2}d||< t |� � D ]}||xx | ||z z cc< � �3||z }d S )Nr )�range�size)� input_line� output_liner �ir@ s r r z'test_generic_filter1d.<locals>.filter1d8 so � ��{�'�(�(� 2� 2�A� �K��N��;�'�'� 2� 2���A����*�Q�q�S�/�1����� 2��{�"� � � r c �z �� t | }t j t j t j d� � t j d� � f� � d� � }d}t j | ||� � |� � }t j |�||f�� � }t ||d� | � � �� � d S )Nr2 )r2 r3 � �r5 r6 r7 ) �FILTER1D_FUNCTIONSr: �tile�hstack�zerosr; r �generic_filter1dr r? )r@ rA rB r rD rE r s �r rF z$test_generic_filter1d.<locals>.check? s� �� �!�!�$�� �W�R�Y���� � �b�g�b�k�k� :�;�;�W� E� E��� ��&�r�4�4� �+<�+<�'2�4� 4���&�r�8�[�8C�~�G� G� G����S�,�*=�*=�a�*@�*@�A�A�A�A�A�Ar )rG rS )rF r@ rA r s @r �test_generic_filter1drX 7 sg �� �#� #� #� B� B� B� B� B� � /�0�0� � ���4� ��a������ r c �b �� d� ��fd�} t t � � D ]\ }} | |� � �d S )Nc �. � | d |z | d |z fS )Nr r3 � )�output_coordinatesr& s r r$ z+test_geometric_transform.<locals>.transformP s" � �!�!�$�u�,�.@��.C�e�.K�K�Kr c �f �� t | }t j d� � � dd� � � t j � � }d}t j | ||� � � � }t j |�|f�� � }t ||d� | � � �� � d S )N� � rQ g �?rR r6 r7 ) �TRANSFORM_FUNCTIONSr: �arange�reshape�astype�float64r �geometric_transformr r? )r@ rA rB r& rD rE r$ s �r rF z'test_geometric_transform.<locals>.checkS s� �� �"�1�%�� �Y�r�]�]� "� "�1�a� (� (� /� /�� � ;� ;�����)�"�d�d�5�k�k�:�:���)�"�i�%��R�R�R����S�,�*=�*=�a�*@�*@�A�A�A�A�A�Ar )rG r` )rF r@ rA r$ s @r �test_geometric_transformrf O sj �� �L� L� L�B� B� B� B� B� � 0�1�1� � ���4� ��a������ r )�numpyr: � numpy.testingr �scipyr � scipy.ndimager r �scipy._lib._ccallbackr rS r9 r` rH rX rf r[ r r �<module>rl s �� � � � � )� )� )� )� )� )� � � � � � � � � � � � � !� !� !� !� !� !� 2� 2� 2� 2� 2� 2� 5�4�K�K�\�\�\� \� � � -�,�C�C�T�T�i�i� � � *�)�@�@�Q�Q�g�g� � �� � �.� � �0� � � � r
Save File
Cancel