← Back
Editing: test_backend.cpython-311.pyc
� d�c� � � � d dl mZ d dlZd dlZd dlmZmZmZ d dl m Z d dl m Z m Z d dlZdZej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej ej ej j ej ej! ej" ej# ej$ ej% ej& ej' ej( ej) eej* dd� � � eej+ dd� � � f Z,ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j! ej j" ej j# ej j$ ej j% ej j& ej j' ej j( ej j) eej j* dd� � � eej j+ dd� � � f Z-e j e j e j e j e j e j e j e j e j e j e j e j e j e j e j e j e j e j! e j" e j# e j$ e j% e j& e j' e j( e j) e j* e j+ f Z.ej/ �0 d e1e-e,e.� � � � d � � � Z2ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j ej j! fZ3e j e j e j e j e j e j e j e j e j e j e j e j e j e j e j e j e j e j! fZ4ej/ �0 d e1e3e4� � � � d � � � Z5dS )� )�partialN)�_fftlog� _pocketfft� set_backend)� mock_backend)�assert_allclose� assert_equal)�fft�fft2�fftn�ifft�ifft2�ifftn�rfft�rfft2�rfftn�irfft�irfft2�irfftn�dct�idct�dctn�idctn�dst�idst�dstn�idstn�fht�ifht� g �?)�dln�muzfunc, np_func, mockc � � t j d� � � d� � } ||� � }t | |� � |d�� � t t d�� � 5 d|_ | |� � }t ||j � � t |j d� � d d d � � n # 1 swxY w Y t | |� � |d�� � d S ) N� �� r g�����|�=)�atolT��onlyr � ) �np�arange�reshaper r r � number_callsr � return_value)�func�np_func�mock�x�answer�ys �>/usr/lib/python3/dist-packages/scipy/fft/tests/test_backend.py�test_backend_callr7 8 s � � � �"� � � � �f�%�%�A� �W�Q�Z�Z�F��D�D��G�G�V�%�0�0�0�0� �\�� -� -� -� +� +� ��� �D��G�G���Q��)�*�*�*��T�&��*�*�*� +� +� +� +� +� +� +� +� +� +� +���� +� +� +� +� �D�D��G�G�V�%�0�0�0�0�0�0s �$=B-�-B1�4B1z func, mockc �� � t j d� � � d� � }t j t d�� � 5 | |d�� � d d d � � n # 1 swxY w Y t t d�� � 5 d |_ | |d�� � }t ||j � � t |j d � � t |j d d d� � d d d � � d S # 1 swxY w Y d S ) Nr$ r% zprecomputed plan)�match�foo)�planTr( r r* r; ) r+ r, r- �pytest�raises�NotImplementedErrorr r r. r r/ � last_args)r0 r2 r3 r5 s r6 �test_backend_planr@ V su � � � �"� � � � �g�&�&�A� � �*�2D� E� E� E� � � ��Q�U����� � � � � � � � � � � ���� � � � � �\�� -� -� -� 7� 7� ��� �D��� � � ���Q��)�*�*�*��T�&��*�*�*��T�^�A�&�v�.��6�6�6� 7� 7� 7� 7� 7� 7� 7� 7� 7� 7� 7� 7���� 7� 7� 7� 7� 7� 7s% �A � A!�$A!�>A C+�+C/�2C/)6� functoolsr �numpyr+ � scipy.fft�scipyr r r �scipy.fft.testsr � numpy.testingr r r< �fnamesr r r r r r r r r r r r �hfft�hfft2�hfftn�ihfft�ihfft2�ihfftnr r r r r r r r r r �np_funcs�funcs�mocks�mark� parametrize�zipr7 � plan_funcs� plan_mocksr@ � � r6 �<module>rX s< �� � � � � � � � � � � � � � � 6� 6� 6� 6� 6� 6� 6� 6� 6� 6� (� (� (� (� (� (� 7� 7� 7� 7� 7� 7� 7� 7� � � � � �� �F�J��� �R�V�[� �F�K��� �r�v�|� �F�K��� �r�v�|� �F�L�"�&�-��� � �F�K��)�:�+;� �F�L�*�+�Z�->� �N�J�O�Z�_�j�>N� �N�J�O�Z�_�j�>N� �G�G�K�Q�3� /� /� /� �G�G�L�a�C� 0� 0� 0� 2�� �� �� ��� �� �� ��� �%�)�/� �� ��� �%�)�/� �� �%�)�*�E�I�,<� �� ��� �%�)�/� �� �%�)�*�E�I�,<� �� �� ��� ��� �� �� �� ��� ��� �� ���� �A�#� .� .� .� ���� �Q�3� /� /� /� 1�� � �<�,�l�.?� � �L� .� �0B� � �L� .� �0B� � �\� 0�,�2E� � �L� .� �0B� � �\� 0�,�2E� � �<�,� � �L� .� � �<�,� � �L� .� � �<�,� .�� ����.���E�8�U�0K�0K�L�L� 1� 1� M�L� 1� �i�m�U�Y�^�U�Y�^��i�n�e�i�o�u�y���i�n�e�i�o�u�y���i�o�u�y� /���1A��i�n�e�i�o�u�y���i�o�u�y� /���1A� C� � � � � 1�<�3D�� � �!3�\�5G�� � �!3�\�5G�� �,�"5�|�7J�� � �!3�\�5G�� �,�"5�|�7J� L� � ���� �s�s�:�z�'B�'B�C�C� 7� 7� D�C� 7� 7� 7rW
Save File
Cancel