
    bMhk                         d dl Z d dlZd dlZd dlZd dlmc mZ d dl	m
Z
 d dlZd dlmZmZmZmZmZmZmZmZmZmZ d dlmZ d dlmZ 	 d dlZn# e$ r dZY nw xY w G d d          ZdS )    N)is_integer_dtype)

ArrowDtypeCategoricalCategoricalDtypeCategoricalIndex	DataFrameIndex
RangeIndexSeriesSparseDtypeget_dummies)SparseArrayc                      e Zd Zej        d             Z ej        ddej        edg          d             Z	 ej        ddg          d	             Z
d
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zej                            dd e dd gi          i e d!d"gi          fd e d#dgi          i e d$d"gi          f e d#d gi          dd% e d!d"gi          f e d#d gi          dd& e d'd"gi          fg          d(             Z!d) Z"d* Z#d+ Z$d, Z%d- Z&d. Z'd/ Z(d0 Z)ej                            d1d"d2g          d3             Z*ej                            dd"d2g          d4             Z+d5 Z,d6 Z-ej                            d7d8g          d9             Z.d: Z/d; Z0ej                            d<d=d>g          d?             Z1 e2j3        d@          dA             Z4dS )BTestGetDummiesc                 4    t          g dg dg dd          S )Nabr   r   r   c         )ABC)r   )selfs    e/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/reshape/test_get_dummies.pydfzTestGetDummies.df"   s&    ___999UUVVV    uint8i8N)paramsc                 4    t          j        |j                  S N)npdtypeparamr   requests     r   r(   zTestGetDummies.dtype&   s    x&&&r!   densesparsec                     |j         dk    S )Nr-   )r)   r*   s     r   r-   zTestGetDummies.sparse*   s     }((r!   c                 "    |t           j        S |S r&   )r'   r"   )r   r(   s     r   effective_dtypezTestGetDummies.effective_dtype0   s    =8Or!   c                     d}t          j        t          |          5  t          |d           d d d            d S # 1 swxY w Y   d S )Nz1dtype=object is not a valid dtype for get_dummiesmatchobjectr(   )pytestraises
ValueErrorr   )r   r    msgs      r   'test_get_dummies_raises_on_dtype_objectz6TestGetDummies.test_get_dummies_raises_on_dtype_object5   s    A]:S111 	, 	,(++++	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	,s   =AAc                 v   t          d          }t          |          }t          |t          d                    }t          g dg dg dd|                     |                    }|rD|j        dk    r|                    t          d	
          }n|                    t          d
          }t          |||          }t          j	        ||           t          |||          }t          j	        ||           t          d          |_
        t          |||          }t          j	        ||           d S )NabcABCr   r   r   r   r   r   r   r   r   r   r   r   r5   r   F
fill_value        r-   r(   )listr   r   r0   kindapplyr   r   tmassert_frame_equalindex)r   r-   r(   s_lists_seriess_series_indexexpectedresults           r   test_get_dummies_basicz%TestGetDummies.test_get_dummies_basic:   s7   e&>>U44))))))))<<&&u--
 
 
  	GzS  #>>+%>HH#>>+#>FFVF%@@@
fh///XfEBBB
fh///e^F%HHH
fh/////r!   c                    t          d          }t          |          }t          g dg dg dd          }t          g dg dg dd|                     |          t          d          	          }|r>t	          |          rd
}n|t
          k    rd}nd}|                    t          |          }t          |||          }	t          j
        |	|           t          |||          }	t          j
        |	|           t          ||j        ||          }	|r"d|                     |          j         d| d}
n|                     |          j        }
t          |
did          }|	j                                        }	d |	j        D             |	_        t          j        |	|           t          |dg||          }	|rdnd}dd|di}d|                    |
d
          z   ||
