403Webshell
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/reshape/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

Current File : C:/Python312/Lib/site-packages/pandas/tests/reshape/__pycache__/test_melt.cpython-312.pyc
�

0	�g���"�ddlZddlZddlZddlZddlmZmZmZm	Z	m
Z
mZddlm
Zejd��Zejd��Zejd��Zejd��ZGd�d�ZGd	�d
�ZGd�d�Zd
�Zy)�N)�	DataFrame�Index�
date_range�lreshape�melt�wide_to_longc	�b�ttjjd�j	d�tt
d�t��tddd���	�}|d
dkDjtj�|d<|ddkDjtj�|d
<|S)N�)�
��ABCD��dtypez
2000-01-01r�B)�periods�freq��columns�index�Ar�id1�id2)r�np�random�default_rng�standard_normalr�list�objectr�astype�int64��ress �@C:\Python312\Lib\site-packages\pandas/tests/reshape/test_melt.py�dfr$s���
�
�	�	���a� �0�0��9��d�6�l�&�1���r��<��C�
�c�(�Q�,�&�&�r�x�x�0�C��J��c�(�Q�,�&�&�r�x�x�0�C��J��J�c��tgd�gd�gd�g�}td�td�g|_ddg|j_|S)N)�5��:�?�S���t��gK�Բ��?)�U��y$����>�M��?g�K���)�i�����<�8b->�?ghB�Ē���ABC�abc�CAP�low)rrr�namesr!s r#�df1r2sE��
�*�,�+�	
��C���;��U��,�C�K����C�K�K���Jr%c��y)N�var�r5r%r#�var_namer6,���r%c��y)N�valr5r5r%r#�
value_namer:1r7r%c�X�eZdZd�Zd�Zd�Zd�Zejjde
eejf�d��Zd�Zejjdd	gd
gdedd
dd�d
d
d
d�dddd�d��fdgdgdedd
dd�dddd�dddd�d��fg�d��Zejjdddggdgdgddgg�d��Zd�Zd�Zd�Zejjd dd!g�d"��Zd#�Zejjd$ej2ed%d&d'�(��ej2gd)�d*�+�ej2gd,��g�d-��Zd.�Zd/�Zd0�Zd1�Zd2�Z d3�Z!d4�Z"d5�Z#ejjd6d7d8g�d9��Z$d:�Z%d;�Z&d<�Z'd=�Z(d>�Z)y?)@�TestMeltc�\�t|�}|jj�ddgk(sJ�y)N�variable�value�rr�tolist)�selfr$�results   r#�test_top_level_methodzTestMelt.test_top_level_method7s+���b����~�~�$�$�&�:�w�*?�?�?�?r%c	��tj|j�t|��tj|jddgddg��t|ddgddg���tj|j||��t|||���tj|jd��t|d���y)	Nrrrr��id_vars�
value_vars�r6r:r��	col_level)�tm�assert_frame_equalr)rBr$r2r6r:s     r#�test_method_signatureszTestMelt.test_method_signatures;s���
���b�g�g�i��b��2�
����G�G�U�E�N��S�z�G�B���e�U�^��c�
�C�	
�
	����G�G�X�*�G�=���h�:�>�	
�
	���c�h�h��h�3�T�#��5K�Lr%c�8�|j�}|jj�ddgk(sJ�|jdg��}|jj�gd�k(sJ�|jddg��}|jj�gd�k(sJ�y)Nr>r?r)rG)rr>r?r�rrr>r?r@)rBr$rC�result1�result2s     r#�test_default_col_nameszTestMelt.test_default_col_namesJs���������~�~�$�$�&�:�w�*?�?�?�?��'�'�5�'�'�*�����%�%�'�+G�G�G�G��'�'�5�%�.�'�1�����%�%�'�+N�N�N�Nr%c��|jddgd��}t|�dk(sJ�|jddgddg��}t|dj�dz|dj�dzdgdzdgdzz|dj�|dj�zd�gd��	�}t	j
