
    g                        d dl Zd dlZd dlZd dlmc mZ  ej                  ej                        d        Z ej                  ej                        d        Z ej                  ej                        d        Zej                   j"                   ej                  ej$                        d               Zej                   j"                   ej                   ej(                  ej*                  d            d               Z ej                  ej.                        d	        Z ej                  ej2                        d
        Zej                   j"                   ej                  ej6                        d               Zy)    Nc                 P    t        | t        j                  j                        sJ y N)
isinstancepalibDataType)tys    R/var/www/openai/venv/lib/python3.12/site-packages/pyarrow/tests/test_strategies.py
test_typesr      s    b"&&//***    c                 P    t        | t        j                  j                        sJ y r   )r   r   r   Field)fields    r
   test_fieldsr      s    eRVV\\***r   c                 P    t        | t        j                  j                        sJ y r   )r   r   r   Schema)schemas    r
   test_schemasr   $   s    fbffmm,,,r   c                 P    t        | t        j                  j                        sJ y r   )r   r   r   Arrayarrays    r
   test_arraysr   )        eRVV\\***r   F)nullablec                 &    | j                   dk(  sJ y )Nr   )
null_countr   s    r
   test_array_nullabilityr   /   s     q   r   c                 P    t        | t        j                  j                        sJ y r   )r   r   r   ChunkedArray)chunked_arrays    r
   test_chunked_arraysr"   5   s    mRVV%8%8999r   c                 P    t        | t        j                  j                        sJ y r   )r   r   r   RecordBatch)record_baths    r
   test_record_batchesr&   :   s    k266#5#5666r   c                 P    t        | t        j                  j                        sJ y r   )r   r   r   Table)tables    r
   test_tablesr*   ?   r   r   )
hypothesishpytestpyarrowr   pyarrow.tests.strategiestests
strategiespastgiven	all_typesr   
all_fieldsr   all_schemasr   marknumpy
all_arraysr   arraysprimitive_typesr   all_chunked_arraysr"   all_record_batchesr&   
all_tablesr*    r   r
   <module>r@      sy  $    ' ' 	+ + 	+ + 			- - +  + T))E	:;! < ! 		 	 !: ": 		 	 !7 "7 +  +r   