<   t          |d                                          }|	j                                        }	d |	j        D             |	_        |	                                }	t          j        |	|           d S )Nr<   )r   r   r   r   r   )r   r   r   r   r   )r   r   r   r   r   rA   r>   r?   r@   )r(   columnsr   FrD   rB   rE   )rS   r-   r(   zSparse[z, ]   countnamec                 ,    g | ]}t          |          S  str.0is     r   
<listcomp>z?TestGetDummies.test_get_dummies_basic_types.<locals>.<listcomp>v       5551A555r!   r   r\   r4   int64r   r   c                 ,    g | ]}t          |          S rZ   r[   r]   s     r   r`   z?TestGetDummies.test_get_dummies_basic_types.<locals>.<listcomp>   ra   r!   )rF   r   r   r0   r   boolrH   r   r   rI   rJ   rS   rX   dtypesvalue_countsrK   assert_series_equalget
sort_index)r   r-   r(   using_infer_stringrL   rM   s_dfrO   rC   rP   
dtype_namekeyexpected_countss                r   test_get_dummies_basic_typesz+TestGetDummies.test_get_dummies_basic_typesR   s   e&>>!//(A(A(AXX
 
 ))))))))<<&&u--KK
 
 

  	J&& !

$"

 
~~kj~IIHVF%@@@
fh///XfEBBB
fh///T4<eTTT 	:T4#7#7#>#>#CTTzTTTJJ--e449J:q/888++--55555
vx000TC5uMMM)7eex"AsA.&'/*=*=j!*L*L&L
#/888CCEE++--55555""$$
vx00000r!   c                    t           j        g}t          |          }t          |dg          }t          ||          }t          ||          }t          ||          }|j        sJ |j        sJ |j        sJ |j                                        dgk    sJ |j                                        dgk    sJ |j                                        dgk    sJ d S )Nr   rK   r-   r   )r'   nanr   r   emptyrK   tolist)r   r-   just_na_listjust_na_seriesjust_na_series_indexres_list
res_seriesres_series_indexs           r   test_get_dummies_just_naz'TestGetDummies.test_get_dummies_just_na   s    x--%l3%@@@|F;;; ???
&';FKKK~%%%%~$$&&1#----&&((QC////%,,..3%777777r!   c           	         ddt           j        g}t          |||          }t          g dg dd|                     |                    }|rD|j        dk    r|                    t          d	          }n|                    t          d
	          }t          j	        ||           t          |d||          }t          t           j        g ddg ddg di|                     |                    }|
                    ddt           j        gd          }|j        |_        |rD|j        dk    r|                    t          d	          }n|                    t          d
	          }t          j	        ||           t          t           j        gd||          }t          t          ddg          t           j        g|                     |                    }	t          j        |j        |	j                   d S )Nr   r   rE   r>   r?   )r   r   r5   FrB   rD   Tdummy_nar-   r(   r@   r   axisr   rq   rS   r(   )r'   rs   r   r   r0   rG   rH   r   rI   rJ   reindexrS   r   assert_numpy_array_equalvalues)
r   r-   r(   sresexpres_naexp_nares_just_naexp_just_nas
             r   test_get_dummies_include_naz*TestGetDummies.test_get_dummies_include_na   s   #rv!F%888))))),,D4H4H4O4O
 
 
  	=zS  iii>>iii<<
c3''' QfEJJJVYYYYYYYYY?&&u--
 
 
 c26 2;; 	CzS  keDDkcBB
ff---!26(T&PUVVV1QC   26($:N:Nu:U:U
 
 
 	#K$68JKKKKKr!   c                     d}t          j        d          }|||g}t          |d|          }t          dg dd| g di          }|r|                    t
          d	
          }t          j        ||           d S )NezLATIN SMALL LETTER E WITH ACUTEletterprefixr-   letter_e)TFFletter_)FTTFrB   )unicodedatalookupr   r   rH   r   rI   rJ   )r   r-   r   eacuter   r   r   s          r   test_get_dummies_unicodez'TestGetDummies.test_get_dummies_unicode   s    #$EFF!HV<<<---/A/A/ACVCVCVW
 
  	;))KE)::C
c3'''''r!   c           	      `   |ddg         }t          ||          }t          g dg dg dg ddt          	          }|rXt          t          g dd
	          t          g dd
	          t          g dd
	          t          g dd
	          d          }t	          j        ||           d S )Nr   r   rr   r   r   r   r?   r   r   r   r@   A_aA_bB_bB_cr5   rd   )r   r   rd   r   rI   rJ   r   r    r-   rP   rO   s        r   test_dataframe_dummies_all_objz-TestGetDummies.test_dataframe_dummies_all_obj   s    c