||�y)
NrrrrFrrr
rP�r)r�lenrrArLrM)rBr$�result3�result4�	expected4s     r#�test_value_varszTestMelt.test_value_varsTs����'�'�5�%�.�S�'�A���7�|�r�!�!�!��'�'�5�%�.�c�3�Z�'�H����%�y�'�'�)�A�-��%�y�'�'�)�A�-� �E�B�J�#����3��S�'�.�.�*�R��W�^�^�-=�=�	
�8�
�	�	���g�y�1r%�type_c�>�t|dj�dz|dj�dzdgdzdgdzz|dj�|dj�zd�gd���}|jddg|d	��
�}tj||�y)Nrr
rrrrrPrU�rrrF)rrArrLrM)rBr[r$�expectedrCs     r#�test_value_vars_typeszTestMelt.test_value_vars_typesds�����%�y�'�'�)�A�-��%�y�'�'�)�A�-� �E�B�J�#����3��S�'�.�.�*�R��W�^�^�-=�=�	
�8�
�����%���E�*�<M��N��
���f�h�/r%c���t|ddgt|�zdgt|�z|dd�gd���}|jdgdg��}tj||�y)N�r�ar�b�rrc)rar/r0r?rUrF)rrVrrLrM)rBr2r^rCs    r#�test_vars_work_with_multiindexz'TestMelt.test_vars_work_with_multiindexssh����
�O��u�s�3�x�'��u�s�3�x�'��Z��	
�8�
�����:�,�J�<��H��
���f�h�/r%z(id_vars, value_vars, col_level, expectedrrrr'r)r+�r�r
r(r*r,)rr/r?rbrcrg)rbr0r?c�X�|j|||��}tj||�y)NrJ)rrLrM)rBrGrHrKr^r2rCs       r#�%test_single_vars_work_with_multiindexz.TestMelt.test_single_vars_work_with_multiindex�s)��@���'�:���C��
���f�h�/r%zid_vars, value_varsrardc��d}tjt|��5|j||��ddd�y#1swYyxYw)NzF(id|value)_vars must be a list of tuples when columns are a MultiIndex��matchrF)�pytest�raises�
ValueErrorr)rBrGrHr2�msgs     r#�$test_tuple_vars_fail_with_multiindexz-TestMelt.test_tuple_vars_fail_with_multiindex�s=��X��
�]�]�:�S�
1�	=��H�H�W��H�<�	=�	=�	=�s	�;�Ac���|j|��}|jj�ddgk(sJ�|jdg|��}|jj�gd�k(sJ�|jddg|��}|jj�gd�k(sJ�|jddgd	|�
�}|jj�gd�k(sJ�|jddgd	dg|�
�}td|dj�dzd|dj�dz|d	gd
zdgd
zzd|d	j�|dj�zidd|dg��}t	j
||�y)N)r6r4r?r�rGr6)rr4r?r)rrr4r?r)rGrHr6rr
rrU�rrrArrLrM)	rBr$r6�result5�result6�result7�result8�result9�	expected9s	         r#�test_custom_var_namezTestMelt.test_custom_var_name�s���'�'�8�'�,�����%�%�'�E�7�+;�;�;�;��'�'�5�'�H�'�=�����%�%�'�+B�B�B�B��'�'�5�%�.�8�'�D�����%�%�'�+I�I�I�I��'�'�5�%�.�S�8�'�T�����%�%�'�+I�I�I�I��'�'��E�N��S�z�H��
����r�%�y�'�'�)�A�-��r�%�y�'�'�)�A�-��3�%�"�*��u�r�z�1��"�S�'�.�.�*�R��W�^�^�-=�=�	
��E�8�W�5�
�	�	���g�y�1r%c���|j|��}|jj�ddgk(sJ�|jdg|��}|jj�gd�k(sJ�|jddg|��}|jj�gd�k(sJ�|jddgd	|�
�}|jj�gd�k(sJ�|jddgd	dg|�
�}td|dj�dzd|dj�dzdd	gd
zdgd
zz||d	j�|dj�ziddd|g��}t	j
||�y)N)r:r>r9r�rGr:)rr>r9r)rrr>r9r)rGrHr:rr
rrUrt)	rBr$r:�result10�result11�result12�result13�result14�
expected14s	         r#�test_custom_value_namezTestMelt.test_custom_value_name�s����7�7�j�7�1�����&�&�(�Z��,?�?�?�?��7�7�E�7�z�7�B�����&�&�(�,F�F�F�F��7�7�E�5�>�j�7�I�����&�&�(�,M�M�M�M��7�7��E�N�s�z��
�����&�&�(�,M�M�M�M��7�7��E�N��S�z�j��
����r�%�y�'�'�)�A�-��r�%�y�'�'�)�A�-��S�E�B�J�#����3��R��W�^�^�-��3����0@�@�	
��E�:�z�:�
�
�	���h�
�3r%c��|j||��}|jj�ddgk(sJ�|jdg||��}|jj�gd�k(sJ�|jddg||��}|jj�gd�k(sJ�|jddgd	||�
�}|jj�gd�k(sJ�|jddgd	dg||�
�}td|dj�dzd|dj�dz|d	gd
zdgd
zz||d	j�|dj�zidd||g��}	t	j
||	�|j
�}
d|
j_|
j�}|jj�ddgk(sJ�y)NrIr4r9r�rGr6r:)rr4r9r)rrr4r9r�rGrHr6r:rr
rrU�foor?)rrrArrLrM�copy�name)rBr$r:r6�result15�result16�result17�result18�result19�
expected19�df20�result20s            r#�test_custom_var_and_value_namez'TestMelt.test_custom_var_and_value_name�s����7�7�H��7�D�����&�&�(�U�E�N�:�:�:��7�7�E�7�X�*�7�U�����&�&�(�,A�A�A�A��7�7��E�N�X�*��
�����&�&�(�,H�H�H�H��7�7��E�N���!�	�
�����&�&�(�,H�H�H�H��7�7��E�N��S�z��!�	�
����r�%�y�'�'�)�A�-��r�%�y�'�'�)�A�-��3�%�"�*��u�r�z�1��R��W�^�^�-��3����0@�@�	
��E�8�Z�8�
�
�	���h�
�3��w�w�y��!������9�9�;�����&�&�(�U�G�,<�<�<�<r%rKr/c�j�|j|��}|jj�ddgk(sJ�y)NrJr/r?r@)rBrKr2r"s    r#�test_col_levelzTestMelt.test_col_levels2���h�h��h�+���{�{�!�!�#��w�'7�7�7�7r%c�f�|j�}|jj�gd�k(sJ�y)N)r/r0r?r@)rBr2r"s   r#�test_multiindexzTestMelt.test_multiindexs)���h�h�j���{�{�!�!�#�'>�>�>�>r%�col�2010�z
US/Pacific)r�tz)rbrc�crb�d�categoryr)rrgrrrc��ttd�|gd�|d��}tjtjgd��|gd��}t|ddgdd	�
�}tt
td��dztj|gdzd��dgdzd
gdzz|d��}gd�|_tj||�y)Nr�)rgrrrr)�klassr��attr1�attr2T��ignore_indexr�r��	attributer?r�r
r�r��rrgr
�)r�r�r�r?)
r�range�pd�concat�SeriesrrrrLrM)rBr�r$�expected_valuerCr^s      r#�test_pandas_dtypeszTestMelt.test_pandas_dtypess�����A�h�s�_�s�S�
�����B�I�I�o�$>��#D�SW�X�����%�(�;�7�
�����a��>�A�%��9�9�c�U�Q�Y�T�:��9�q�=�G�9�q�=�0�!�	
�
��B���
���f�h�/r%c���tddgtjddg�d��}t|dgdg�}ttjddg�ddgddgd��}t	j
||�y)	Nrgr
�X�Yr]rr)rr>r?)rr��CategoricalrrLrM)rB�datarCr^s    r#�test_preserve_categoryzTestMelt.test_preserve_category;sq����1�v�B�N�N�C��:�,F�G�H���d�S�E�C�5�)����.�.�#�s��,�3��*�PQ�ST�v�V�
��	���f�h�/r%c��ttjjd�j	d�td���}d}t
jt|��5|jddgd	d
g�ddd�t
jt|��5|jddgdd
g�ddd�t
jt|��5|jgd
�dd
g�ddd�|j�}td�td�g|_t
jt|��5|jdgdg�ddd�t
jt|��5|jdgdgd��ddd�y#1swY��xYw#1swY��xYw#1swY��xYw#1swY�ixYw#1swYyxYw)Nr
)r�r�abcdrUzEThe following id_vars or value_vars are not present in the DataFrame:rkrbrc�Cr�rr�)rbrc�not_here�or_therer
)�Erbrd�FrrJ)rrrrrrrmrn�KeyErrorrr�r)rBr$rp�multis    r#� test_melt_missing_columns_raisesz)TestMelt.test_melt_missing_columns_raisesEs�����I�I�!�!�!�$�4�4�V�<�d�6�l�
��
V��
�]�]�8�3�
/�	,��G�G�S�#�J��c�
�+�	,��]�]�8�3�
/�	,��G�G�S�#�J��c�
�+�	,��]�]���
�	D�
�G�G�6��c�
�C�		D����	���f��t�F�|�4��
�
�]�]�8�3�
/�	3��J�J�
�|�j�\�2�	3��]�]�8�3�
/�	2��J�J��u�s�e�q�J�1�	2�	2�)	,�	,��	,�	,��	D�	D��	3�	3��	2�	2�s<� F�F�F*�9F6�1G�F�F'�*F3�6F?�Gc���tdgdgdgdgd��}t|ddgdd	g�
�}tdgdzdgdztd�ddgd��}tj||�y)
Nr��barrgr
)rrbrcr�rrbrcr�rF�bd)rrbr>r?)rrrrLrM�rBr$rCr^s    r#�test_melt_mixed_int_str_id_varsz(TestMelt.test_melt_mixed_int_str_id_varshsr��
�E�7�%���s�!��E�
F���b�1�c�(��S�z�B�����!��5�'�A�+�4��:�QR�TU�PV�W�
��	���f�h�/r%c��tdgdgd��}t|ddg��}tddgddgd��}tj||�y)Nr�r�)rrbrrb)rH�r>r?�rrrLrMr�s    r#�"test_melt_mixed_int_str_value_varsz+TestMelt.test_melt_mixed_int_str_value_varsqsL��
�E�7�%��1�
2���b�a��X�.���1�c�(�e�U�^�L�M��
���f�h�/r%c��tdgdgd�dg��}t|d��}tdd	gddgd
�ddg��}tj||�y)Nrrg�r�r��first�rFr�r�r�r�r�r�s    r#�test_ignore_indexzTestMelt.test_ignore_indexxsX��
��s�A�3�/��y�
A���b�u�-������1�a�&�9�'�7�AS�
��	���f�h�/r%c�F�tjjddgddg��}tddgdd	gd
�|��}t	|d�
�}tjjddgdzddg��}tdgdzdgdzzgd�d�|��}tj||�y)N)r��second)r��third�baz�foobar�r1rrgr
r�r�r�Fr�r�r�r�r�)r��
MultiIndex�from_tuplesrrrLrM�rBrr$rC�expected_indexr^s      r#�test_ignore_multiindexzTestMelt.test_ignore_multiindex�s����
�
�)�)�
 �"4�5�e�X�=N�*�
