← Back
Editing: test_splines.cpython-311.pyc
� d�c� � �� � d Z ddlZddlZddlmZ ddlmZ d� Zd d�Z ej � dg d �� � ej � d g d �� � d � � � � � Z dS )zTests for spline filtering.� N)�assert_almost_equal)�ndimagec �8 � dgdgddgddgg d�g d�d�}|| S )z0Knot values to the right of a B-spline's center.� � � )�� �L r )�B � r �r r � � r � � )�order� knot_valuess �B/usr/lib/python3/dist-packages/scipy/ndimage/tests/test_splines.py�get_spline_knot_valuesr sA � ��c��c��!�f��!�f�"�l�l�!�k�k� #� #�K� �u� � � �mirrorc � � t |� � }t j | | f� � }t |� � D ]<\ }}t j || � � }|dk r||||f< �(|||||z f< ||||z |f< �=|d dt |dd� � � z z }|dk rd\ } } n:|dk rd\ } } n.|d k rd \ } } n"t d � |� � � � �t t |� � dz � � D ]Z} t || dz d� � � D ]=\ } }|| | | | z z fxx |z cc< || dz | dz | | z z fxx |z cc< �>�[||z S ) z1Matrix to invert to find the spline coefficients.r r r Nr )r r �reflect)r r � grid-wrap)���r zunsupported mode {}) r �np�zeros� enumerate�arange�sum� ValueError�format�range�len) �nr �moder �matrix�diag� knot_value�indices�knot_values_sum�start�step�row�idxs r �make_spline_knot_matrixr0 s� � �(��/�/�K� �X�q�!�f� � �F� %�k� 2� 2� 9� 9���j��)�D�!�$�$�� �1�9�9�'1�F�7�G�#� $� $�.8�F�7�G�d�N�*� +�.8�F�7�T�>�7�*� +� +�!�!�n�q�3�{�1�2�2��+?�+?�'?�?�O� �x���� ��t�t� �� � �� ��t�t� �� � � � ��t�t��.�5�5�d�;�;�<�<�<��S��%�%��)�*�*� B� B�� (��S�1�W�X�X�)>� ?� ?� B� B�O�C�� �3���S��(�(� )� )� )�Z� 7� )� )� )� �C�4�!�8�e�V�a�Z�$�s�(� 2�2� 3� 3� 3�z� A� 3� 3� 3� 3� B� �O� #�#r r r r& )r r r c �Z � d}t j |t �� � }t j |d| |�� � }t j |d| |�� � }t || |�� � }t |t j ||� � � � t |t j ||j � � � � d S )N�d )�dtyper )�axisr r& r )r& ) r �eye�floatr �spline_filter1dr0 r �dot�T)r r& r% r5 �spline_filter_axis_0�spline_filter_axis_1r' s r �%test_spline_filter_vs_matrix_solutionr< 6 s� � � �A� �&��%� � � �C�"�2�3�Q�e�8<� >� >� >��"�2�3�Q�e�8<� >� >� >�� $�Q��D� 9� 9� 9�F���R�V�$8�&� A� A�B�B�B���R�V�$8�&�(� C� C�D�D�D�D�Dr )r ) �__doc__�numpyr �pytest� numpy.testingr �scipyr r r0 �mark� parametrizer< r r r �<module>rD s� �� !� !� � � � � � � � � -� -� -� -� -� -� � � � � � � � � � $� $� $� $�@ �����"4�"4�"4�5�5������!C�!C�!C�D�D� E� E� E�D� 6�5� E� E� Er
Save File
Cancel