^R///IIiii			)))TT
 
 
  	 &yyy???&yyy???&yyy???&yyy???	  H 	fh/////r!   c                 N   |ddg         }|                     d|d          }t          |          }t          g dg dg dg dd	t          
          }|dk    r5|j        t
          j        u r"|ddg                              d          |ddg<   t          j        ||           d S )Nr   r   r\   r   r   r   r?   r   r@   r   r5   stringr   r   boolean)	astyper   r   rd   na_valuepdNArI   rJ   )r   r    any_string_dtyperP   rO   s        r   #test_dataframe_dummies_string_dtypez2TestGetDummies.test_dataframe_dummies_string_dtype   s    c
^YYU)9::;;R yy yy yy yy	  
 
 
 x'',<,E,N,N'/'?'F'Fy'Q'QHeU^$
fh/////r!   c           
         t          |||          }|r4t          }|j        dk    rt          |d          }nt          |d          }nt          j        }|}t          g d |g d|           |g d|           |g d	|           |g d
|          d          }|g d         }t          j        ||           d S )NrE   r   Fr   r   r   r5   r?   r   r@   r   r   r   r   r   )	r   r   rG   r   r'   arrayr   rI   rJ   r   r    r-   r(   rP   arrtyprO   s           r   "test_dataframe_dummies_mix_defaultz1TestGetDummies.test_dataframe_dummies_mix_default   s   Re<<< 	CzS  !%//!%++(CCYYs999C000s999C000s999C000s999C000 
 
 ===>
fh/////r!   c                 8   ddg}t          |||          }t          g dg dg dg dg dd	          }|d
g         |d
g<   g d}|d
g|z            }|rt          nt          ||                             fd          ||<   t          j        ||           d S )Nfrom_Afrom_Br   r   TFTFTFTTFFFTr   from_A_afrom_A_bfrom_B_bfrom_B_cr   r   r   r   r   c                      |           S r&   rZ   )xr   s    r   <lambda>zCTestGetDummies.test_dataframe_dummies_prefix_list.<locals>.<lambda>  s    A r!   )r   r   r   r   rH   rI   rJ   )r   r    r-   prefixesrP   rO   colsr   s          @r   "test_dataframe_dummies_prefix_listz1TestGetDummies.test_dataframe_dummies_prefix_list  s    h'R@@@YY///000///000 
 
 cU)#???SEDL)#/kk!$--.>.>.>.>??
fh/////r!   c                    t          |d|          }g d}t          g dg dg dgdg|z             }|                    dt          j        i          }|rtt          j        t          g d	d
          t          g ddd          t          g ddd          t          g ddd          t          g ddd          gd          }t          j	        ||           d S )Nbadr   )bad_abad_br   bad_cr   TFTFr   FTTFr   TFFTr   rS   r   rW   r   r   zSparse[bool])rX   r(   r   r   r   r   r   r   r   )
r   r   r   r'   rb   r   concatr   rI   rJ   )r   r    r-   rP   bad_columnsrO   s         r   !test_dataframe_dummies_prefix_strz0TestGetDummies.test_dataframe_dummies_prefix_str  s7   Rf===:::---------
 EK'
 
 
 ??C?33 	 y9993///...WNSSS///g^TTT...WNSSS///g^TTT 	 	 	H 	fh/////r!   c                    t          |dgdg|          }t          g dg dg dg dd          }|j        }||d	d                                       t                    ||d	d          <   |d
g         |d
g<   |r0ddg}||                             t          dd                    ||<   t          j        ||           d S )Nr   r   )r   rS   r-   r   r   r   r?   )r   r   r   r   r   r   r   r   rd   F)r   r   rS   r   rd   r   rI   rJ   r   r    r-   rP   rO   r   s         r   test_dataframe_dummies_subsetz,TestGetDummies.test_dataframe_dummies_subset3  s    R
SE&QQQ$__YY%II%II	 
 
 %d122h/66t<<abbcU)# 	O
