
    Pi                         d dl Z d dlZd dlmc mZ d dlZd dlm	Z
  ej        dd          Zd Zd Zd Z G d d	          Z G d
 d          Z ej        dd          d             ZdS )    Npyarrowz16.0)
minversionc                  f   t          j        g d          } |                                 }t          j                            t          j        |          d          dk    sJ t                              |           }t                              g dg          }|	                    |          sJ t                              | t          
                                          }t                              g dgt          
                                          }|	                    |          sJ d S )N            arrow_array_streamr   typepdSeries__arrow_c_stream__ctypes	pythonapiPyCapsule_IsValid	py_objectpachunked_arrayequalsint32scapsulecaexpecteds       |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/series/test_arrow_interface.pytest_series_arrow_interfacer      s   
	)))A""$$G**W%%'<	
 	
 	 	 	 	 
		!		B,,H99X			!"((**		-	-B"((**==H99X    c                  j   t          j        g dd          } |                                 }t          j                            t          j        |          d          dk    sJ t                              |           }t                              g dg          }|	                    |          sJ t                              | t          
                                          }t                              g dgt          
                                          }|	                    |          sJ d S )Nr   zInt64[pyarrow]dtyper
   r   r   r   r   s       r   (test_series_arrow_interface_arrow_dtypesr$       s   
	)))#3444A""$$G**W%%'<	
 	
 	 	 	 	 
		!		B,,H99X			!"((**		-	-B"((**==H99Xr    c                     t          j        ddgd          } |                                 }t          j                            t          j        |          d          dk    sJ t                              |           }t                              ddggt          	                                          }|
                    |          sJ d S )Nfoobarzstring[pyarrow]r"   r
   r   r   )r   r   r   r   r   r   r   r   r   large_stringr   r   s       r   'test_series_arrow_interface_stringdtyper)   3   s    
	5%.(9:::A""$$G**W%%'<	
 	
 	 	 	 	 
		!		B% 0r7H7HIIH99Xr    c                       e Zd Zd ZddZdS )ArrowArrayWrapperc                     || _         d S N)array)selfr.   s     r   __init__zArrowArrayWrapper.__init__D   s    


r    Nc                 6    | j                             |          S r-   )r.   __arrow_c_array__r/   requested_schemas     r   r2   z#ArrowArrayWrapper.__arrow_c_array__G   s    z++,<===r    r-   )__name__
__module____qualname__r0   r2    r    r   r+   r+   C   s7          > > > > > >r    r+   c                       e Zd Zd ZddZdS )ArrowStreamWrapperc                     || _         d S r-   )stream)r/   r   s     r   r0   zArrowStreamWrapper.__init__L   s    #r    Nc                 6    | j                             |          S r-   )r<   r   r3   s     r   r   z%ArrowStreamWrapper.__arrow_c_stream__O   s    {--.>???r    r-   )r5   r6   r7   r0   r   r8   r    r   r:   r:   K   s=        $ $ $@ @ @ @ @ @r    r:   z14.0)min_versionc                  >   t                               g dddgg          } t          j                            |           }t          j        g d          }t          j        ||           t          j                            t          |                     }t          j        ||           t                               g dg dd          }t          j                            |          }t          j        dddd	d
ddddg          }t          j        ||           t           	                    g d          } t          j        g d          }t          j                            |           }t          j        ||           t          j                            t          |                     }t          j        ||           t          j        t          d          5  t          j                            g d           d d d            d S # 1 swxY w Y   d S )N)r   r	      r      )r   r	   r@   r   rA   )abc)rB   rC   r   rB   r	   rC   r@   rD   z.Expected an Arrow-compatible array-like object)match)r   r   r   r   
from_arrowtmassert_series_equalr:   tabler.   r+   pytestraises	TypeError)arrresultr   rI   s       r   test_dataframe_from_arrowrO   S   s.    

III1v.
/
/CY!!#&&Fy))H68,,, Y!!"4S"9"9::F68,,, HH999???;;<<EY!!%((Fy,,AC.@.@PSBTBTUVVH68,,, ((999

Cy##HY!!#&&F68,,,Y!!"3C"8"899F68,,, 
I
 
 
 ( ( 		YYY'''( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (s   #"HHH)r   rJ   pandas.util._test_decoratorsutil_test_decoratorstdpandasr   pandas._testing_testingrG   importorskipr   r   r$   r)   r+   r:   
skip_if_norO   r8   r    r   <module>rY      s#     ) ) ) ) ) ) ) ) )          Vv666  &  &   > > > > > > > >@ @ @ @ @ @ @ @ yf---!( !( .-!( !( !(r    