
    o[we4e                         d dl Z d dlZd dlZd dl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 d dlmZ d dlmZ  G d d          ZdS )    N)is_integer_dtype)CategoricalCategoricalIndex	DataFrame
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; Z0dS )<TestGetDummiesc                 4    t          g dg dg dd          S )Nabr   r   r   c         )ABC)r   )selfs    Elib/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_dtype&   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_object+   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   r2   r   F
fill_value        r*   r%   )listr   r   r-   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_basic0   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||          }ddd}
d|
                    |	d
          z   |
|	<   t          |
d                                          }|j                                        }d |j        D             |_        |                                }t          j        ||           d S )Nr9   )r   r   r   r   r   )r   r   r   r   r   )r   r   r   r   r   r>   r;   r<   r=   )r%   columnsr   FrA   r?   rB   )rP   r*   r%   zSparse[z, ]   countnamec                 ,    g | ]}t          |          S  str.0is     r   
<listcomp>z?TestGetDummies.test_get_dummies_basic_types.<locals>.<listcomp>l       5551A555r   r   r   )int64r1   r   c                 ,    g | ]}t          |          S rW   rX   rZ   s     r   r]   z?TestGetDummies.test_get_dummies_basic_types.<locals>.<listcomp>v   r^   r   )rC   r   r   r-   r   boolrE   r   r
   rF   rG   rP   rU   dtypesvalue_countsrH   assert_series_equalget
sort_index)r   r*   r%   rI   rJ   s_dfrL   r@   rM   
dtype_nameexpected_countss              r   test_get_dummies_basic_typesz+TestGetDummies.test_get_dummies_basic_typesH   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$%33&'/*=*=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   rH   r*   r   )r$   nanr   r
   emptyrH   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_naz   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   rB   r;   r<   )r   r   r2   Fr?   rA   Tdummy_nar*   r%   r=   r   axisr   rl   rP   r%   )r$   rn   r
   r   r-   rD   rE   r   rF   rG   reindexrP   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_)FTTFr?   )unicodedatalookupr
   r   rE   r   rF   rG   )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   rm   r   r   r   r<   r   r   r   r=   A_aA_bB_bB_cr2   ra   )r
   r   ra   r   rF   rG   r   r   r*   rM   rL   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                     |ddg         }|                     ddd          }t          |          }t          g dg dg dg d	d
t                    }t	          j        ||           d S )Nr   r   r1   stringr   r   r   r<   r   r=   r   r2   )astyper
   r   ra   rF   rG   r   r   rM   rL   s       r   #test_dataframe_dummies_string_dtypez2TestGetDummies.test_dataframe_dummies_string_dtype   s    c
^YYXH5566R yy yy yy yy	  
 
 
 	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 )NrB   r   Fr   r   r   r2   r<   r   r=   r   r   r   r   r   )	r
   r   rD   r	   r$   arrayr   rF   rG   r   r   r*   r%   rM   arrtyprL   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#   rW   )xr   s    r   <lambda>zCTestGetDummies.test_dataframe_dummies_prefix_list.<locals>.<lambda>  s    A r   )r
   r   r   r   rE   rF   rG   )r   r   r*   prefixesrM   rL   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   rP   r   rT   r   r   zSparse[bool])rU   r%   r   r   r   r   r   r   r{   )
r
   r   r   r$   r_   pdconcatr   rF   rG   )r   r   r*   rM   bad_columnsrL   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   rP   r*   r   r   r   r<   )r   r   r   r   r   r   r   r   ra   F)r
   r   rP   r   ra   r	   rF   rG   r   r   r*   rM   rL   r   s         r   test_dataframe_dummies_subsetz,TestGetDummies.test_dataframe_dummies_subset&  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   ra   F__B__bB__c)r   r   r   r   )r
   r   r   r	   rF   rG   renamer   s         r   !test_dataframe_dummies_prefix_sepz0TestGetDummies.test_dataframe_dummies_prefix_sep8  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)r/   ztoo fewr   reescaper3   r4   r5   r
   r   r   r*   r6   s       r   (test_dataframe_dummies_prefix_bad_lengthz7TestGetDummies.test_dataframe_dummies_prefix_bad_lengthR  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)r/   r   r   r   r   s       r   ,test_dataframe_dummies_prefix_sep_bad_lengthz;TestGetDummies.test_dataframe_dummies_prefix_sep_bad_lengthZ  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   ra   F)r   r
   r   ra   r	   rF   rG   )r   r*   r   r   rM   rL   rP   s          r   "test_dataframe_dummies_prefix_dictz1TestGetDummies.test_dataframe_dummies_prefix_dictb  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   Try   r   r{   r   Fr   r   )r   r   r   r   r2   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$   rn   locr
   rf   r   rD   r	   r   r   rF   rG   r   s           r   test_dataframe_dummies_with_naz-TestGetDummies.test_dataframe_dummies_with_nax  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   catrB   r   r{   r   Fr   r   r   r2   r<   r   r=   r;   r   r   r   )r   r   r   r   r   cat_xcat_y)r   r
   rf   r   rD   r	   r$   r   r   rF   rG   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 )NrW   )r
   rF   rG   )r   get_dummies_kwargsrL   rM   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 )Nr9   r:   r<   r=   )r   r   r2   T