+D%d^22;vu3M3MNNHTN
fh/////r!   c                    t          |d|          }t          g dg dg dg dg dd          }|d	g         |d	g<   |g d         }|r0g d
}||                             t          dd                    ||<   t	          j        ||           t          |ddg|          }|                    ddd          }t	          j        ||           t          |ddd|          }t	          j        ||           d S )Nz..
prefix_sepr-   r   r   r   r   r   )r   A..aA..bB..bB..cr   )r   r   r   r   rd   F__B__bB__c)r   r   r   r   )r   r   r   r   rI   rJ   renamer   s         r   !test_dataframe_dummies_prefix_sepz0TestGetDummies.test_dataframe_dummies_prefix_sepE  sE   RD@@@YY+++,,,+++,,, 
 
 cU)#AAAB 	O333D%d^22;vu3M3MNNHTN
fh///RT4LHHH??FF+K+K?LL
fh///R$T,B,B6RRR
fh/////r!   c                     t          j        d          }t          j        t          |          5  t          |dg|           d d d            d S # 1 swxY w Y   d S )NzPLength of 'prefix' (1) did not match the length of the columns being encoded (2)r2   ztoo fewr   reescaper6   r7   r8   r   r   r    r-   r9   s       r   (test_dataframe_dummies_prefix_bad_lengthz7TestGetDummies.test_dataframe_dummies_prefix_bad_length_  s    i
 
 ]:S111 	? 	?I;v>>>>	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	?   AAAc                     t          j        d          }t          j        t          |          5  t          |dg|           d d d            d S # 1 swxY w Y   d S )NzTLength of 'prefix_sep' (1) did not match the length of the columns being encoded (2)r2   r   r   r   r   s       r   ,test_dataframe_dummies_prefix_sep_bad_lengthz;TestGetDummies.test_dataframe_dummies_prefix_sep_bad_lengthg  s    i
 
 ]:S111 	? 	?wv>>>>	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	?r   c                 x   ddd}t          g dg dg dd          }t          |||          }t          g dg d	g d
g dg dd          }g d}||                             t                    ||<   |r,||                             t	          dd                    ||<   t          j        ||           d S )Nr   r   r   r   r   r   )r   r   r   r   r   r?   r   r@   r   r   rd   F)r   r   r   rd   r   rI   rJ   )r   r-   r   r    rP   rO   rS   s          r   "test_dataframe_dummies_prefix_dictz1TestGetDummies.test_dataframe_dummies_prefix_dicto  s    !11YYY___???SSTTR@@@YY%II%II%II%II 
 
 CBB$W-44T:: 	U ( 1 8 8VU9S9S T THW
fh/////r!   c                    t           j        t           j        t           j        g|j        dd d f<   t          |d||                              d          }|r4t
          }|j        dk    rt          |d          }nt          |d          }nt           j        }|}t          dd	dt           j        g |g d
|           |g d|           |g d|           |g d|           |g d|           |g d|          d                              d          }t          j        ||           t          |d||          }|g d         }t          j        ||           d S )Nr   Tr~   r   r   r   Fr   r   )r   r   r   r   r5   r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   A_nanr   r   B_nanr   )r'   rs   locr   ri   r   rG   r   r   r   rI   rJ   r   s           r   test_dataframe_dummies_with_naz-TestGetDummies.test_dataframe_dummies_with_na  s   /q!!!tR$vUKKKVV W 
 
  	CzS  !%//!%++(CCArv&s<<<s333s<<<s333\\\555s<<<s333s<<<s333\\\555 

 

 *!*

 	 	fh///R%eLLL===>
fh/////r!   c                 "   t          g d          |d<   t          |||                              d          }|r4t          }|j        dk    rt          |d          }nt          |d          }nt          j        }|}t          g d	 |g d
|           |g d|           |g d|           |g d|           |g d|           |g d|          d                              d          }t          j
        ||           d S )Nr   yr   catrE   r   r   r   Fr   r   r   r5   r?   r   r@   r>   r   r   r   )r   r   r   r   r   cat_xcat_y)r   r   ri   r   rG   r   r'   r   r   rI   rJ   r   s           r   'test_dataframe_dummies_with_categoricalz6TestGetDummies.test_dataframe_dummies_with_categorical  sW   005	Re<<<GGQGOO 	CzS  !%//!%++(CCYYs999C000s999C000s999C000s999C000YYYc222YYYc222 

 

 *!*

 	 	fh/////r!   zget_dummies_kwargs,expecteddata   är   u   ä_aTr   u   x_ä)r   r   )r   r   u   xäac                 H    t          di |}t          j        ||           d S )NrZ   )r   rI   rJ   )r   get_dummies_kwargsrO   rP   s       r   test_dataframe_dummies_unicodez-TestGetDummies.test_dataframe_dummies_unicode  s1    0 22122