����1�v�q�!�f�5�U�
C���b�u�-�����2�2�
 �"4�5��9�%��AR�3�
�����1���w��{�2�\�J� �
��
	���f�h�/r%c���tddgdd��}tddgdd	gd
�|��}t|d�
�}tddgdzdd��}tgd�gd�d�|��}tj||�y)Nr�r�r�r�)rr�rrgr
r�)�x�yr�Fr�)r�r�r�r�r�r�)rrrrLrMr�s      r#�test_ignore_index_name_and_typez(TestMelt.test_ignore_index_name_and_type�sw���u�e�n�J�U�C��
�a��V�1�a�&�1��
?���b�u�-����u�~��1��%�P���-��E� �
��
	���f�h�/r%c��tgd�ggd���}|jdgdg��}tgd�gd�ggd	���}tj||�y)
N)�idr
r�)rbrcrcrUrbrcrF)r�rcr
)r�rcr�)rbr>r?r�r�s    r#� test_melt_with_duplicate_columnsz)TestMelt.test_melt_with_duplicate_columns�sM��
��~��
?�����#��C�5��9���
�^�,�6P�
��	���f�h�/r%r�Int8�Int64c��ttjddgd��tjddg|��d��}|j�}tgd�tjgd	�|��d
��}t	j
||�y)Nrgr
r�rr�r�rbrc)rbrbrcrc)rgr
r�rr�)rr�r�rrLrM)rBrr$rCr^s     r#�test_melt_ea_dtypezTestMelt.test_melt_ea_dtype�sv����Y�Y��1�v�V�4��Y�Y��1�v�U�3�
�
��������0����<�u�=�
�
��	���f�h�/r%c�6�tdddd�dddd�dd	d
d�d��}|jjd�|_|jd
gdg��}tt	d�tjdgdzd��gd�d��}tj||�y)Nrbrcr�rfrgr�r�r
r�)rrr�zstring[python]rrrFr.r)rgr�r�)rr>r?)	rrrrrr�r�rLrMr�s    r#�test_melt_ea_columnszTestMelt.test_melt_ea_columns�s���
����-��q�Q�'��q�Q�'�
�
���Z�Z�&�&�'7�8��
����#��C�5��9����%�[��I�I�s�e�a�i�7G�H�"�
�
��	���f�h�/r%c��tdtjdd��tjdd��d�dtjdd��tjdd��d�gd	d
g��}|jdgd
dgdd��}tddddd�d
d
ddd�tjdd��tjdd��tjdd��tjdd��d�d��}t	j
||�y)N�A0z
2023/03/01z
Asia/Tokyo)r�z
2023/03/10)�type�
start_date�end_date�A1z
2023/03/11�aaaa�bbbb�r�rr�r�r��	start/end�dater�r�z2023-03-01 00:00:00+0900z2023-03-10 00:00:00+0900z2023-03-11 00:00:00+0900)r�r�r�)rr��	TimestamprrLrMr�s    r#�test_melt_preserves_datetimez%TestMelt.test_melt_preserves_datetime�s��
�!�"$�,�,�|��"M� "���\�l� K��!�"$�,�,�|��"M� "���\�l� K��
��6�"�
������H�$�j�1� ��	�
��� �T�d�t�<�#�#�!�!�	��|�|�$>�<�P��|�|�$>�<�P��|�|�$>�<�P��|�|�$>�<�P�	�
�
��"	���f�h�/r%c��tgd�gd�d�gd���}|jddd�	�}tgd�d
gdzgd�d��}tj||�y�
N�rgr
r��rr�r�r���11�22�33r�rbrrgr�rcr�)rbrrgr�r�s    r#�#test_melt_allows_non_scalar_id_varsz,TestMelt.test_melt_allows_non_scalar_id_vars�s^��
� �y�1�$�
���������
��
�9�#����y�I�J��
���f�h�/r%c��tgd�gd�d�gd���}|jdgdd�	�}tgd�d
gdzgd�d��}tj||�yr�r�r�s    r#�$test_melt_allows_non_string_var_namez-TestMelt.test_melt_allows_non_string_var_names`��
� �y�1�$�
������E����
��
�9�#����y�I�J��
���f�h�/r%c��tgd�gd�d�gd���}tjtd��5|j	dgd	d
g��ddd�y#1swYyxYw)Nr�r�r�r�r�z.* must be a scalar.rkrbrgr
rs)rrmrnror�rBr$s  r#�$test_melt_non_scalar_var_name_raisesz-TestMelt.test_melt_non_scalar_var_name_raisessS��
� �y�1�$�
���]�]�:�-D�
E�	4��G�G�S�E�Q��F�G�3�	4�	4�	4�s�A�AN)*�__name__�
__module__�__qualname__rDrNrSrZrm�mark�parametrize�tuplerr�arrayr_rerrirqr{r�r�r�r�r�r�rr�r�r�r�r�r�r�r�r�r�r�r�r�r�rr5r%r#r<r<6sJ��@�
M�O�2� �[�[���W�u�d�B�H�H�&=�>�0�?�0�0��[�[���2�������!)�i�I�F�#&�3�3�7�%.�8��!H���	
�������!)�i�I�F�#&�3�3�7�%.�8��!H���	
�	
��:0�;�:0��[�[����
�*��&��\�:�&�
��$�	
��=��=�2�64�:(=�T�[�[���[�1�e�*�5�8�6�8�?��[�[���
��B�I�I�j���|�D�E��B�I�I�/�z�B��B�I�I�o�&�	
��0��0�(0�!2�F0�0�0�0�$0�0��[�[���W�v�w�&7�8�0�9�0�"0�('0�R0�0�4r%r<c��eZdZd�Zy)�TestLreshapec���gd�gd�gd�gd�gd�dtjddd	gd
tjtjddggd
�dtjdddgdtjtjddgd�
}t|�}tdd�D�cgc]}d|d����
c}tdd�D�cgc]}d|d����
c}d�}t	||�}gd�gd�gd�gd�gd �gd!�d"�}t||j
�#�}t
j||�t	||d$�%�}gd&�gd'�gd(�gd)�d*d+d,d-d.dtjddd	d
tjtjddgd/dd0d1d2dtjddddtjtjddgd"�}t||j
�#�}t
j||�tdd3�D�cgc]}d|d����
c}tdd�D�cgc]}d|d����
c}d�}d4}tjt|�5�5t	||�ddd�ycc}wcc}wcc}wcc}w#1swYyxYw)6N)�	08jan2009�	20dec2008�	30dec2008�	21dec2008�	11jan2009)������C�%)�e�f�g�h�i)�Male�Femalerrr)r�	22dec2008�	04jan2009�	29dec2008�	20jan2009�	21jan2009�	22jan2009�	31dec2008�	03feb2009�	05feb2009�	02jan2009�	15feb2009)ii

