← Back
Editing: test_fftlog.cpython-311.pyc
� d�c1 � �L � d dl Z d dlZd dlmZ d dlZd dlZd dlmZm Z m Z d dl m Z d� Z ej � dddg� � ej � d g d �� � ej � d g d �� � ej � d ddg� � d� � � � � � � � � Zej � ej � � � d� � d�� � d� � � Zej � d ddg� � d� � � ZdS )� N)�assert_allclose)�fht�ifht� fhtoffset)�pochc �V � d� } t j ddd� � }t j |d |d z � � }d}d}d} | ||� � }t |||||� � � }g d �}t ||� � t |||� � � }t |||||� � � }g d �}t ||� � d }t |||� � � }t |||||� � � }g d�}t ||� � d}t |||� � � }t |||||� � � }g d�}t ||� � d S )Nc �J � | |dz z t j | dz dz � � z S )N� � )�np�exp)�r�mus �=/usr/lib/python3/dist-packages/scipy/fft/tests/test_fftlog.py�fz&test_fht_agrees_with_fftlog.<locals>.f s' � ��2�a�4�y����A���a�� � �(�(� ���� � r r g333333�?� ��offset�bias)g��=lS�g�l�1�Z?gp� R��_�g�~]� o?g��0d �@?g��� ���?g0�2�u�?g7c 4Uo�?gU��/5�?g�ޠ3��J�g�Q�i��I�g!m ���9?g ���1�g?��J�b7?g �tj3B�g �� 'K?�r )g �Dy��?g �T�I�g ��9�4?g�B ��M?g�Q�89�w?g��ĥ��?gZ�T3%�?gz;_x���?g9%P��?gv@���݆�gA�A��Y]?gayi��g;�gG�I<!?g F��\m �g�H��� ?g @(�LG��g�������?)g��U�_ �g0E>y7��?g4⬞ F�?gR^ء*9��g�-�g��?g ���!a�?gR�� �?g�m�MUU�?g� ��.�?g'?�eH��g�Z7Z��P?g�mܷ`�%�gCG��T?g�2�T�g҅���u�>g�/�&� ��g�������)g�G��3��>g`4]<]?g[8����+?g,8G�d�P?g��� �u?g�,ח$5�?g~�}���?gr�a,�4�?g�V�����?gh�`>E��gDg�yٖ?g�P2�ZS��g͆�˝��?gܺJ����?g[�U�+@g�"�� %@)r �logspace�logr r r ) r r �dlnr r r �a�ours�theirss r �test_fht_agrees_with_fftlogr! s� � � )� )� )� � �B��2� � �A� �&��1��a��d�� � �C� �B� �F� �D� ��!�R���A� �q�#�r�&�t� 4� 4� 4�D�@� @� @�F� �D�&�!�!�!� �s�B�T� *� *� *�F� �q�#�r�&�t� 4� 4� 4�D�@� @� @�F� �D�&�!�!�!� �D� �s�B�T� *� *� *�F� �q�#�r�&�t� 4� 4� 4�D�@� @� @�F� �D�&�!�!�!� �D� �s�B�T� *� *� *�F� �q�#�r�&�t� 4� 4� 4�D�@� @� @�F� �D�&�!�!�!�!�!r �optimalTFr )r � �?g �r )r g�������?g���������n�@ �? c �` � t j � d� � }|� | � � }|� dd� � }|� dd� � }|rt ||||�� � }t |||||�� � }t |||||�� � } t || � � d S )N� �E3 ���r ���r ��initialr r ) r �random� RandomState�standard_normal�uniformr r r r ) r$ r r r" �rngr r r �A�a_s r �test_fht_identityr4 S s� � � �)� � � � +� +�C� ���A� � �A� �+�+�b�!� � �C� ���R�� � �B�� ?��3��F��>�>�>�� �A�s�B�v�D�1�1�1�A� �a��b��d� 3� 3� 3�B��A�r�����r �mips64z test known to fail on mips64el)�reasonc �< � t j � d� � } | � d� � }| � dd� � }d\ }}t j d�� � 5 }t ||||�� � |r J d � � � d d d � � n # 1 swxY w Y d \ }}t j d�� � 5 }t ||||�� � |r J d � � � d d d � � n # 1 swxY w Y d \ }}t j t � � 5 }t ||||�� � |s J d � � � d d d � � n # 1 swxY w Y d \ }}t j t � � 5 }t ||||�� � |s J d � � � d d d � � d S # 1 swxY w Y d S )Nr( r% r) r )g �r# T)�recordr z)fht warned about a well-defined transform)g �� �?)g �r9 z+fht did not warn about a singular transformz,ifht did not warn about a singular transform) r r- r. r/ r0 �warnings�catch_warningsr �pytest�warns�Warningr )r1 r r r r r8 s r �test_fht_special_casesr? g s � � �)� � � � +� +�C� ���B� � �A� �+�+�b�!� � �C� �H�B�� � �� -� -� -� G�� �A�s�B�T�"�"�"�"��F�F�F�F�F�F�F�G� G� G� G� G� G� G� G� G� G� G���� G� G� G� G� �H�B�� � �� -� -� -� G�� �A�s�B�T�"�"�"�"��F�F�F�F�F�F�F�G� G� G� G� G� G� G� G� G� G� G���� G� G� G� G� �H�B�� � �g� � � E�&� �A�s�B�T�"�"�"�"��D�D�D�D�D�D�D�E� E� E� E� E� E� E� E� E� E� E���� E� E� E� E� �H�B�� � �g� � � F�&� �Q��R�d�#�#�#�#��E�E�E�E�E�E�E�F� F� F� F� F� F� F� F� F� F� F� F���� F� F� F� F� F� FsH �% B�B�B�7 C$�$C(�+C(� D:�:D>�D>�# F�F�Fc � � t j � d� � }|� dd� � }|� d|z d� � }t j dd| � � }||z }t j |d |d z � � }t ||d |� � � }t |||||� � � }t j |� � |d d d� z } d| z |z t |dz |z dz |� � z } t || � � d S ) Nr( r � r) r9 r* r r r r+ r ) r r- r. r0 r r r r r r r ) r$ r1 r �gammar r r r r2 �k�Ats r �test_fht_exactrE � s � � �)� � � � +� +�C� ���Q�� � �B� �K�K��2��s� #� #�E� � �B��1� � �A� �5��A� �&��1��a��d�� � �C� �s�B��%� 8� 8� 8�F� �A�s�B�v�E�2�2�2�A� ��v���q���2��w� �A� �A�#�� ��b��d�5�j�!�^�U�3�3� 3�B��A�r�����r )r: �numpyr � numpy.testingr r< �platform�scipy.fft._fftlogr r r � scipy.specialr r! �mark� parametrizer4 �skipif�machine� startswithr? rE � r r �<module>rQ s� �� ���� � � � � )� )� )� )� )� )� � � � � ���� 2� 2� 2� 2� 2� 2� 2� 2� 2� 2� � � � � � �E"� E"� E"�P �����T�5�M�2�2������#3�#3�#3�4�4������ � � �0�0������r�2�h�'�'� � � (�'� 1�0� 5�4� 3�2� � ����$�H�$�&�&�1�1�(�;�;�Dd��e�e� F� F� f�e� F�B �����r�2�h�'�'�� � (�'�� � r
Save File
Cancel