← Back
Editing: test_multithreading.cpython-311.pyc
� d�cT � �V � d dl mZ d dlZd dlZd dlmZ d dlZd dlZ ej d�� � d� � � Z ej � dej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej ej! ej" ej# ej$ ej% g� � ej � dd d g� � d � � � � � Z&d � Z'd � Z(d� Z)d� Z*d� Z+dS )� ��fftN��assert_allclose�module)�scopec �B � t j � dd� � S )Ni � )�np�random�randn� � �E/usr/lib/python3/dist-packages/scipy/fft/tests/test_multithreading.py�xr s � � �9�?�?�3�� $� $�$r �func�workers� ���c �Z � || d�� � } || |�� � }t ||� � d S )N� �r r )r r r �expected�actuals r �test_threaded_samer sB � � �t�A�q�!�!�!�H� �T�!�W� %� %� %�F��F�H�%�%�%�%�%r c �. � t j | d�� � S )Nr r r )r s r �_mt_fftr s � � �7�1�a� � � � r c �J � � t j � d�� � }t j d� � 5 }|� t � fd�t d� � D � � � � }d d d � � n # 1 swxY w Y |D ]}t ||� � �t j � d�� � d S )Nr r c � �� g | ]}���S r r )�.0�_r s �r � <listcomp>z0test_mixed_threads_processes.<locals>.<listcomp>& s �� � 2� 2� 2�A�a� 2� 2� 2r � )r �multiprocessing�Pool�mapr �ranger )r �expect�p�res�rs ` r � test_mixed_threads_processesr, s� �� � �W�Q�� "� "� "�F� � �a� � � 4�A��e�e�G� 2� 2� 2� 2��q��� 2� 2� 2�3�3��4� 4� 4� 4� 4� 4� 4� 4� 4� 4� 4���� 4� 4� 4� 4� � #� #����6�"�"�"�"��G�A�q������s �5A-�-A1�4A1c � � t j � � }t j dg| �� � t j t d�� � 5 t j | d�� � d d d � � n # 1 swxY w Y t j t d�� � 5 t j | | dz �� � d d d � � d S # 1 swxY w Y d S )Nr r �workers must not be zero��matchr �workers value out of range)�os� cpu_countr �ifft�pytest�raises� ValueError)r �cpuss r �test_invalid_workersr9 . s= � � �<�>�>�D��H�a�S�4�%� � � � � � �z�)C� D� D� D� � � ���1� � � � � � � � � � � � � � � ���� � � � � � �z�)E� F� F� F� %� %� ���T�E�!�G�$�$�$�$�%� %� %� %� %� %� %� %� %� %� %� %���� %� %� %� %� %� %s$ �A*�*A.�1A.�B8�8B<�?B<c �j � t j � � } t j � � dk sJ �t j d� � 5 t j � � dk sJ �t j d� � 5 t j � � | k sJ � d d d � � n # 1 swxY w Y t j � � dk sJ � d d d � � n # 1 swxY w Y t j � � dk sJ �t j | � � 5 t j � � dk sJ � d d d � � d S # 1 swxY w Y d S )Nr r# r )r2 r3 r � get_workers� set_workers)r8 s r �test_set_get_workersr= : s� � � �<�>�>�D� �?� � �� !� !� !� !� � �� � � &� &��� � �A�%�%�%�%� �_�R� � � -� -��?�$�$��,�,�,�,�,� -� -� -� -� -� -� -� -� -� -� -���� -� -� -� -� �� � �A�%�%�%�%�%� &� &� &� &� &� &� &� &� &� &� &���� &� &� &� &� �?� � �� !� !� !� !� � �$�� � � &� &��� � �A�%�%�%�%�%�&� &� &� &� &� &� &� &� &� &� &� &���� &� &� &� &� &� &sH �.C�/B� C�B �C� B � C�C �C � D(�(D,�/D,c � � t j t d�� � 5 t j d� � 5 d d d � � n # 1 swxY w Y d d d � � n # 1 swxY w Y t j t d�� � 5 t j t j � � dz � � 5 d d d � � n # 1 swxY w Y d d d � � d S # 1 swxY w Y d S )Nr. r/ r r1 r )r5 r6 r7 r r< r2 r3 r r r �test_set_workers_invalidr? K s� � � � �z�)C� D� D� D� � � �_�Q� � � � � � � � � � � � � � � � ���� � � � �� � � � � � � � � � ���� � � � � � �z�)E� F� F� F� � � �_�b�l�n�n�_�Q� .� /� /� � � � � � � � � � � � � � ���� � � � �� � � � � � � � � � � ���� � � � � � sg �A�?� A�A �A�A �A�A� A�<*C �&B4�( C �4B8 �8C �;B8 �<C � C�C),�scipyr �numpyr r5 � numpy.testingr r$ r2 �fixturer �mark� parametrizer4 �fft2�ifft2�fftn�ifftn�rfft�irfft�rfft2�irfft2�rfftn�irfftn�hfft�ihfft�hfft2�ihfft2�hfftn�ihfftn�dct�idct�dctn�idctn�dst�idst�dstn�idstnr r r, r9 r= r? r r r �<module>r^ s� �� � � � � � � � � � � � � � � )� )� )� )� )� )� � � � � � � � � �� �h� � � �%� %� � �%� ������G�S�X�s�x���C�H�c�i��H�c�i���C�J�� �3�:��H�c�i���C�J�� �3�:��G�S�X�s�x����G�S�X�s�x��� "� � � �����Q��G�,�,�&� &� -�,�� �&� !� !� !� � � � %� %� %�&� &� &�"� � � � r
Save File
Cancel