i
i�i��l�@���@��@�߱@��@�b�@�Ų@)
�birthdt�birthwtr��sex�visitdt1�visitdt2�visitdt3�wt1�wt2�wt3rgr�visitdtr��wt)r7r8)rrr
rrrr
rrrrr)rrrrrrrrrrrr)rrrrrrrrrrrr)rrrrrrrrrrrr)rrrrrr r!r"r#r$r%r&)�|�@r)�4�@�ĩ@�Ұ@r'r(r)r*r+r,r-)r.r/r�r0r7r8rUF)�dropna)rrr
rrrrr
rrrrr
rr)rrrrrrrrrrrrrrr)rrrrrrrrrrrrrrr)rrrrrrrrrrrrrrrrrrrrr9r:r;r<r�z$All column lists must be same lengthrk)r�nanrr�rrrLrMrmrnro)	rBr�r$�i�specrC�exp_data�exprps	         r#�
test_pairszTestLreshape.test_pairssr���6�+�C��%�b�f�f�k�;��T�$�b�f�f�b�f�f�k�;�O�1��B�F�F�F�F�F�;��B�F�F�B�F�F�F�F�;�-
��2�t�_��27�q�!��=�A�'�!�A���=�',�Q��{�3�!�R��!�u�:�3�
���"�d�#��
�
�O�
�
�
�uH
��R��&�.�.�9��
���f�c�*��"�d�5�1���"�"�"�$����������������������$����������������������mg
��P��&�.�.�9��
���f�c�*�27�q�!��=�A�'�!�A���=�',�Q��{�3�!�R��!�u�:�3�
��5��
�]�]�:�S�
1�	��R���	�	��E>��3��z>��3�	�	�s$�
I�8
I�$
I�
I�0
I�I#N)rrrrCr5r%r#r	r	s��`r%r	c��eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
�Zd�Zd�Zd�Zd�Zd�Zej,j/dddg�d��Zy)�TestWideToLongc
���tjjd�jd�}t	dddd�ddd	d�d
ddd�d
ddd�tt
td�|��d��}|j|d<|j�|j�zgd�gd�gd�gd�d�}t	|�}|jddg�gd�}t|ddgdd��}tj||�y)Nr
r�rbrcr�rfr��e�f�@�333333�?�ffffff�?皙����	@��������?皙�����?)�A1970�A1980�B1970�B1980r�r��rbrcr�r�rGrH�rIrJrKrLrMrN��rVrV�rWrW�rrgr
rrgr
�r�rr�yearr�rZ�r�rrrr�r?�j�rrrrr�dict�zipr�rrA�	set_indexrrLrM�rBr�r$rAr^rCs      r#�test_simplezTestWideToLong.test_simples����I�I�!�!�!�$�4�4�Q�7��
� �S�S�1� �S�S�1� �S�S�1� �S�S�1��#�e�A�h��*�+�
�
���8�8��4������a�h�h�j�(�/�/�8�$�
���X�&���%�%�t�V�n�5�o�F���b�3��*���?��
���f�h�/r%c�p�tgd�gd�g�}gd�|_ddg}t||dd��|ddgk(sJ�y)	N)rrgr
r��)rr�r���	)r��inc1�inc2�edu1�edu2�inc�edur��ager\)rrr)rBr$�stubss   r#�
test_stubszTestWideToLong.test_stubssA��
���9�
:��;��
������R��$�%�0�����&�&�&r%c
���tjjd�jd�}t	dddd�ddd	d�d
ddd�d
ddd�tt
td�|��d��}|j|d<|j�|j�zgd�gd�gd�gd�d�}t	|�}|jddg�gd�}t|ddgddd��}tj||�y)Nr
r�rbrcr�rfr�rGrHrIrJrKrLrMrN)zA.1970zA.1980zB.1970zB.1980r�r�rSrTrUrXrYrZr[rr�.�r?r]�sepr^rbs      r#�test_separating_characterz(TestWideToLong.test_separating_character#s���
�I�I�!�!�!�$�4�4�Q�7��
�!�c�c�2�!�c�c�2�!�c�c�2�!�c�c�2��#�e�A�h��*�+�
�
���8�8��4������a�h�h�j�(�/�/�8�$�
���X�&���%�%�t�V�n�5�o�F���b�3��*���C�H��
���f�h�/r%c
���tjjd�jd�}t	dddd�ddd	d�d
ddd�d
ddd�tt
td�|��d��}|j|d<|j�|j�zgd�gd�gd�gd�d�}t	|�}|jddg�gd�}t|ddgdd��}tj||�y)Nr
r�rbrcr�rfr�rGrHrIrJrKrLrMrN)zA(quarterly)1970zA(quarterly)1980zB(quarterly)1970zB(quarterly)1980r�r�rSrTrUrX)r��A(quarterly)�B(quarterly)rZr�rZ)r�rwrxrwrxr\r^rbs      r#�test_escapable_charactersz(TestWideToLong.test_escapable_characters=s����I�I�!�!�!�$�4�4�Q�7��
�(+���$<�(+���$<�(+���$<�(+���$<��#�e�A�h��*�+�
�
���8�8��4������a�h�h�j�(�:�:�8�$�
���X�&���%�%�t�V�n�5�1�
���b�>�>�"B�d�f�U��
���f�h�/r%c�F�tddgddgddgddgd	��}|j|d
<gd�gd�ddtjtjggd
�gd�d�}t|�}|j	d
dg�gd�}t|ddgd
d��}t
j||�y)N��?�@�@�@�@�@�X1�X2��A2010�A2011�B2010r�r��r�r�r�r��r{r|r}r~)rrgrrg)��r���r�)r�rrr�rZrZr[rrr\)rrrr>rarrLrM�rBr$rAr^rCs     r#�test_unbalancedzTestWideToLong.test_unbalancedWs���
��s���s���s���D�\�	
�
���8�8��4��)�%��s�B�F�F�B�F�F�+��,�
���X�&���%�%�t�V�n�5�o�F���b�3��*���?��
���f�h�/r%c�j�tgd�gd�gd�gd�gd�gd�gd�gd�d��}|j|d	<tgd
�gd
�gd�gd�gd
�gd�gd�d��}|jd	dg�gd�}t|gd�d	d��}t	j
|j
d��|j
d���y)N��a11�a22�a33��a21r��a23��b11�b12�b13��b21�b22�b23r�r���[�\�])�A11�A12�B11�B12�BB11�BB12�BBBX�BBBZr��r�r�r�r�r�r��r�r�r�r�r�r��r�r�r�r�r�r��rgr
r�rr�r�rX��r�r��r�r�)r�r�rr�BBr�rZrZ)r�r�rrr��rrr�r\rg��axis�rrrarrLrM�
sort_index�rBr$r^rCs    r#�test_character_overlapz%TestWideToLong.test_character_overlapns���
�,�,�,�,�!�!�$�$�	
�
���8�8��4���0�0�?�?�(�(�0�
�

���%�%�t�V�n�5�6V�W���b�"2�d�f�E��
���f�/�/�Q�/�7��9L�9L�RS�9L�9T�Ur%c	���d}tddgddgddgdd	gd
��}|j|d<dgggggggd
�}t|�jdtji�}|jddg�gd�}|jj
ddgd��|_t|ddgdd|��}tj|jd��|jd���y)Nznope!r{r|r}r~rr�r�r�r�r��)r�r�r�r�r�rZrrrZ)r�r�r�r�rrrrg��levelrrrsr��rrrrr ra�
set_levelsrrLrMr�)rBrtr$rAr^rCs      r#�test_invalid_separatorz%TestWideToLong.test_invalid_separator�s����
��s���s���s���D�\�	
�
���8�8��4����������	
���X�&�-�-�v�r�x�x�.@�A���%�%�t�V�n�5�6�
��"���2�2�A�q�6��2�C����b�3��*���C�H��
���f�/�/�Q�/�7��9L�9L�RS�9L�9T�Ur%c�j�tgd�gd�gd�gd�gd�gd�gd�gd�d��}|j|d	<tgd
�gd
�gd�gd�gd
�gd�gd�d��}|jd	dg�gd�}t|gd�d	d��}t	j
|j
d��|j
d���y)Nr�r�r�r�r�r�r�)r�r�r�r�r�r��Arating�Arating_oldr�r�r�r�r�rXr�)r�r�rrr�r�rZrZ)r�r�rrr�r�r\rgr�r�r�s    r#�test_num_string_disambiguationz-TestWideToLong.test_num_string_disambiguation�s����,�,�,�,�!�!�'�+�	
�
���8�8��4���3�7�?�?�(�(�0�
�

���%�%�t�V�n�5�6�
���b�"2�d�f�E��
���f�/�/�Q�/�7��9L�9L�RS�9L�9T�Ur%c	��tddgddgddgddgd	��}|j|d
<dgggggggd�}t|�jd
tji�}|jd
d
g�}|jj
ddgd��|_t|ddgd
d
��}tj|jd��|jd���y)Nr{r|r}r~rr�r�r�)�Aone�Atwo�Boner�r�r�)r�r�r�r�r�rZrrrZrrgr�rrr\r�r�r�s     r#�test_invalid_suffixtypez&TestWideToLong.test_invalid_suffixtype�s�����c�
��c�
��c�
��D�\�	
�
���8�8��4����������	
���X�&�-�-�v�r�x�x�.@�A���%�%�t�V�n�5��!���2�2�A�q�6��2�C����b�3��*���?��
���f�/�/�Q�/�7��9L�9L�RS�9L�9T�Ur%c���tgd�gd�gd�gd�d��}tgd�gd�gd�gd	�d
��}|jgd��dg}t|dd
dgd��}tj||�y)N)	rgrgrgr
r
r
r�r�r�)	rgr
r�rgr
r�rgr
r�)	�ffffff@�333333@皙����@r
�������?�ffffff�?r��ffffff@�������@)	�333333@�ffffff@r�rLr��333333@�ffffff
@r�r�)�famid�birth�ht1�ht2)r�r�r�r�r�r�r|rLr�r�r�r�r�r�r�r�r�r�)rgrgrgrgrgrgr
r
r
r
r
r
r�r�r�r�r�r�)rgrgr
r
r�r�rgrgr
r
r�r�rgrgr
r
r�r�)rgr
rgr
rgr
rgr
rgr
rgr
rgr
rgr
rgr
)�htr�r�rn)r�r�rnr�r�r�rnr\�rrarrLrMr�s    r#�test_multiple_id_columnsz'TestWideToLong.test_multiple_id_columns�s~��
�4�4�B�D�	
�
����(P�O�M�/
�
��6�%�%�&?�@�$��H���b�$�7�G�*<��F��
���f�h�/r%c��tgd�gd�gd�d��}d}tjt|��5t	|ddgdd	�
�ddd�y#1swYyxYw)N)rgr
r�rr�)rgrgrgrgrg)�A_A1�B_B1r�z3the id variables need to uniquely identify each rowrk�A_A�B_Br��colnamer\�rrmrnror�rBr$rps   r#�test_non_unique_idvarsz%TestWideToLong.test_non_unique_idvarssY���$�o�O�T�
��D��
�]�]�:�S�
1�	A���e�U�^�s�i�@�	A�	A�	A�s�A�Ac���tgd�gd�gd�gd�gd�d��}tgd�gd�gd	�gd
�d��jdd
g�}t|ddgdd
d��}tj||�y)N)�CCH Pounder�Johnny Depp�Christoph Waltz)�Joel David Moore�
Orlando Bloom�Rory Kinnear)�@�@���@�|�@)�@�@���@��x@)�Avatar�Pirates of the Caribbean�Spectre)�actor_1�actor_2�actor_fb_likes_1�actor_fb_likes_2�title)r�r�r�r�r�r�)r�r�r�r�r�r�)rgrgrgr
r
r
)r�r�r�r�r�r�)�actor�actor_fb_likes�numr�r�r�r�r��_rsr�r�s    r#�test_cast_j_intzTestWideToLong.test_cast_j_ints���
�L�P�$>�$:�J�
�
����#S�)��
�
�*�)�W�e�$�
%�+	�,���*�+�w�%�S�
��	���f�h�/r%c��tddgddgddgddgd	��}d
}tjt|��5t	|dd
gdd��ddd�y#1swYyxYw)Nr{r|r}r~rr�r�r�)r�r�r�rz,stubname can't be identical to a column namerkrrr�r\r�r�s   r#�test_identical_stubnamesz'TestWideToLong.test_identical_stubnamesBsi��
��s���s���s���D�\�	
�
��=��
�]�]�:�S�
1�	=���c�3�Z�3�)�<�	=�	=�	=�s�A�Ac��tddgddgddgddgd	��}tgd
�gd�ddtjtjggd�d
��}|jddg�}t	|ddgdddd��}tj||�y)Nr{r|r}r~rr�r�r�)�treatment_placebo�treatment_test�result_placeborr�)�placebor�testrr��rr�rC�	treatmentrr�rCrz[a-z]+r��r?r]�suffixrt�rrr>rarrLrMr�s    r#�test_nonnumeric_suffixz%TestWideToLong.test_nonnumeric_suffixOs���
�&)�3�Z�#&��*�#&��*��D�\�	
�
���-�A���R�V�V�R�V�V�4�1�	
�
���%�%�s�I�&6�7�����;�'�3�)�H�RU�
��	���f�h�/r%c���tddgddgddgddgd	d
gd��}tgd�gd
�gd�gd�d��jddg�}t|ddgdddd��}tj||�y)Nr�r�rrgrr�r{r|r}r~)r�result_1�
result_foo�treatment_1�
treatment_foor�)�1rr�r�)��"@rr�r�rrr�rCrz.+r�rr�r�s    r#�test_mixed_type_suffixz%TestWideToLong.test_mixed_type_suffixfs���
��D�\���F�"�C�j� #�S�z�"%�s��
�
���-�3�.�1�	
�
��)�S�)�$�
%�	����;�'�3�)�D�c�
��	���f�h�/r%c
���tddgddgddgddgd	d
gd��}tgd�gd
�ddddtjtjtjtjgtjtjtjtjddddgd��}|jddg�}t	|ddgdddd��}tj||�y)Nr{r|r}r~rr�rrgr�r�)z
treatment_1.1z
treatment_2.1z
result_1.2r
r)r�r�r�r�r�r�r�r�)rJrJr{r{皙�����?rr�r�rrrrr�rCrz[0-9.]+r�rr
r�s    r#�test_float_suffixz TestWideToLong.test_float_suffix}s���
�"%�s��"%�s��"�C�j���F��D�\�
�
���E�C���S�#�r�v�v�r�v�v�r�v�v�r�v�v�N� �f�f�b�f�f�b�f�f�b�f�f�c�3��S�Q�	
�
���%�%�s�I�&6�7�����;�'�3�)�I�SV�
��	���f�h�/r%c
���dddddd�ddd	d
dd�dd
d
ddd�dddddd�dddddd�d�}tj|�}t|dgddgd��}t|dddgd��}tj||�y)Nrrgr
r�r)rrgr
r�rg�������?rg�?r{g�Q����?g�G�z��?g�Q����?g\��(\�?g333333�?g�p=
ף�?g{�G�z�?g�p=
ף�?gq=
ףp�?g�(\���?rK)�node_idr�PA0�PA1�PA3�PArr�time)�	stubnamesr?r])r�	from_dictrrLrM)rB�	wide_data�wide_dfr^rCs     r#�test_col_substring_of_stubnamez-TestWideToLong.test_col_substring_of_stubname�s���
��q�Q�1�5��c�d�s�t�<����$�3�?����$�4�@����$�4�@�
�	��%�%�i�0����D�6�i��=M�QW�X���g��)�S�9I�V�T��
���f�h�/r%c���ttd�tddd�d��}tjt
t
jd���5|jdd�	�ddd�y#1swYyxYw)
Nr-r�r
)r�r?zvalue_name (value) cannot matchrkr?r})	rrr�rmrnro�re�escaperr�s  r#�test_raise_of_column_name_valuez.TestWideToLong.test_raise_of_column_name_value�sb���t�E�{�U�2�r�1�5E�F�
G��
�]�]��b�i�i�(I�J�
�	9�
�G�G�G��G�8�	9�	9�	9�s�A.�.A7r�O�stringc��tddgddgddgd��}|jd|i�}t|d	d
gddd�
�}tgd�d��}tgd�tj
gdzd�|��}|jjdj|�}|jj|d��|_tj||�y)Nr�2�d���,�)r�za-1za-2r�rbrcr��-)rr?r]rt))rrg)r,rg)rr
)r,r
)r�r�)r�)r-r.r/r0rr�r�rr�)rrrrrr>r�levelsr�rLrM)rBrr$rCrr^�	new_levels       r#�test_missing_stubnamez$TestWideToLong.test_missing_stubname�s����s�C�j�#�s��S�#�J�O�
P��
�Y�Y��e�}�
%�����C�j����
���4��
���&�b�f�f�X��\�:��
���N�N�)�)�!�,�3�3�E�:�	�!���2�2�9�A�2�F���
���f�h�/r%N)rrrrcrpruryr�r�r�r�r�r�r�r�r�rrrr#r(rmrrr4r5r%r#rErEs���0�0'�0�40�40�.V�<V�<V�BV�:'0�RA�%0�N=�0�.0�.0�00� 	9��[�[���W�s�H�o�6�0�7�0r%rEc��tjd�tddiddiddiddiddid��}|jj	d�|_t|ddd	d
d��}tddgddgddggt
d
dgt��tjjgd�t
gd�d��gdd	g����}tj||�y)N�pyarrowrrgr
)�ID�R_test1�R_test2�R_test3�Dzstring[pyarrow_numpy]�Rr7�	UNPIVOTEDr�z.*)rr?r]rtr	r;r)rgrgrg)�test1�test2�test3r�r)
rm�importorskiprrrrrrr�r��from_arraysrLrM)r$rCr^s   r#�(test_wide_to_long_pyarrow_string_columnsrC�s���
���	�"�	��a�&��1�v��1�v��1�v��Q��	
�
�B����"�"�#:�;�B�J�
�
�c�T�[�c�$��F��
�Q��!�Q��!�Q�� ��s�C�j��/��m�m�'�'���1�9P�Q�
���%�(�
�
�H����&�(�+r%)r&�numpyrrm�pandasr�rrrrrr�pandas._testing�_testingrL�fixturer$r2r6r:r<r	rErCr5r%r#�<module>rIs���	��
��������������
��
�������������c4�c4�La�a�HF0�F0�R,r%

Youez - 2016 - github.com/yon3zu
LinuXploit