fh/////r!   c                    t          d          }t          |          }t          |t          d                    }t          g dg ddt                    }t	          |d|          }|r|                    t          d	
          }t          j        ||           t	          |d|          }t          j        ||           t          d          |_	        t	          |d|          }t          j        ||           d S )Nr<   r=   r?   r@   )r   r   r5   T
drop_firstr-   FrB   )
rF   r   r   rd   r   rH   r   rI   rJ   rK   r   r-   rL   rM   rN   rO   rP   s          r   !test_get_dummies_basic_drop_firstz0TestGetDummies.test_get_dummies_basic_drop_first  s     e&>>U44999999==TJJJVVDDD 	E~~ke~DDH
fh///X$vFFF
fh///e^VLLL
fh/////r!   c                    t          d          }t          |          }t          |t          d                    }t          t          d                    }t	          |d|          }t          j        ||           t	          |d|          }t          j        ||           t          t          d                    }t	          |d|          }t          j        ||           d S )Naaar=   r   rq   Tr  )rF   r   r   r
   r   rI   rJ   r  s          r   +test_get_dummies_basic_drop_first_one_levelz:TestGetDummies.test_get_dummies_basic_drop_first_one_level  s    e&>>U44:a==111VVDDD
fh///X$vFFF
fh///4;;///^VLLL
fh/////r!   c                    ddt           j        g}t          |d|          }t          dg dit                    }|r|                    t          d          }t          j        ||           t          |dd|	          }t          dg dt           j        g d
it                    	                    dt           j        gd          }|r|                    t          d          }t          j        ||           t          t           j        gdd|	          }t          t          d                    }t          j        ||           d S )Nr   r   Tr  r?   r5   FrB   r   r  r-   r@   r   r   rq   )r'   rs   r   r   rd   rH   r   rI   rJ   r   r
   )	r   r-   s_NAr   r   r   r   r   r   s	            r   $test_get_dummies_basic_drop_first_NAz3TestGetDummies.test_get_dummies_basic_drop_first_NA  sP   S"&!$4???iii(555 	;))KE)::C