drop_firstr*   Fr?   )
rC   r   r   ra   r
   rE   r   rF   rG   rH   r   r*   rI   rJ   rK   rL   rM   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   rl   Tr   )rC   r   r   r   r
   rF   rG   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<   r2   Fr?   rz   r   r*   r=   r   r{   rl   )r$   rn   r
   r   ra   rE   r   rF   rG   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   r2   Fr?   )r
   r   ra   rE   r   rF   rG   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   ra   r   rF   rG   )r   r   r*   r%   rM   rL   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$   rn   r   r
   rf   r   r   ra   r   rF   rG   )r   r   r*   rM   rL   r   r	  s          r   )test_dataframe_dummies_drop_first_with_naz8TestGetDummies.test_dataframe_dummies_drop_first_with_na%  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   ra   rF   rG   r   )r   r   rM   rL   s       r   test_get_dummies_int_intz'TestGetDummies.test_get_dummies_int_int@  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
   rF   rG   )r   r%   r   rP   rL   rM   s         r   test_get_dummies_int_dfz&TestGetDummies.test_get_dummies_int_dfM  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  r2   r;   r<   r}   )r   rC   r
   r$   r   r-   r   r  r   rF   rG   )r   r%   r  r   rM   r   r   rL   s           r   1test_dataframe_dummies_preserve_categorical_dtypez@TestGetDummies.test_dataframe_dummies_preserve_categorical_dtype_  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"  rP   r*   r!  r   )r   	from_dictr
   r~   rF   rG   )r   r*   r   df2s       r   *test_get_dummies_dont_sparsify_all_columnsz9TestGetDummies.test_get_dummies_dont_sparsify_all_columnsm  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   )	rP   r
   rf   r   r   r$   r_   rF   rG   r   s       r   "test_get_dummies_duplicate_columnsz1TestGetDummies.test_get_dummies_duplicate_columnsv  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#  ra   Fr   r2   )r  r  )r   r
   r	   r   rF   rG   )r   r   rM   r%   rL   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            )oner1  r1  twor2  r2  )r   r   r   r   r   r   )r   r   zqwt)barfoor,  zooz1Input must be a list-like for parameter `columns`r/   r   )r   r3   r4   	TypeErrorr
   )r   r   r   r6   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abcar2   r   r   r   r   r   r   r>   )r   rC   r
   r   rF   rG   )r   any_numeric_ea_and_arrow_dtypeserrM   rL   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   r=  r2   r>  r   r   )x_ax_bx_c)r   rC   r
   rF   rG   )r   r?  r   rM   rL   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   )1__name__
__module____qualname__r3   fixturer   r$   float64ra   r%   r*   r-   r7   rN   rj   rw   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&  r)  r+  r;  rA  rF  rW   r   r   r   r      s       ^W W ^W V^GT2:tTBCCC' ' DC' V^GX.///) ) 0/)
  
, , ,
0 0 0001 01 01d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 0 0r   r   )r   r   numpyr$   r3   pandas.core.dtypes.commonr   pandasr   r   r   r   r   r   r	   r
   pandas._testing_testingrF   pandas.core.arrays.sparser   r   rW   r   r   <module>rT     s	   				          6 6 6 6 6 6                            1 1 1 1 1 1`
0 `
0 `
0 `
0 `
0 `
0 `
0 `
0 `
0 `
0r   