
    Mh3                     ~   S SK rS SKrS SKJrJrJrJr  S SKJ	r
  \R                  S 5       r\R                  S 5       rS rS rS rS rS	 rS
 rS rS rS rS rS rS rS rS rS rS rS rS rS r S r!S r"S r#S r$S r%\RL                  RO                  S\RP                  " S\" S/ S Q05      S!S"9\RP                  " S#\" S/ S$Q05      S%S"9\RP                  " S&\" S/ S'Q05      S(S"9\RP                  " S \" S/ S)Q05      S*S"9\RP                  " S+\" S/ S,Q05      S-S"9\RP                  " S.\" S/ S/Q05      S0S"9/5      S1 5       r)S2 r*S3 r+\RL                  RO                  S\RP                  " S4\" / S5Q/ S6QS7.5      S!S"9\RP                  " S \" / S)Q/ S8QS7.5      S*S"9\RP                  " S+\" / S,Q/ S9QS7.5      S:S"9\RP                  " S#S;S<.\" / S=Q/ S>QS7.5      S?S"9\RP                  " SS+S<.\" / S,Q/ S@QS7.5      SAS"9\RP                  " S.S&S+/S<.\" SBSCS&S+/// SDQS7.5      SES"9/5      SF 5       r,SG r-SH r.SI r/g)J    N)	DataFrameSeriesfrom_dummiesget_dummiesc                  8    [        / SQ/ SQ/ SQ/ SQ/ SQS.5      $ )N   r   r	   r   r	   r   r	   r   r   r   r   r	   col1_acol1_bcol2_acol2_bcol2_cr        X/var/www/html/env/lib/python3.13/site-packages/pandas/tests/reshape/test_from_dummies.pydummies_basicr      #    	
 r   c                  8    [        / SQ/ SQ/ SQ/ SQ/ SQS.5      $ )Nr   r
   r   r   r   r   r   r   r   r   r   dummies_with_unassignedr      r   r   c                      / SQn [         R                  " [        SS9   [        U 5        S S S 5        g ! , (       d  f       g = f)Nr
   zBExpected 'data' to be a 'DataFrame'; Received 'data' of type: listmatchpytestraises	TypeErrorr   dummiess    r   test_error_wrong_data_typer%   '   s3    G	S
 	W	
 
 
s	   3
Ac                      [        / SQ/ SQS.5      n [        R                  " [        SS9   [	        U 5        S S S 5        g ! , (       d  f       g = f)Nr   r
   abHDummy DataFrame contains unassigned value\(s\); First instance in row: 2r   r   r    r!   
ValueErrorr   r#   s    r   (test_error_no_prefix_contains_unassignedr-   0   ?    ii89G	(
 	W
 
 
   A
Ac                      [        / SQ/ SQS.5      n [        R                  " [        SS9   [	        U SS/S9  S S S 5        g ! , (       d  f       g = f)	Nr   r   r	   r	   r'   rExpected 'default_category' to be of type 'None', 'Hashable', or 'dict'; Received 'default_category' of type: listr   cddefault_categoryr   r    r!   r"   r   r#   s    r   0test_error_no_prefix_wrong_default_category_typer8   <   sE    ii89G	9
 	WSz:
 
 
s   A
Ac                      [        / SQ/ SQS.5      n [        R                  " [        SS9   [	        U 5        S S S 5        g ! , (       d  f       g = f)Nr   r1   r'   zHDummy DataFrame contains multi-assignment\(s\); First instance in row: 2r   r+   r#   s    r   %test_error_no_prefix_multi_assignmentr:   H   r.   r/   c                      [        / SQSS[        R                  /S.5      n [        R                  " [
        SS9   [        U 5        S S S 5        g ! , (       d  f       g = f)Nr   r   r	   r'   z0Dummy DataFrame contains NA value in column: 'b'r   )r   npnanr    r!   r,   r   r#   s    r   !test_error_no_prefix_contains_nanr>   T   sG    iq!RVVn=>G	M
 	W
 
 
s   A
Ac                      [        / SQ/ SQ/ SQS.5      n [        R                  " [        SS9   [	        U 5        S S S 5        g ! , (       d  f       g = f)N)r	         r	   )r   r	   r      )c1c2c3c4r(   r)   r3   (Passed DataFrame contains non-dummy datar   r7   r#   s    r   test_error_contains_non_dummiesrI   \   sE    4LMG 
9
 	W	
 
 
s   A
Ac                      [        / SQ/ SQ/ SQ/ SQS.5      n [        R                  " [        SS9   [	        U SS9  S S S 5        g ! , (       d  f       g = f)Nr   r
   )r   r   zcol2-azcol2-bz*Separator not specified for column: col2-ar   _sepr+   r#   s    r   *test_error_with_prefix_multiple_seperatorsrN   g   sM    		
G 
<
 	W#&	
 
 
s   A
Ac                     [         R                  " [        SS9   [        U S/S9  S S S 5        g ! , (       d  f       g = f)NzJExpected 'sep' to be of type 'str' or 'None'; Received 'sep' of type: listr   rK   rL   r   r   s    r   %test_error_with_prefix_sep_wrong_typerQ   w   s4    	,
 	].
 
 
s   /
=c                 ~    [         R                  " [        SS9   [        U SS9  S S S 5        g ! , (       d  f       g = f)Nr*   r   rK   rL   r    r!   r,   r   r   s    r   *test_error_with_prefix_contains_unassignedrU      s3    	(
 	,#6
 
 
s   .
<c                     [         R                  " [        SS9   [        U SSS/S9  S S S 5        g ! , (       d  f       g = f)Nr2   r   rK   xyrM   r6   r   rT   s    r   2test_error_with_prefix_default_category_wrong_typerZ      s9    	9
 	,#c
S
 
 
   1
?c                     [         R                  " [        SS9   [        U SSS0S9  S S S 5        g ! , (       d  f       g = f)Nz^Length of 'default_category' \(1\) did not match the length of the columns being encoded \(2\)r   rK   col1rW   rY   rS   rT   s    r   9test_error_with_prefix_default_category_dict_not_completer^      s<     
=
 	,#QTV
 
 
r[   c                     U S   R                  S5      U S'   [        R                  U R                  S'   [        R
                  " [        SS9   [        U SS9  S S S 5        g ! , (       d  f       g = f)Nr   float64rB   r   z5Dummy DataFrame contains NA value in column: 'col2_c'r   rK   rL   )astyper<   r=   locr    r!   r,   r   rP   s    r   #test_error_with_prefix_contains_nanrd      s^    +H5<<YGM(%'VVMk"	R
 	],
 
 
s   A""
A0c                     U S   R                  [        5      U S'   SU R                  S'   [        R                  " [
        SS9   [        U SS9  S S S 5        g ! , (       d  f       g = f)Nr   strra   rH   r   rK   rL   )rb   objectrc   r    r!   r"   r   rP   s    r   +test_error_with_prefix_contains_non_dummiesrh      sR    +H5<<VDM(%*Mk"	y(S	T], 
U	T	Ts   A
A&c                      [        / SQ/ SQ/ SQ/ SQ/ SQS.5      n [        R                  " [        SS9   [	        U S	S
9  S S S 5        g ! , (       d  f       g = f)Nr   )r	   r	   r   r
   r   r   r   zHDummy DataFrame contains multi-assignment\(s\); First instance in row: 0r   rK   rL   r+   r#   s    r   (test_error_with_prefix_double_assignmentrj      sR    	
G 
(
 	W#&
 
 
s   A		
Ac                      [        / SQ5      n [        U 5      n[        U5      n[        S/ SQ05      n[        R
                  " X#5        g )Nr(   r)   r3   r(    )r   r   r   r   tmassert_frame_equal
categoriesr$   resultexpecteds       r   "test_roundtrip_series_to_dataframert      s?    ,-J*%G'"F"234H&+r   c                  |    [        S/ SQ05      n [        U 5      n[        USS9nU n[        R                  " X#5        g )Nrm   rl   rK   rL   r   r   r   rn   ro   rp   s       r   &test_roundtrip_single_column_dataframerw      s;    B 456J*%G's+FH&+r   c                      [        / SQ/ SQS.5      n [        U 5      n[        USS9nU n[        R                  " X#5        g N)r(   r)   r(   )r)   r(   r3   r]   col2rK   rL   rv   rp   s       r   test_roundtrip_with_prefixesr|      s:    O_MNJ*%G's+FH&+r   c                      [        / SQ/ SQ/ SQS.5      n [        S/ SQ05      n[        U 5      n[        R                  " X!5        g )Nr	   r   r   r	   r   r	   r   r   r   r   r	   r   rG   rm   rl   r   r   rn   ro   r$   rs   rr   s      r    test_no_prefix_string_cats_basicr      s;    lLQRG"234H'"F&+r   c                      [        / SQ/ SQ/ SQS.5      n [        S/ SQ05      n[        U 5      n[        R                  " X!5        g )N)TFFTFTFF)FFTFrG   rm   rl   r   r   s      r   ,test_no_prefix_string_cats_basic_bool_valuesr      sD    +,,	
G "234H'"F&+r   c                      [        / SQ/ SQ/ SQS.5      n [        S/ SQ05      n[        U 5      n[        R                  " X!5        g )Nr~   r   r   rG   rm   rl   r   r   s      r   2test_no_prefix_string_cats_basic_mixed_bool_valuesr      sA    !<<PG "234H'"F&+r   c                      [        / SQ/ SQ/ SQ/ SQS.5      n [        S/ SQ05      n[        U 5      n[        R                  " X!5        g )Nr	   r   r   r   r   r   r   r   r   r	   )r	      rB      rm   r   r   s      r   test_no_prefix_int_cats_basicr      sA    l|MG "m,-H'"F&+r   c                      [        / SQ/ SQ/ SQ/ SQS.5      n [        S/ SQ05      n[        U 5      n[        R                  " X!5        g )Nr   r   r   r   )g      ?g      9@      @g\(\@rm   r   r   s      r   test_no_prefix_float_cats_basicr   	  sB    ,\VG "456H'"F&+r   c                      [        / SQ/ SQ/ SQ/ SQ/ SQS.5      n [        S/ SQ0SS	9n[        U 5      n[        R                  " X!5        g )
N)r	   r   r   r   r   )r   r	   r   r   r   )r   r   r	   r   r   )r   r   r   r	   r   )r   r   r   r   r	   )gGz?r3   rB   FNrm   rg   dtyper   r   s      r   test_no_prefix_mixed_cats_basicr     sL    ! "!	
G "9:(KH'"F&+r   c                      [        / SQ/ SQ/ SQS.5      n [        S/ SQ05      n[        U 5      n[        R                  " X!5        g )Nr   r
   r   r(   r)   NaNrm   r   r   s      r   :test_no_prefix_string_cats_contains_get_dummies_NaN_columnr   !  s;    ii	JKG"/01H'"F&+r   zdefault_category, expectedr3   rm   rG   zdefault_category is a str)idr	   )r(   r)   r	   zdefault_category is a int      ?)r(   r)   r   zdefault_category is a float)r(   r)   r   zdefault_category is a 0F)r(   r)   Fzdefault_category is a boolr	   rB   )r(   r)   r   zdefault_category is a tuplec                     [        / SQ/ SQS.5      n[        X0S9nU(       a  US   R                  S5      US'   [        R                  " XA5        g )Nr   r
   r'   r5   rm   zstring[pyarrow_numpy])r   r   rb   rn   ro   )r6   rs   using_infer_stringr$   rr   s        r   +test_no_prefix_string_cats_default_categoryr   (  sH    L ii89G'EF|**+BC&+r   c                 h    [        / SQ/ SQS.5      n[        U SS9n[        R                  " X!5        g ry   r   )r   rs   rr   s      r   test_with_prefix_basicr   U  s+    /?KLH-S1F&+r   c            
          [        / SQ/ SQ/ SQ/ SQ/ SQ/ SQ/ SQS.5      n [        / SQ/ SQS.5      n[        U S	S
9n[        R                  " X!5        g )Nr   r
   r   r   )r   r   col1_NaNr   r   r   col2_NaNr   )r   r(   r3   rz   rK   rL   r   r   s      r   0test_with_prefix_contains_get_dummies_NaN_columnr   [  sS    !!	

G "3=NOPH's+F&+r   rW   )r(   r)   rW   )rW   r(   r3   rz   )r   r(   r3   )Fr(   r3   zdefault_category is a Falser   )r{   r]   )r(   r)   r   )r	   r(   r3   z4default_category is a dict with int and float values)Nr(   r3   z4default_category is a dict with bool and None valuesr(   r)   )r   r(   r3   z5default_category is a dict with list and tuple valuesc                 F    [        U SUS9n[        R                  " X25        g )NrK   rY   )r   rn   ro   )r   r6   rs   rr   s       r   !test_with_prefix_default_categoryr   l  s'    L S;KF &+r   c                      [        / SQ/ SQ/ SQS.5      n U R                  R                  S5      U l        [        U 5      n[        S[	        [        S5      SS905      n[        R                  " X5        g )	Nr~   r   r   rG   string[python]rm   abcar   r   columnsrb   r   r   listrn   ro   dfrr   rs   s      r   test_ea_categoriesr     s]    	L|L	MB""#34BJ"F"fT&\9IJKLH&+r   c                  Z   [        / SQ/ SQ/ SQ/ SQ/ SQS.5      n U R                  R                  S5      U l        [        U SS9n[        [	        [        S	5      SS
9[	        [        S5      SS
9S.5      nUR                  R                  S5      Ul        [        R                  " X5        g )Nr   r
   r   r   r   r   rK   rL   abar   bacrz   r   r   s      r   test_ea_categories_with_sepr     s    		

B ""#34BJ"#&F4;.>?4;.>?	
H  ''../?@H&+r   c                      [        / SQ/ SQ/ SQS.[        S5      S9n [        U 5      n[        S[        S5      0[        S5      S9n[        R                  " X5        g )	Nr~   r   r   rG   abcd)indexrm   r   )r   r   r   rn   ro   r   s      r   test_maintain_original_indexr     sO    	LAf
B "F"d6l+4<@H&+r   )0numpyr<   r    pandasr   r   r   r   pandas._testing_testingrn   fixturer   r   r%   r-   r8   r:   r>   rI   rN   rQ   rU   rZ   r^   rd   rh   rj   rt   rw   r|   r   r   r   r   r   r   r   markparametrizeparamr   r   r   r   r   r   r   r   r   r   <module>r      s       	 	 	 			;	' /7T
W--'(,,,,
,,,,,,  r?+,*	

 	r=)**	

 	r+,-,	

 	r=)*(	

 	r,-.+	

 	r-./,	
5"F,G"F,,,"  HI*	

 	}mDE(	

 	0:KLM,	

 	$FGE	

 	5)0:JKLE	

 	dE]3S4-8BTUVF	
5"F,G"F,,,.,r   