c3'''TDT&QQQCBFIII>dKKKSS"&M T 
 
  	A\\+%\@@F
ff---!VHtV
 
 
  jmm444
k;77777r!   c                     |ddg         }t          |d|          }t          g dg ddt                    }|r|                    t          d	
          }t          j        ||           d S )Nr   r   Tr  r?   r@   )r   r   r5   FrB   )r   r   rd   rH   r   rI   rJ   r   s        r   !test_dataframe_dummies_drop_firstz0TestGetDummies.test_dataframe_dummies_drop_first  s|    c
^RD@@@YYYyyyAANNN 	E~~ke~DDH
fh/////r!   c                 T   t          g d          |d<   t          |d|          }t          g dg dg dg dd	          }g d
}||                             t                    ||<   |g d	         }|r|D ]}t          ||                   ||<   t          j        ||           d S )Nr   r   Tr  r   r?   r@   r   )r   r   r   r   )r   r   r   )r   r   r   r   rd   r   rI   rJ   )r   r    r-   r(   rP   rO   r   cols           r   2test_dataframe_dummies_drop_first_with_categoricalzATestGetDummies.test_dataframe_dummies_drop_first_with_categorical$  s    005	RD@@@))IIIiii)))TT
 
 '&&!$..t448889 	; ; ; +HSM : :
fh/////r!   c                 N   t           j        t           j        t           j        g|j        dd d f<   t          |dd|                              d          }t          dddt           j        gg dg dg d	g dd
          }g d}||                             t                    ||<   |                    d          }|r|D ]}t          ||                   ||<   t          j
        ||           t          |dd|          }|g d         }t          j
        ||           d S )Nr   Tr  r   r   r   r   r   r   )r   r   r   r   r   )r   r   r   r   F)r   r   r   )r'   rs   r   r   ri   r   r   rd   r   rI   rJ   )r   r    r-   rP   rO   r   r  s          r   )test_dataframe_dummies_drop_first_with_naz8TestGetDummies.test_dataframe_dummies_drop_first_with_na2  sU   /q!!!t$v
 
 

*!*

 	 Arv&#||%#||% 
 
 0//!$..t44&&A&.. 	; ; ; +HSM : :
fh///R%DPPP///0
fh/////r!   c                    t          g d          }t          |          }t          ddgddgddggddgt                    }t	          j        ||           t          t          g d                    }t          |          }t          ddgddgddggt          ddg          t                    }t	          j        ||           d S )	Nr   r   r   r   r   r   r   r   r   r   )r   r   r   rd   rI   rJ   r   )r   r   rP   rO   s       r   test_get_dummies_int_intz'TestGetDummies.test_get_dummies_int_intM  s    iii  T""q!fq!fq!f51vTRRR
fh///k///2233T""VaVaV$k3*.E.ET
 
 
 	fh/////r!   c                 B   t          g dt          g d          g dg dd          }g d}t          g dg dg dg|          }||d	d                                       |          ||d	d          <   t          |d
dg|          }t	          j        ||           d S )Nr  r   )      ?       @r  )r   r   r   D)r   r  A_1A_2B_ar   )r   r  r   r   r   r   )r   r  r   r   r   r   r   r   r   r   r   )r   r   r   r   rI   rJ   )r   r(   r   rS   rO   rP   s         r   test_get_dummies_int_dfz&TestGetDummies.test_get_dummies_int_dfZ  s    YY 11YY$__	 
 
 988!!!#7#7#79M9M9MN
 
 
 !) 5 < <U C CTC:UCCC
fh/////r!   orderedFc                    t          t          d          t          d          |          }t          ||          }t          j        g dg dg|                     |                    }t          |j        |j        |          }t          |||                     |                    }t          j
        ||           d S )Nxyxyz)
categoriesr   r5   r>   r?   r   )r   rF   r   r'   r   r0   r   r$  r   rI   rJ   )r   r(   r   r   rP   r   r   rO   s           r   1test_dataframe_dummies_preserve_categorical_dtypez@TestGetDummies.test_dataframe_dummies_preserve_categorical_dtypel  s     $t**egNNNS...xIII.d6J6J56Q6QRRRNs~w
 
 
 T4t7K7KE7R7RSSS
fh/////r!   c                     t          j        ddgddgd          }t          |dg|          }|                    dg	          }t	          j        |dg         |           d S )
Nr   r   ABCD)GDPNationr*  rS   r-   r)  r   )r   	from_dictr   r   rI   rJ   )r   r-   r    df2s       r   *test_get_dummies_dont_sparsify_all_columnsz9TestGetDummies.test_get_dummies_dont_sparsify_all_columnsz  sn      !QD$<!H!HIIhZ???jj%j))
b%k3/////r!   c                 *   g d|_         t          |                              d          }t          g dg dg dgg d                              d          }|                    d	t
          j        i          }t          j        ||           d S )
N)r   r   r   r   r   r   r   r   )r   r   r   r   A_cr   r   )	rS   r   ri   r   r   r'   rb   rI   rJ   r   r    rP   rO   s       r   "test_get_dummies_duplicate_columnsz1TestGetDummies.test_get_dummies_duplicate_columns  s    $__
R+++33---------
 655
 
 
 *!*

 	 ??C?33
fh/////r!   c                    t          dddgi          }t          |dgd          }t          dd          }t          t          ddg|	          t          ddg|	          d
          }t	          j        ||           d S )Nr   r   r   Tr+  rd   Fr   r5   )r  r  )r   r   r   r   rI   rJ   )r   r    rP   r(   rO   s        r   test_get_dummies_all_sparsez*TestGetDummies.test_get_dummies_all_sparse  s    aV}%%R#t<<<FE**"Aq6777"Aq6777 
 
 	fh/////r!   r   bazc                     t          g dg dg dg dd          }d}t          j        t          |          5  t	          ||           d d d            d S # 1 swxY w Y   d S )	N)r   r   r            )oner:  r:  twor;  r;  )r   r   r   r   r   r   )r   r   zqwt)barfoor5  zooz1Input must be a list-like for parameter `columns`r2   r   )r   r6   r7   	TypeErrorr   )r   r   r    r9   s       r   #test_get_dummies_with_string_valuesz2TestGetDummies.test_get_dummies_with_string_values  s     )))AAA555555	 
 
 B]9C000 	, 	,F++++	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	,s   AA Ac                     t          t          d                    }t          ||          }t          g dg dg dd|          }t	          j        ||           d S )Nabcar5   r   r   r   r   r   r   rA   )r   rF   r   r   rI   rJ   )r   any_numeric_ea_and_arrow_dtypeserrP   rO   s        r    test_get_dummies_ea_dtype_seriesz/TestGetDummies.test_get_dummies_ea_dtype_series  ss    T&\\""S(FGGG,,\\\EE0
 
 
 	fh/////r!   c                     t          dt          d          i          }t          ||          }t          g dg dg dd|          }t          j        ||           d S )Nr   rF  r5   rG  r   r   )x_ax_bx_c)r   rF   r   rI   rJ   )r   rH  r    rP   rO   s        r   #test_get_dummies_ea_dtype_dataframez2TestGetDummies.test_get_dummies_ea_dtype_dataframe  sx    T&\\*++R'EFFF LLlllKK0
 
 
 	fh/////r!   
dtype_typer   categoryc                 P   |}|j         t          j        u rdnd}|dk    rt          t	          dg|                    }t          t          dg|          dd          }t          |          }t          dt          dg|          d	          }t          j	        ||           d S )
Nr   rd   rQ  r   r5   r   rX   r   Tr   name_a)
r   r   r   r   r	   r   r   r   rI   rJ   )r   rP  string_dtype_no_objectr(   	exp_dtyper    rP   rO   s           r   test_get_dummies_ea_dtypez(TestGetDummies.test_get_dummies_ea_dtype  s     '!&25!8!8IIf	##$UC5%%8%899EuE : : :CCDDR1vY0O0O0OPPQQ
fh/////r!   pyarrowc                    t          t          dgt          t          j                                        dd          }t          |          }t          dt          dgd          d          }t          j        ||           t          t          dgt          t          dgt          t          j                                                            dd          }t          |          }t          j        ||           d S )Nr   r5   r   rS  Tzbool[pyarrow]rT  )
r   r   r   par   r   rI   rJ   r   r	   r1  s       r   test_get_dummies_arrow_dtypez+TestGetDummies.test_get_dummies_arrow_dtype  s     uJry{{4K4K L L LSTUUVVR1v_0U0U0UVVWW
fh///E*5#j>U>U+V+V+VWW    
 
 R
fh/////r!   )5__name__
__module____qualname__r6   fixturer    r'   float64rd   r(   r-   r0   r:   rQ   ro   r|   r   r   r   r   r   r   r   r   r   r   r   r   r   r   markparametrizer   r  r  r	  r  r  r  r  r  r  r%  r.  r2  r4  rD  rJ  rO  rX  td
skip_if_nor\  rZ   r!   r   r   r   !   sl       ^W W ^W V^GT2:tTBCCC' ' DC' V^GX.///) ) 0/)
  
, , ,
0 0 0011 11 11f8 8 8"!L !L !LF( ( (0 0 0&0 0 0$0 0 0.0 0 0(0 0 0:0 0 0$0 0 04? ? ?? ? ?0 0 0, 0  0  0D0 0 06 [% D3%=112	6D6*++
 C$=112	6D6*++
 #C#<00DAA	6D6*++
 #C#<00EE	6D6*++	
 *0 0+ *00 0 0*0 0 0$8 8 800 0 00 0 00 0 060 0 00 0 0$ [Yu660 0 760 [Xe}550 0 6500 0 0$
0 
0 
0 [Xw//, , 0/, 0 0 00 0 0 [\Hj+ABB	0 	0 CB	0 R]90 0 0 0 0r!   r   ) r   r   numpyr'   r6   pandas.util._test_decoratorsutil_test_decoratorsrd  pandas.core.dtypes.commonr   pandasr   r   r   r   r   r   r	   r
   r   r   r   pandas._testing_testingrI   pandas.core.arrays.sparser   rY  r[  ImportErrorr   rZ   r!   r   <module>rp     sz   				          ) ) ) ) ) ) ) ) ) 6 6 6 6 6 6                                  1 1 1 1 1 1   	BBBC0 C0 C0 C0 C0 C0 C0 C0 C0 C0s   	A AA