
    Mh                         S SK rS SKrS SKJrJrJrJrJr  S SK	J
r  \R                  " S5         " S S5      r " S S5      rg)    N)Categorical	DataFrame
MultiIndexSeries
date_rangexarrayc                   J    \ rS rSr\R
                  S 5       rS rS rS r	Sr
g)TestDataFrameToXArray   c                    [        [        S5      [        [        SS5      5      [        R                  " SS5      R                  S5      [        R                  " SSS	S
9/ SQ[        [        S5      5      [        SSS9[        SSSS9S.5      $ )Nabcd            u1g      @g       @float64dtype)TFTF20130101   )periodsz
US/Eastern)r   tz)abcdefgh)r   listrangenparangeastyper   r   )selfs    U/var/www/html/env/lib/python3.13/site-packages/pandas/tests/generic/test_to_xarray.pydfTestDataFrameToXArray.df   sv    &\%1+&YYq!_++D1YYsCy9/ f.
A6
A,G	
 	
    c                    Un[        U5      S:X  a  [        R                  " S5        SSKJn  US S Ul        SUR
                  l        SUR                  l        UR                  5       nUR                  S   S:X  d   e[        UR                  5      S:X  d   e[        UR                  5      S:X  d   e[        R                  " [        UR                  R                  5       5      S/5        [!        Xe5      (       d   eUR#                  5       nUS	   R%                  U(       d  [&        OS
5      US	'   S UR                  l        [        R(                  " UR+                  5       U5        g )Nr   z'Test doesn't make sense for empty indexDatasetr   foobarr      r   string[pyarrow_numpy])lenpytestskipr   r.   indexnamecolumns	to_xarraysizescoords	data_varstmassert_almost_equalr"   keys
isinstancecopyr&   objectassert_frame_equalto_dataframe)r'   
index_flatr)   using_infer_stringr6   r.   resultexpecteds           r(   test_to_xarray_index_types0TestDataFrameToXArray.test_to_xarray_index_types    s"   u:?KKAB"!9

||E"a'''6==!Q&&&6##$)))
tFMM$6$6$89E7C&****
 779 ,,,F2I
 !%
f113X>r+   c                     SSK Jn  SUR                  l        USS R	                  5       nUR
                  S   S:X  d   e[        X25      (       d   eg )Nr   r-   r/   )r   r.   r6   r7   r9   r:   r@   )r'   r)   r.   rG   s       r(   test_to_xarray_empty*TestDataFrameToXArray.test_to_xarray_empty<   sL    "Aa""$||E"a'''&****r+   c                    SSK Jn  [        R                  " S/[	        S5      /SS/S9Ul        UR                  5       nUR                  S   S:X  d   eUR                  S   S:X  d   e[        UR                  5      S	:X  d   e[        UR                  5      S
:X  d   e[        R                  " [        UR                  R                  5       5      SS/5        [        XC5      (       d   eUR!                  5       nUR#                  5       nUS   R%                  U(       d  [&        OS5      US'   S UR(                  l        [        R,                  " XE5        g )Nr   r-   r   r   onetwonamesr      r1   r   r2   )r   r.   r   from_productr#   r6   r9   r:   r3   r;   r<   r=   r>   r"   r?   r@   rD   rA   r&   rB   r8   r7   rC   )r'   r)   rF   r.   rG   rH   s         r(   test_to_xarray_with_multiindex4TestDataFrameToXArray.test_to_xarray_with_multiindexD   s!   " **SE58+<UENS||E"a'''||E"a'''6==!Q&&&6##$)))
tFMM$6$6$89E5>J&****$$&779 ,,,F2I
 !%
f/r+    N)__name__
__module____qualname____firstlineno__r4   fixturer)   rI   rL   rU   __static_attributes__rW   r+   r(   r
   r
      s'    ^^
 
?8+0r+   r
   c                   &    \ rS rSrS rS rS rSrg)TestSeriesToXArrayZ   c                    UnSSK Jn  [        [        [	        U5      5      USS9nSUR
                  l        UR                  5       n[        U5        [	        U5      [	        U5      :X  d   e[	        UR                  5      S:X  d   e[        R                  " [        UR                  R                  5       5      S/5        [        XS5      (       d   e[        R                  " UR!                  5       U5        g )Nr   	DataArrayint64)r6   r   r/   r   )r   rc   r   r#   r3   r6   r7   r9   reprr;   r=   r>   r"   r?   r@   assert_series_equal	to_series)r'   rE   r6   rc   serrG   s         r(   rI   -TestSeriesToXArray.test_to_xarray_index_types[   s     	%U3u:&e7C		V6{c%j(((6==!Q&&&
tFMM$6$6$89E7C&,,,, 	v//137r+   c                 \   SSK Jn  [        / [        S9nSUR                  l        UR                  5       n[        U5      S:X  d   e[        UR                  5      S:X  d   e[        R                  " [        UR                  R                  5       5      S/5        [        X15      (       d   eg )Nr   rb   r   r/   r   )r   rc   r   rB   r6   r7   r9   r3   r;   r=   r>   r"   r?   r@   )r'   rc   rh   rG   s       r(   rL   'TestSeriesToXArray.test_to_xarray_emptym   s    $Rv&		6{a6==!Q&&&
tFMM$6$6$89E7C&,,,,r+   c                    SSK Jn  [        R                  " SS/[	        S5      /SS/S9n[        [	        S	5      S
US9nUR                  5       n[        U5      S:X  d   e[        R                  " [        UR                  R                  5       5      SS/5        [        XA5      (       d   eUR                  5       n[        R                  " XS5        g )Nr   rb   r   r   r   rO   rP   rQ      rd   )r   r6   rS   )r   rc   r   rT   r#   r   r9   r3   r=   r>   r"   r;   r?   r@   rg   rf   )r'   rc   mirh   rG   ress         r(   rU   1TestSeriesToXArray.test_to_xarray_with_multiindexx   s    $$$sCj%(%;E5>RU1XWB76{a
tFMM$6$6$89E5>J&,,,, 
s(r+   rW   N)rX   rY   rZ   r[   rI   rL   rU   r]   rW   r+   r(   r_   r_   Z   s    8$	-
)r+   r_   )numpyr$   r4   pandasr   r   r   r   r   pandas._testing_testingr=   importorskipr
   r_   rW   r+   r(   <module>rv      s?          H G0 G0T() ()r+   