
    bMh                        d dl Zd dlZd dlZd dlmZ d Zd Z	d Z
d Zd Zd Zd Zej                            d	 ej        g d
           ej         ej        ddd                    g          d             Zd Zd Zd Zd Zd Zd Zej                            dddg          d             Zej                            dddg          d             ZdS )    Nc            	      J   t          j        g dt          j        g dgt	          d          d          } |                                 }t          j        dddt          j        t          j        d	d
gt	          d          t          d          }t          j        ||           d S )Nr               abcdfoo)indexnamer   r   r   r   r	   aaabcddr   dtyper   )	pdSeriesnpnanlistexplodeobjecttmassert_series_equalsresultexpecteds      h/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/series/methods/test_explode.py
test_basicr      s    
	999bfb&1fERRRAYY[[Fy	
Aq"&"&!Q'tIfSX  H 68,,,,,    c            
      h   t          j        g dt          j        d t          j        g           t          j        ddg          gd          } |                                 }t          j        dddt          j        d t          j        ddgg d	t          d
          }t          j        ||           d S )Nr   abr   )r   r   r   r   )r   r   r   r   r   r   r	   r	   r   )	r   r   r   r   arrayr   r   r   r   r   s      r   test_mixed_typer%      s    
		BFD"(2,,	3*0E0EFU	 	 	A YY[[Fy	
Aq"&$S1&&&	  H 68,,,,,r    c                      t          j        t                    } |                                 }|                                 }t          j        ||           d S )Nr   )r   r   r   r   copyr   r   r   s      r   
test_emptyr)      sG    
	AYY[[FvvxxH68,,,,,r    c                      t          j        g dgddgdg          } |                                 }t          j        g ddddgg d          }t          j        ||           d S )Nr   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   r   s      r   test_nested_listsr.   &   sm    
	III;A*++AYY[[Fy)))Q1-\\\BBBH68,,,,,r    c            
         t          j        g dt          j        g dgdt           j                            t          d          t          d          gddg                    } |                                 }t           j        	                    g d	ddg          }t          j        d
ddt          j        t          j        ddg|t          d          }t          j        ||           d S )Nr   r   r   abr   bar)names)r   r   )r"   r   r3   r3   )r"   r   )r#   r   r#   r   r4   r   r   r   r	   r   )r   r   r   r   
MultiIndexfrom_productr   ranger   from_tuplesr   r   r   )r   r   r   r   s       r   test_multi_indexr9   -   s    
		BFB'm(($t**eAhh)?u~(VV	 	 	A
 YY[[FM%%NNNen &  E y	
Aq"&"&!Q'uF  H 68,,,,,r    c                      t          j        t          d          g                                          } |                                 }t	          j        ||            d S )N   )r   r   r7   r   r   r   r   r   s     r   
test_larger=   >   sK    
	5::,''))AYY[[F61%%%%%r    c                     t          j        dt          j        ddd          i          } |                     d d          }|                                }t          j        || d                                                    d S )	Nr"   20190101r   UTCperiodstzc                     | j         S N)r$   )xs    r   <lambda>z#test_invert_array.<locals>.<lambda>G   s     r    r   )axis)r   	DataFrame
date_rangeapplyr   r   r   rename)dflistifyr   s      r   test_invert_arrayrO   D   sv    	sBM*aEJJJK	L	LBhh((qh11G__F62c7>>#3#344444r    r   r+   2019r   r@   rA   c                 X    |                                  }t          j        ||            d S rE   )r   r   r   r<   s     r   test_non_object_dtyperR   L   s*     YY[[F61%%%%%r    c                  d   t          j        ddddddgddg          } | j        j                            d	                                          }| dg                             |          }t          j        g d
t          d          dddgg d          }t          j	        ||           d S )Nza,b,cr   )var1var2zd,e,fr   rT   rU   )columns,)r   r   r   r   r   r   abcdef)rU   rT   r   r   r   r   r   r   )rV   r   )
r   rI   rT   strsplitr   joinr   r   assert_frame_equal)rM   explodedr   r   s       r   test_typical_usecaser_   T   s    	1	%	%'C'CD 
 
 
B w{  %%--//H\x((F|###T(^^<<      H
 &(+++++r    c                  2   t          j        t          j        ddd          t          j        ddd          g          } |                                 }t          j        t          j        ddd          g d          }t	          j        ||           d S )	N20170101r   r@   rA   20170104   rY   r,   )r   r   rJ   r   r   r   r   s      r   test_nested_EArd   c   s    
	M*aE:::M*aE:::	
	 	A YY[[Fy
j!666>P>P>P  H 68,,,,,r    c                      t          j        ddgddggddg          } |                                 }t          j        g dg dt          	          }t	          j        ||           d S )
Nr   r   r   r	   r   r,   r   r   r   r	   )r   r   r   r   r   r   r   r   r   r   r   r   r   s      r   test_duplicate_indexri   r   sk    
	Aq6Aq6"1a&111AYY[[Fy\\\HHHH68,,,,,r    c                      t          j        ddgddgg          } |                     d          }t          j        g dg dt          	          }t	          j        ||           d S )
Nr   r   r   r	   Tignore_indexrf   )r   r   r   r   rg   rh   r   s      r   test_ignore_indexrm   z   sh    
	Aq6Aq6"##AYYDY))Fy\\\HHHH68,,,,,r    c                      t          j        h dgdg          } |                                                                 }t          j        g dg d          }t	          j        ||           d S )N>   r"   r#   cr   r,   r"   r#   ro   )r   r   r   )r   r   r   sort_valuesr   r   r   s      r   test_explode_setsrr      sm    
	???#A3///AYY[[$$&&Fy			:::H68,,,,,r    c                      t          j        g dg d          } |                     d          }t          j        g d          }t          j        ||           d S )Nr+   rp   r,   Trk   r-   r   s      r   %test_explode_scalars_can_ignore_indexrt      s]    
	)))???333AYYDY))Fy##H68,,,,,r    rl   TFc           	         t          j        d          }d d gdgg ddgd g}t          j        |t          j        |                    |                                                              }|                    |           }t          j        g d| rd ng dt          j        |                                          	          }t          j	        ||           d S )
Npyarrowr   r   r   r'   rk   )NNr   Nr   r   N)r   r   r   r   r   r   r	   )datar   r   )
pytestimportorskipr   r   
ArrowDtypelist_int64r   r   r   rl   parw   serr   r   s         r   test_explode_pyarrow_list_typer      s     
	Y	'	'B 
t	

	
AD )Dbhhrxxzz.B.B C C
D
D
DC[[l[33Fy..."=dd(=(=(=mBHHJJ''  H
 68,,,,,r    c                 6   t          j        d          }g d}t          j        |t          j        |                                                    }|                    |           }t          j        g ddg d          }t          j        ||           d S )Nrv   r+   r'   rk   zint64[pyarrow]r   )r   r   )	rx   ry   r   r   rz   r|   r   r   r   r}   s         r   "test_explode_pyarrow_non_list_typer      s    		Y	'	'B99D
)Dbhhjj 9 9
:
:
:C[[l[33Fy*:)))LLLH68,,,,,r    )numpyr   rx   pandasr   pandas._testing_testingr   r   r%   r)   r.   r9   r=   rO   markparametrizer   rJ   rR   r_   rd   ri   rm   rr   rt   r   r    r    r   <module>r      s                 - - -- - -- - -- - -- - -"& & &5 5 5 )")III

		-"-e*T*T*T U U	V & & &
, , ,- - -- - -- - -- - -- - - $77- - 87-* $77- - 87- - -r    