| Server IP : 209.209.40.120 / Your IP : 216.73.217.112 Web Server : Microsoft-IIS/10.0 System : Windows NT NEWWWW 10.0 build 17763 (Windows Server 2019) i586 User : NEWWWW$ ( 0) PHP Version : 8.3.30 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : OFF | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : C:/Python312/Lib/site-packages/pandas/tests/apply/__pycache__/ |
Upload File : |
�
. �g� � � � d dl Zd dlZd dlmc mZ d dlmZm Z d dl
m Z ej d� ej j gZ ej" d dg�� d� � Zd� Zd� Zd � Zej j- d
d � d � d
� d� g� d� � Zej j- dg d�g d�g� d� � Zd� Zd� Zd� Zy)� N)� DataFrame�Index�numba� )�paramsc � � | j S �N)�param)�requests �?C:\Python312\Lib\site-packages\pandas/tests/apply/test_numba.py�
apply_axisr
s � � �=�=�� c � � d� }| j |d|�� }| j |d|�� }t j ||� y )Nc � � | S r � ��xs r �<lambda>z+test_numba_vs_python_noop.<locals>.<lambda> � � �Q� r r ��engine�axis�python)�apply�tm�assert_frame_equal)� float_framer
�func�result�expecteds r �test_numba_vs_python_noopr! sE � � �D�
�
�
�t�G�*�
�
E�F�� � ��h�Z� �H�H����&�(�+r c � � t j d� t dt ddgd�� t ddgd�� � � } d
� }| j |d d �
� }| j |dd �
� }t j ||dd�� y )N�pyarrowr �a�bzstring[pyarrow_numpy]��dtyper �y)�index�columnsc � � | S r r r s r r z3test_numba_vs_python_string_index.<locals>.<lambda># r r r r r r F)�check_column_type�check_index_type)�pytest� importorskipr r r r r )�dfr r r s r �!test_numba_vs_python_string_indexr1 s� � �
��� �"� � ��S�#�J�&=�>��s�C�j�(?�@�
�B�
�D�
�X�X�d�7��X�
3�F��x�x��X�A�x�6�H������E�E�r c �L � t g d�g d�g d�d�t g d�� �� } d� }| j |dd �
� }| j |d d �
� }t j ||� d � }| j |dd
�
� }| j |d d
�
� }t j ||� y )N�r � � �� � � )g @g @g "@�r$ r% �c)�A�B�C�r) c � � | d S )Nr; r r s r r z/test_numba_vs_python_indexing.<locals>.<lambda>0 �
� ��3�� r r r r r c � � | d S )Nr< r r s r r z/test_numba_vs_python_indexing.<locals>.<lambda>5 rA r r )r r r r �assert_series_equal)�frame�row_funcr r �col_funcs r � test_numba_vs_python_indexingrG + s� � � ��i�o�>��O�$�
�E� �H�
�[�[��'��[�
:�F��{�{�8�H�1�{�=�H����6�8�,� �H�
�[�[��'��[�
:�F��{�{�8�H�1�{�=�H����6�8�,r � reductionc �" � | j � S r )�meanr s r r r = s � �q�v�v�x� r c �" � | j � S r )�minr s r r r = s � �1�5�5�7� r c �" � | j � S r )�maxr s r r r = s � �a�e�e�g� r c �" � | j � S r )�sumr s r r r = s � ����� r c �� � t t j dt j �� � }|j | d|�� }|j | d|�� }t j ||� y )N)r7 r7 r& r r r )r �np�ones�float64r r rC )rH r
r0 r r s r � test_numba_vs_python_reductionsrU ; sU � �
�2�7�7�6����4� 5�B�
�X�X�i��j�X�
A�F��x�x� �(��x�D�H����6�8�,r �colnamesr3 )g �?g @g @c � �� t t j g d�g d�g d�gt j �� | �� }| d ��fd�}|j |dd �
� }|j |d d �
� }t j ||� y ) Nr3 r6 )� � � r& )r* r c � �� | � S r r )r � first_cols �r r z-test_numba_numeric_colnames.<locals>.<lambda>M s �� �!�I�,� r r r r r )r rR �array�int64r r rC )rV r0 �fr r r\ s @r �test_numba_numeric_colnamesr` F sr �� �
�
���)�Y� �2�"�(�(�C�X�
�B� �� �I� �A�
�X�X�a��a�X�
0�F��x�x��(��x�3�H����6�8�,r c � � d� }t j t d�� 5 | j |dddi�� d d d � y # 1 sw Y y xY w)Nc � � | S r r r s r r z1test_numba_parallel_unsupported.<locals>.<lambda>T � � �!� r zAParallel apply is not supported when raw=False and engine='numba'��matchr �parallelT)r �
engine_kwargs)r. �raises�NotImplementedErrorr )r r_ s r � test_numba_parallel_unsupportedrj S sN � ��A� � � ��Q�
� O� ���!�G�J��;M��N� O� O� O�s �?�Ac �� � d� }t dddgit ddg� �� }t j t d�� 5 |j |d| � � d d d � y # 1 sw Y y xY w)
Nc � � | S r r r s r r z2test_numba_nonunique_unsupported.<locals>.<lambda>] rc r r$ r r4 r? zBThe index/columns must be unique when raw=False and engine='numba'rd r r )r r r. rh ri r �r
r_ r0 s r � test_numba_nonunique_unsupportedrn \ s_ � ��A� �C�!�Q��=��s�C�j�(9� :�B� � � ��R�
� 5� ����7���4� 5� 5� 5�s �A�A"c � � d� }t ddgddgddgd�� }|d j d
� |d < t j t d � � 5 |j |d
| �� d d d � t j t d� � 5 |d j
� j |d
| �� d d d � y # 1 sw Y �SxY w# 1 sw Y y xY w)Nc � � | S r r r s r r z/test_numba_unsupported_dtypes.<locals>.<lambda>g rc r r r4 r$ r% r7 r8 r: r; zdouble[pyarrow]zAColumn b must have a numeric dtype. Found 'object|string' insteadrd r r zUColumn c is backed by an extension array, which is not supported by the numba engine.)r �astyper. rh �
ValueErrorr �to_framerm s r � test_numba_unsupported_dtypesrt f s� � ��A� �!�Q��s�C�j��1�v�>� ?�B���g�n�n� .�/�B�s�G� � � ��Q�
� 5� ����7���4� 5�
� � ��6�
� E�
�3����� � ��7�� �D� E� E�
5� 5�� E� E�s � B2�&B>�2B;�>C) �numpyrR r. � pandas.util._test_decorators�util�_test_decorators�td�pandasr r �pandas._testing�_testingr �
skip_if_no�mark�
single_cpu�
pytestmark�fixturer
r! r1 rG � parametrizerU r` rj rn rt r r r �<module>r� s� �� �
� )� )�� ��b�m�m�G�$�f�k�k�&<�&<�
=�
� �� � ��1�v� �� ��,�
�
-� �������*�,=�?P�Q��-� �-� ������i� �%A�B� -� C� -�O�5�Er