
    jp                        d dl Zd dlm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                   ej,                  ddg       ej.                               d        Zej$                  j&                   ej                   e
j2                  e
j4                  d	
            d               Z ej                   e
j8                  e
j4                              d        Z ej                  e
j<                        d        Zej$                  j&                   ej                  e
j@                        d               Z!y)    Nc                 P    t        | t        j                  j                        sJ y N)
isinstancepalibDataType)tys    D/tmp/pip-target-jj7clzyk/lib/python/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   z+01:30z-00:30c           
          |j                  t        j                  t        j                  t        j                  d|             d            }t        |t
        j                  j                        sJ |j                  j                  | k(  sJ y )Ns   )size)drawpastarraysstjustr   	timestampr   r   r   typetz)timezonedatar   s      r
   +test_timestamp_array_fixed_offset_timezonesr)   0   sa    IIBGGBLLh78qACEeRVV\\***::==H$$$r   F)nullablec                 &    | j                   dk(  sJ y )Nr   )
null_countr   s    r
   test_array_nullabilityr-   8   s     q   r   c                 P    t        | t        j                  j                        sJ y r   )r   r   r   ChunkedArray)chunked_arrays    r
   test_chunked_arraysr1   >   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_batchesr5   C   s    k266#5#5666r   c                 P    t        | t        j                  j                        sJ y r   )r   r   r   Table)tables    r
   test_tablesr9   H   r   r   )"
hypothesishhypothesis.strategies
strategiesr"   pytestpyarrowr   pyarrow.tests.strategiestestsr    given	all_typesr   
all_fieldsr   all_schemasr   marknumpy
all_arraysr   sampled_fromr(   r)   r!   primitive_typesr-   chunked_arraysr1   all_record_batchesr5   
all_tablesr9    r   r
   <module>rO      s  $  "   ' ' 	+ + 	+ + 			- - +  + 	(H-	.	:% ;% T))E	:;! < ! 				T11	23: 4: 		 	 !7 "7 +  +r   