
    m*i                         S SK rS SKrS SKJrJrJrJrJrJ	r	  S SK
Jr  S SKJr  \R                  " S5      r " S S5      r " S S5      rg)	    N)Categorical	DataFrame
MultiIndexSeriesStringDtype
date_range)Version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    c/var/www/html/land-doc-ocr/venv/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!   str)lenpytestskipr
   r0   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_stringr8   r0   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 ,,,F%
 !%
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/   r1   )r
   r0   r8   r9   r;   r<   rB   )r)   r+   r0   rI   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      r3   r!   r4   )r
   r0   r   from_productr%   r8   r;   r<   r5   r=   r>   r?   r@   r$   rA   rB   rF   rC   r(   rD   r:   r9   rE   )r)   r+   rH   r0   rI   rJ   s         r*   test_to_xarray_with_multiindex4TestDataFrameToXArray.test_to_xarray_with_multiindexF   s    " **SE58+<UENS||E"a'''||E"a'''6==!Q&&&6##$)))
tFMM$6$6$89E5>J&****$$&779 ,,,F%
 !%
f/r-    N)__name__
__module____qualname____firstlineno__r6   fixturer+   rK   rN   rW   __static_attributes__rY   r-   r*   r   r      s'    ^^
 
?8+0r-   r   c                   &    \ rS rSrS rS rS rSrg)TestSeriesToXArray\   c                 \   Un[        UR                  [        5      (       a  UR                  R                  S:X  a}  [	        [
        R                  5      [	        S5      :  aW  [	        [
        R                  5      [	        S5      :  a1  UR                  [        R                  R                  S[        S95        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(                  R1                  5       5      S
/5        [        Xd5      (       d   e[*        R2                  " UR5                  5       U5        g )Npyarrowz2024.9.0z2025.6.0z-xarray calling reshape of ArrowExtensionArray)reasonraisesr   	DataArrayint64)r8   r   r1   r   )rB   r   r   storager	   r
   __version__applymarkerr6   markxfailNotImplementedErrorrh   r   r%   r5   r8   r9   r;   reprr=   r?   r@   r$   rA   assert_series_equal	to_series)r)   rG   requestr8   rh   serrI   s          r*   rK   -TestSeriesToXArray.test_to_xarray_index_types]   s;   u{{K00##y0**+gj.AA**+gj.AA!!J. "  	%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   rg   r   r1   r   )r
   rh   r   rD   r8   r9   r;   r5   r=   r?   r@   r$   rA   rB   )r)   rh   rt   rI   s       r*   rN   'TestSeriesToXArray.test_to_xarray_empty{   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   rg   r   r   r   rQ   rR   rS      ri   )r   r8   rU   )r
   rh   r   rV   r%   r   r;   r5   r?   r@   r$   r=   rA   rB   rr   rq   )r)   rh   mirt   rI   ress         r*   rW   1TestSeriesToXArray.test_to_xarray_with_multiindex   s    $$$sCj%(%;E5>RU1XWB76{a
tFMM$6$6$89E5>J&,,,, 
s(r-   rY   N)rZ   r[   r\   r]   rK   rN   rW   r_   rY   r-   r*   ra   ra   \   s    8<	-
)r-   ra   )numpyr&   r6   pandasr   r   r   r   r   r   pandas._testing_testingr?   pandas.util.versionr	   importorskipr
   r   ra   rY   r-   r*   <module>r      sD        '			X	&G0 G0T4) 4)r-   