
    chk9                        S SK r S SKrS SKrS SKrS SKJr   S SKJ	s  J
r   S SKJ	s  Jr   S SKr\R"                  S:X  a   S SKr S SK
rS SKr\R,                  " \R.                  " SSS9S9r\R2                  " \R4                  " 5       5      r\R2                  " \R8                  " 5       5      r\R2                  " \R<                  " 5       5      r\R2                  " \R@                  " 5       5      r!\R2                  " \RD                  " 5       5      r#\R2                  " \RH                  " 5       5      r%\RL                  " \R<                  \RN                  " S SS95      r(\RR                  " \\!\#\%\(5      r*\RV                  " \RX                  " 5       \RZ                  " 5       \R\                  " 5       \R^                  " 5       /5      r0\RV                  " \Rb                  " 5       \Rd                  " 5       \Rf                  " 5       \Rh                  " 5       /5      r5\RR                  " \0\55      r6\RV                  " \Rn                  " 5       \Rp                  " 5       \Rr                  " 5       /5      r:\RL                  " \Rv                  \RN                  " S	S
S9\RN                  " S	S
S9S9r<\RL                  " \Rz                  \RN                  " S	SS9\RN                  " S	SS9S9r>\RL                  " \R~                  \RN                  " S	SS9\RN                  " S	SS9S9r@\RL                  " \R                  \RN                  " S	SS9\RN                  " S	SS9S9rB\RR                  " \6\:\@\B5      rC\RV                  " \R                  " 5       \R                  " 5       /5      rF\RV                  " \R                  " S5      \R                  " S5      \R                  " S5      \R                  " S5      /5      rI\(       aI  \(       aB  \RR                  " \R                  " 5       \R                  " 5       \R                  " 5       5      rKO\(       a2  \RR                  " \R                  " 5       \R                  " 5       5      rKOJ\(       a2  \RR                  " \R                  " 5       \R                  " 5       5      rKO\R                  " 5       rK\RL                  " \R                  \RV                  " / SQ5      \KS9rM\RL                  " \R                  \RV                  " / SQ5      5      rO\R2                  " \R                  " 5       5      rQ\RR                  " \F\I\M\O\Q5      rR\RR                  " \\\C\R\*5      rS\R                  " \R,                  " 5       \R,                  " 5       5      rU\R                  \S4S j5       rW\S4S jrX\R                  \S4S j5       rYS%S jrZ\R                  \S\S4S j5       r[\SS4S jr\\R                  " S 5      r^\W" \^5      r_\\" \^5      r`\RN                  " S SS9ra\R                  S&S j5       rb\R                  S&S j5       rc\R                  S'S j5       rd\R                  S(S  j5       re\R                  S%S! j5       rf\R                  S%S" j5       rg\d" \^5      rh\e" \^5      ri\f" \^5      rj\g" \^5      rk\RR                  " \\\6\RV                  " \Rp                  " 5       \Rr                  " 5       /5      \@\F\I\Q\\!\#\%5      rl\RR                  " \\6\\!\(5      rm\l4S# jrn\R                  " S$ 5      rog! \ a    Sr GN&f = f! \ a    Sr GN+f = f! \ a    Sr GN5f = f! \ a    Sr GN/f = f! \ a    Sr GN9f = f))    Nwin32A   ~   )min_codepointmax_codepoint)alphabet   	min_value	max_value   	   )	precisionscale   &   L   smsusns)r   r   r   r   )unittzc                     U " [         5      nU " U5      n[        R                  R                  U5      (       a  SnOU " [        R
                  " 5       5      nU " [        5      n[        R                  " X#XES9$ )NT)typenullablemetadata)custom_textpatypesis_nullstbooleansr   field)drawtype_strategynametypr   metas         kC:\Users\julio\OneDrive\Documentos\Trabajo\Ideas Frescas\venv\Lib\site-packages\pyarrow/tests/strategies.pyfieldsr+      sZ    D
}
C	xx&>D88DXEE    c           
         [         R                  " [        R                  U 5      [         R                  " [        R                  U 5      -  [         R                  " [        R                  U [         R
                  " SSS95      -  [         R                  " [        R                  U 5      -  [         R                  " [        R                  U 5      -  $ )Nr   r	   r
   )r"   buildsr   list_
large_listintegers	list_viewlarge_list_viewitem_strategys    r*   
list_typesr6      s    
		"((M*
		"--/	0
		HHKK!r2	
	
 			",,.	/ 			"$$m4	5
r,   c                 $   [         R                  " [        U5      5      nU " U5      nU Vs/ s H  oDR                  PM     nn[        R
                  " [        [        U5      5      [        U5      :H  5        [        R                  " U5      $ s  snf N)
r"   listsr+   r'   hassumelensetr   struct)r%   r5   fields_strategyfields_renderedr$   field_namess         r*   struct_typesrB      si    hhvm45O?+O+:;?%::?K;HHS[!"c+&66799_%% <s   Bc           	      :   U c  [         n Uck  [        R                  " [        [        [        R
                  " [        R                  " 5       [        R                  " 5       /5      [        [        [        5      n[        R                  " [        R                  X5      $ r8   )signed_integer_typesr"   one_of	bool_typeinteger_typessampled_fromr   float32float64binary_typestring_typefixed_size_binary_typer.   
dictionary)key_strategyvalue_strategys     r*   dictionary_typesrQ      sd    +OORZZ\2::<89"
 99R]]LAAr,   c                     U " U5      n[         R                  " [        R                  R	                  U5      (       + 5        U " U5      n[        R
                  " X45      $ r8   )r:   r;   r   r    r!   map_)r%   rO   r5   key_type
value_types        r*   	map_typesrV      sE     L!HHH!!(++,m$J778((r,   c                     [         R                  " [        U 5      US9n[         R                  " [        R
                  U5      $ )N)max_size)r"   r9   r+   r.   r   schema)r&   
max_fieldschildrens      r*   schemasr\      s,    xx}-
CH99RYY))r,   c                      [         [        5       -  [        5       -  [        5       -  [	        5       -  [        [
        5      -  [        [
        5      -  $ r8   )primitive_typesr6   rB   rQ   rV   	all_types r,   r*   <lambda>ra      sP    		 		 			
 	9	 	Y	 r,      c                 B    U " [        XSS95      nUR                  5       $ NFsizer   )arrays	to_pylist)r%   rU   rf   r   arrs        r*   _pylistrj      s    
vje<
=C==?r,   c                 z    U " U5      nU " [        XSS95      nU " [        X%US95      n[        [        Xg5      5      $ rd   )rj   listzip)r%   rT   rU   rf   r   lengthkeysvaluess           r*   _pymaprq     s=    $ZF>?D'*HEFFD!""r,   c           
      V   [        U[        R                  5      (       a	  U " U5      nO-[        U[        R                  5      (       a  UnO[        S5      e[        U[        R                  5      (       a	  U " U5      nO0Uc  U " [        5      nO [        U[        5      (       d  [        S5      e[        R                  R                  U5      (       a-  [        R                  " U5        [        R                  " 5       nGO[        R                  R                  U5      (       a  [        R                  " 5       nGO[        R                  R                  U5      (       a>  U " [         R"                  " UR%                  5       U4S95      n[        R&                  " XdS9$ [        R                  R)                  U5      (       aW  U " [         R"                  " UR%                  5       U4S95      nSU[*        R,                  " U5      '   [        R&                  " XdS9$ [        R                  R/                  U5      (       a  [        R0                  " 5         GO[        R                  R3                  U5      (       a  [        R4                  " 5       nGO[        R                  R7                  U5      (       a  [        R8                  " 5       nGOQ[        R                  R;                  U5      (       Ga*  [<        c  [>        R@                  " S5        URB                  c  [>        R@                  " S5        SnS	n[D        RD                  RG                  US
-  5      [D        RH                  " SS9-   n	[D        RD                  RG                  US
-  5      [D        RH                  " SS9-
  n
 URB                  RK                  S5      n[        US   5      n[        US   5      n[D        RH                  " XS9n[        RP                  " [        RR                  " U5      U	U
S9nGO[        R                  RU                  U5      (       a  [        RV                  " 5       nGO[        R                  RY                  U5      (       a  [        RV                  " 5       nGO[        R                  R[                  U5      (       d$  [        R                  R]                  U5      (       a  [        R^                  " 5       nGO-[        R                  Ra                  U5      (       d$  [        R                  Rc                  U5      (       a  [        Rd                  " 5       nGO[        R                  Rg                  U5      (       a+  [        R^                  " URh                  URh                  S9nGO[        R                  Rk                  U5      (       a  [m        URn                  X#S9nGOE[        R                  Rq                  U5      (       a  [m        URn                  X#S9nGO[        R                  Rs                  U5      (       a!  [m        URn                  URt                  US9nGO[        R                  Rw                  U5      (       a  [m        URn                  X#S9nGO[        R                  Ry                  U5      (       a  [m        URn                  X#S9nGOR[        R                  R{                  U5      (       a/  [m        URn                  X#S9n[        R&                  " U " U5      US9$ [        R                  R}                  U5      (       a%  [        UR                  UR                  [        US9nO[        R                  R                  U5      (       a  [        R                  " [        U5      S:  5        / / nnU H=  nUR                  U5        UR                  U " [#        UR                  US95      5        M?     [        R                  R                  UUS9$ [        U5      eU(       a*  [        R                  " [        R                  " 5       W5      n[        R                  " WX"S9n[        R&                  " U " U5      US9$ ! [L         a$    [<        RN                  " URB                  5      n GNf = f)NzType must be a pyarrow DataTypezSize must be an integer)shaper   g      Ez/no module named zoneinfo (or tzdata on Windows)zrequires timezone not Nonel         l    i ʚ;   )hours:r   r   )rv   minutes)	timezonesr   r   min_sizerX   re   rf   )r+   )K
isinstancer"   SearchStrategyr   DataType	TypeError_default_array_sizesintr    r!   r:   r;   none
is_booleanr#   
is_integernpstrg   to_pandas_dtypearrayis_floatingnpisnan
is_decimalrejectis_timetimesis_datedatesis_timestampzoneinfopytestskipr   datetimefromtimestamp	timedeltasplit
ValueErrorZoneInfo	datetimesjustis_duration
timedeltasis_interval	is_binaryis_large_binarybinary	is_stringis_large_stringtextis_fixed_size_binary
byte_widthis_listrj   rU   is_large_listis_fixed_size_list	list_sizeis_list_viewis_large_list_viewis_dictionaryis_maprq   rT   	item_type	is_structr<   appendr   StructArrayfrom_arraysNotImplementedErrorrE   r9   )r%   r   rf   r   tyvaluerp   	min_int64	max_int64min_datetimemax_datetimeoffsetoffset_hours
offset_minr   r+   child_arraysr$   s                     r*   rg   rg     s   $))**$Z	D"++	&	&9::$))**Dz	()c""122	xx					R	 	 			R	 	 dkk""4"4"6tgFGxx((			b	!	!dkk""4"4"6tgFG $)rxx xx((			R	 	  	

			"		
			"		
			r	"	"KKIJ55=KK45		((66 "*"4"42">?((66 "*"4"42">?	*UU[[%Fvay>LVAYJ##,KB rwwr{l'35			b	!	!			b	!	!			B		288#;#;B#?#?					B		288#;#;B#?#?			&	&r	*	*		2==2==I			"		DD				#	#DD		$	$R	(	(BLL8L			r	"	"DD		$	$R	(	(DD				#	#TExxV2..			r{{BLL7K (*			B			R1!2EMM% VEJJT%B CD  ~~)),v)FF!"%%		"'')U+XXed:F88DLr**Y  	*""255)B	*s    Aa: :*b('b(c                 2   [        U[        R                  5      (       a  U " U5      n[        R                  " [
        R                  R                  U5      (       + 5        [        XS9n[        R                  " XRUS9n[
        R                  " U " U5      US9$ )Nr|   rz   rt   )r}   r"   r~   r:   r;   r   r    r   rg   r9   chunked_array)r%   r   
min_chunks
max_chunks
chunk_sizechunkchunkss          r*   chunked_arraysr   p  so    $))**Dz HH##D))*4)EXXe:FFDLt44r,   c           
      f   [        U[        R                  5      (       a	  U " U5      nO0Uc  U " [        5      nO [        U[        5      (       d  [        S5      eU " [        XS95      nU Vs/ s H  oP" [        UR                  US95      PM     nn[        R                  R                  XdS9$ s  snf NzRows must be an integer)rZ   r|   )rY   )r}   r"   r~   r   r   r   r\   rg   r   r   RecordBatchr   r%   r   rowsrZ   rY   r$   r[   s          r*   record_batchesr   ~  s    $))**Dz	()c""122'$67FAGHVEJJT23HH >>%%h%>> I   ,#B.c           
      f   [        U[        R                  5      (       a	  U " U5      nO0Uc  U " [        5      nO [        U[        5      (       d  [        S5      eU " [        XS95      nU Vs/ s H  oP" [        UR                  US95      PM     nn[        R                  R                  XdS9$ s  snf r   )r}   r"   r~   r   r   r   r\   rg   r   r   Tabler   r   s          r*   tablesr     s    $))**Dz	()c""122'$67FAGHVEJJT23HH8888 Ir   c                     [         R                  " [        R                  U 5      [         R                  " [        R                  U 5      -  $ r8   )r"   r.   r   r/   r0   r4   s    r*   pandas_compatible_list_typesr     s2     			"((M*
		"--/	0r,   c            
          [         R                  " [        [        [        5      [	        [        5      [        [        S9[        [        5      [	        [        5      5      $ )N)rP   )r"   rE   !pandas_compatible_primitive_typesr   rB   rQ   (pandas_compatible_dictionary_value_typespandas_compatible_typesr`   r,   r*   ra   ra     sA    BII)$%FG67C	
 	%%<=,-	r,   )NN)T)NT)r   NN)pr   sysr   
hypothesisr:   hypothesis.strategies
strategiesr"   hypothesis.extra.numpyextranumpyr   ImportErrorhypothesis.extra.pytzpytztzstr   platformtzdatar   pyarrowr   r   
charactersr   r   null	null_typebool_rF   r   rK   stringrL   large_binarylarge_binary_typelarge_stringlarge_string_typer.   r1   rM   rE   binary_like_typesrH   int8int16int32int64rD   uint8uint16uint32uint64unsigned_integer_typesrG   float16rI   rJ   floating_types	decimal32decimal32_type	decimal64decimal64_type
decimal128decimal128_type
decimal256decimal256_typenumeric_typesdate32date64
date_typestime32time64
time_typesr   ry   	timestamptimestamp_typesdurationduration_typesmonth_day_nano_intervalinterval_typestemporal_typesr^   dictionariesr   	compositer+   r6   rB   rQ   rV   r\   deferredr_   
all_fieldsall_schemasr   rj   rq   rg   r   r   r   
all_arraysall_chunked_arraysall_record_batches
all_tablesr   r   r   r   r`   r,   r*   <module>r     s  $  
   "))(( <<7  gg]] GGBGGI	GGBHHJ	ggbiik"ggbiik"GGBOO-. GGBOO-. IIKK!r*  II  GGIHHJHHJHHJ	(   HHJIIKIIKIIK	*   		.0FGJJLJJLJJL" 
 LLkkA3
++Q
/
 LLkkA4
++R
0
 ))MMkkA4
++R
0
 ))MMkkA4
++R
0
 		-)?< __IIKIIK 
 __IIcNIIdOIIdOIIdO	 
 H		"'')T^^%5r||~FI			"'')T^^%56I		"'')R\\^4I	I))LL	0	1
 KKOO+, 3356 )) ??2779bggi0 . F F -  %4 & &B !0+) ) *d *
 KK
	 I
i  {{Q"=   
 # # a+ a+H 
5 
5 ? ? 
9 
9 I
#I. #I. I
 %'IIOORZZ\2::<01
 !% !* ,.99, ( 4 ++	 i  D  D  H
    	BsW   	\< 	] ] ]) ]8 <]]]]]&%]&)]54]58^^