o h`<*@sFddlZddlZddlZddlZddlZddlZddlZddlZzddl Z Wn e y1dZ Ynwddl m Z ddlZddlmmZdefdefdefdefdefdefd efd efgZee\ZZGd d d Z Gd ddZ!GdddZ"ddZ#ddZ$ddZ%ddZ&ddZ'ddZ(ddZ)dd Z*d!d"Z+Gd#d$d$Z,d%d&Z-d'd(Z.d)d*Z/d+d,Z0d-d.Z1d/d0Z2d1d2Z3e.e/gZ4e dure45e2e3e1ge4Z6e0ge6Z7ej89d3e6Z:ej89d3e4Z;ej89d3e7Ze;d8d9Z?ej8j e;d:d;Z@ej8j e;dd?ZBe;d@dAZCe;ej89dBejDejEejFejGgdCdDZHed?Zʐd@dAZːdBdCZej89dDdejvdEdfdejvdEdFfdejvdEdGfdejvdHdGfgdIdJZej8jdKdLZej8j eejСdMdNZej8jdOdPZej8jdQdRZej8jdSdTZej8jej8jdUdVZej8jej89dWezdXfexdYfgdZd[Zej8jd\d]Zej8jej8jd^d_Zer?r.rrr$test_limited_iterator_size_underflowkrCrGcCsVtd}tttd}||sJtjttdtd}||s)JdS)Nr>r+r5)r#r,r@rAr-rr(r/rrrtest_iterator_without_sizeqs rJcCsXtd}tjtddd}||sJtjtdtdd}||s*JdS)Nr>rr+r=r<)r#r, itertoolscountr-rrIrrrtest_infinite_iteratorzs rNcCstttddtdDWdn1swYtttjddtdDddWddS1s@wYdS)Ncs|]}ddVqdSr!rNr.0xrrr z(test_failing_iterator..rFcsrOrPrrQrrrrTrUrK)pytestraisesZeroDivisionErrorr#r,rArrrrtest_failing_iterators  "rYc@r)ObjectWithOnlyGetitemcCsdS)Nr+r)rkeyrrr __getitem__z!ObjectWithOnlyGetitem.__getitem__N)rrrr\rrrrrZr"rZcCs@tjtddttWddS1swYdS)Nz has no len()match)rVrW TypeErrorr#r,rZrrrrtest_object_with_getitems"racCs|Srrxsrrr_as_listr]rdcCt|Sr)tuplerbrrr _as_tuplergcCs t|Sr) collectionsdequerbrrr _as_deques rkcCsddt|D}|S)NcSsi|]\}}||qSrr)rRkvrrr z#_as_dict_values..) enumeratevalues)rcdctrrr_as_dict_valuessrscCs"tjt|td}||dd<|S)Ndtype)npemptyr3object)rcr9rrr_as_numpy_arrays rycCrer)r&rbrrr_as_setrhrzseqcCs2t|gd}tgd}||sJdSNr!r*r+r#r,r-r{r/r0rrrtest_sequence_typessrcCs6t|gdg}tgdg}||sJdSr|r~rrrrtest_nested_sequence_typessrcCZgd}t||}t|dksJ|jdksJ|jtks#J||ks+JdS)NTNFNr*)r#r,r3r4r5bool_r7r{r(r9rrrtest_sequence_boolean rcCsPtddtddg}t||}|jtksJ|gdks&JdS)NTFr)rvrr#r,r5r7rrrrtest_sequence_numpy_booleansrcCsXtddg}t||dd|dddg}|jtks J|gdks*JdS)NTFrr!)TNFTF)rvr,r#r5rr7)r{rqr9rrr&test_sequence_mixed_numpy_python_boolss rcCsRt|g}t|dksJ|jdksJ|jtksJ|gks'JdSr2r#r,r3r4r5r6r7r{r9rrrtest_empty_lists rcCsfgddgdg}t||}t|dksJ|jdksJ|jttks)J||ks1JdS)Nr!r*r+)r#r,r3r4r5list_rr7r{datar9rrrtest_nested_listss rfactorycCspgddgdg}tj|||td}t|dksJ|jdks#J|j|tks.J||ks6JdSNr!r*rHr+)r#r,rr3r4r5r7)r{rrr9rrr$test_nested_lists_with_explicit_types rcCsFtttj|gddgdgttdWdn1s#wYtttj|gddgdgttdWdn1sKwYtttj|gddgdgttdWdn1sswYtttj|gddgdgt tdWddS1swYdS)Nr!r*r+rH) rVrWr`r#r,rr large_list list_viewlarge_list_view)r{rrrtest_list_with_non_lists ( ( ( ("r inner_seqcCs~tj||g|ddgdg|td}t|dksJ|jdks%J|j|tks0J|gddgdgks=JdSr)r#r,rr3r4r5r7)r{rrr9rrrtest_nested_arrayss   rc Csddgddgdg}tj||ttdd}t|dks J|jdks'J|jttdks4J||ksJdS)NrFg?float64rHfloat32)rvrrr#r,r-r)rr9r(rrrtest_numpy_scalars_mixed_types  rz)Type inference for uint64 not implemented)reasonrWcCsbdttjjg}tj|td}ttj|dd}||s#Jt|}||s/JdS)NrrHr rt)rvrr rr#r,r-)rr9r( arr_inferredrrrtest_uint64_max_converts  rbits) @cCsttd|}tttjftd|dg|Wdn1s&wYtttjftd|d dg|WddS1sMwYdS)Nzint%dr*r!rr#rVrW OverflowErrorrr,rr'rrrtest_signed_integer_overflows"rcCsttd|}tttjftd|g|Wdn1s$wYtttjftdg|WddS1sDwYdS)Nzuint%dr*rrrrrtest_unsigned_integer_overflows"rtypcCsFttjtj|dg|dWddS1swYdS)N1rH)rVrWr#rr,)r{rrrrtest_integer_from_string_errors"rcCsgd}gd}tj||d}tgd}||sJtttj||dddWddS1s9wYdS)N)r!r*r+r)FTFFT)mask)r!Nr+rNr!)r#r,r-rVrWr)rrresultr(rrrtest_convert_with_masks "rcCsBddl}|t}tgd|t|ksJdS)Nrr)gccollectr#total_allocated_bytesr,)r bytes_beforerrrtest_garbage_collection s rcCVgd}t|}t|dksJ|jdksJ|jtks!J||ks)JdS)N??N@NNrr+)r#r,r3r4r5rr7rr9rrrtest_sequence_double  rcCshgd}t|}gd}t|}||sJgd}t|}gd}t|}||s2JdS)Nr)rr!NrNN)r!rNrNN)rrNrNNr~)rr9data2r0data3arr3data4arr4rrr$test_double_auto_coerce_from_integer"s    rcCsgd}gd}gd}t|tt t|Wdn1s&wYttt|WddS1sAwYdS)N)rr!r*Nll)rr!r*Nl)rr!r*Nl)r#r,rVrWr) valid_valuesinvalid_valuesinvalid_values2rrr.test_double_integer_coerce_representable_range5s     "rcCstj}|dddg}|ddg}|ddg}tj|tdtttj|tdWdn1s8wYtttj|tdWddS1sWwYdS)NriiiirH)rvrr#r,rVrWr)f32rrrrrr/test_float32_integer_coerce_representable_rangeEs    "rcCstjtddtjddgtdWdn1swYtjtddtjddgtdWdn1s?wYtjtddtddgWddS1s^wYdS) Nztried to convert to booleanr^TfoorHztried to convert to float32rztried to convert to double)rVrWrr#r,rrrrrrtest_mixed_sequence_errorsWs"rznp_str_scalar,pa_typefloat16rrrTFcCstt|}|d|dd|ddtjg}tj|||d}t|dks%J|r/|jdks.Jn|jdks6J|r@|j|ks?Jn |jtksIJ| dd|ddksYJ|rg| d duseJdSt | d srJdS) Nrr!rrrr+r*rr) rrvrr#r,r3r4r5rr7isnan)r{rrrrrr9rrrtest_sequence_numpy_doublebs  rcCstjtfD]b}tj|ddg|gd|tjgdgtd}tj||d}t|dks-J|jdks4J|jt t ks@J|rT| ddggddgdgksSJqtj | ddggdtjgdgqdS)Nr@)rr@rtrrr!)rvr,rrrxr#r3r4r5rrr7testing assert_equal)rrrr9rrr test_ndarray_nested_numpy_doubles&   $ rcCs6tjdtd}tjddgtjdtjddgtjdg|dd<tjdtd}ddg|d<ddg|d<ttt}t|g|ksFJt||g}tjddgddggddgddggg|d }||siJtjdtd}tdgtdgg|dd<t||g}| dgdggdgdgggksJdS) Nr*rtr!r+rrrrrH) rvrwrxr,rr#r infer_typer-r7r9r0 expected_typerr(rrr#test_nested_ndarray_in_object_arrays$   " (rz?Type inference for multidimensional ndarray not yet implementedcCstjddgddggtjd}tjddgddggtjd}ttt}t|g|ks0Jt||g}tjddgddggddgddggg|d}||sSJdS) Nr!r*r+rtrrrrH)rvr,rr#rrr-rrrr,test_multidimensional_ndarray_as_nested_lists"r)r value_typer!r*N)r!rrNr!sr*usrF)secondsab)saaasbbbscccr+)rrcabcrrrrrcCsNt|}tj|td}tj|g|d}|j|sJ||gks%JdS)NrtrH)r#rrvr,rxr5r-r7)rrr'ndarrayr9rrr#test_list_array_from_object_ndarrays r rcCshtjt|td}dd|D|dd<t|}tj|g|d}|j|s)J| |gks2JdS)NrtcSsg|] }tj|tdqS)rt)rvr,rx)rRitemrrr z>test_nested_list_array_from_object_ndarray..rH) rvrwr3rxr#rr,r5r-r7)rrr r'r9rrr*test_nested_list_array_from_object_ndarrays  rcCsrttttddgWdn1swYtjtddgdd}tddg}||s7JdS)NrstrTr)rVrWrr#r,floatr-)r9r(rrr!test_array_ignore_nan_from_pandass rcCstjgddddtjgdddg}t|}tjgddgdgttd}||s2Jtt}tj||d}||}||sLJdS)Nr}rrt)rrrr rH)rvr,r#rrr-r cast)rr9r(t2r0 expected2rrr$test_nested_ndarray_different_dtypess   rcCr)NrbarNmañanarr!)r#r,r3r4r5stringr7rrrrtest_sequence_unicoderrr'cCsVgd}tj||d}t|dksJ|jdksJ|j|ks!J||ks)JdS)NrrHrr!)r#r,r3r4r5r7)r'rr9rrr#test_sequence_unicode_explicit_types rcCsddtdg}gd}gd}t|}tj|td}|jtks&J||s-Jtj||d}tj||d}|j|ksBJ||sIJtj||d}tj||d}|j|ks^J||seJdS)Nquxfoobarz)squxrr)rrbarzrH) bytearrayr#r,binaryr5r-) binary_type string_typerqb_valuesu_valuesr9r(rrrcheck_array_mixed_unicode_bytes%s r'cCs:tttttttttdSr)r'r#r"r large_binary large_string binary_view string_viewrrrrtest_array_mixed_unicode_bytes:sr,cCs`d}tdt|}|g|}tj||d}t|tjsJ|j|ks&Jt||ks.JdS)Nsh0123456789abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyz0123456789abcdefghijklmnopqrstuvwxyzrH)mathceilr3r#r, isinstanceArrayr5)r'rnrepeatsrr9rrrtest_large_binary_array@s r3cCsd}tdt|}tjd||ddg|d}t|tjs J|j|ks'Jt|dks/J|d}t|t||ksAJdS)N$0123456789abcdefghijklmnopqrstuvwxyzr-rbarrHrr!) r.r/r3r#r,r0r1r5 as_buffer)r'rr2r9bufrrrtest_large_binary_valueMs r8cCsbd}tdt|}ttjtjd||ddg|dWddS1s*wYdS)Nr4r-rr5rH)r.r/r3rVrWr#ArrowCapacityErrorr,)r'rr2rrrtest_string_too_large\s "r:cCsd}dtdtdddd|dtddg}dtttfD]1}tj||d}t|d ks5J|j d kscCsgd}tj||d}|j|ksJ|ddksJdd}ttjtj|g|dWddS1s;wYdS)N)rNr5rHrrrz utf-16-le)r#r,r5as_pyencoderVrWr)r'rr9valrrrtest_sequence_utf8_to_unicodews "rBcCsnddtddg}tj|tdd}t|dksJ|jdks!J|jtdks+J|gdks5JdS)Nfoofbarb2346rrHr!)rCNrDrE)r!r#r,r"r3r4r5r7rrrrtest_sequence_fixed_size_bytess rFcCsNgd}ttjtj|tddWddS1s wYdS)N)rNrDrErrH)rVrWr#rr,r"rrrr5test_fixed_size_bytes_does_not_accept_varying_lengthss"rHcCsFdg}t|ddks Jtd}tj||d}||ks!JdS)Ns h Ar rH)r3r#r"r,r7)rr'r9rrr#test_fixed_size_binary_length_checks  rJcCstddddtdddtdddg}t|}t|dks"J|jtks+J|jdks2J|dtdddksAJ|ddusKJ|dtdddksZJ|d tdddksiJdS) Nr!iir*rrr+) datetimedater#r,r3r5date32r4r?rrrrtest_sequence_dates  "rPinputi*l,UqcCs|\}}tddddg}tj||d}tj||d}||fD]1}t|dks)J|j|ks0J|jdks7J|dtdddksFJ|ddusPJqdS)NrKr!rHr*r)rMrNr#r,r3r5r4r?)rQt ex_valuesrr9r0rSrrrtest_sequence_explicit_typess rTcCsPddg}tttjftj|tdWddS1s!wYdS)NlrH)rVrWrr#ArrowExceptionr,rO)rrrrtest_date32_overflows"rV) time_typeunitint_typemsnsc sRfdd}fdd}tddddtd d d d dtd d ddtddd dg}|}tj||d}t|d ks>J|j|ksEJ|jd ksLJ||}tt ||} | | ks`J|d  ||d ksnJ|d  ||d ks|J|d dusJ|d ||dksJ|d ||dksJddd} dS)NcsVgd}d|d}|durdS|jd|jd|j|jd}t||S)NrrZrr[rFr+i<gư>)indexhourminutesecond microsecondint)rRunits multiplierrrXrrexpected_integer_values z@test_sequence_time_with_timezone..expected_integer_valuecs6dkr |jddSdkr|j|jdddS|S)NrrrbrZ)replacerb)rRrfrrexpected_time_values  z=test_sequence_time_with_timezone..expected_time_valuer"@rrrir! 8雙rFírHr*r+rcSstj||d}t|S)N)hoursminutes)rM timedeltatimezone)rsrtoffsetrrrtzs z,test_sequence_time_with_timezone..tz)r) rMtimer#r,r3r5r4rrrr7r?) rWrXrYrgrkrr'r9rqr(rxrrfr test_sequence_time_with_timezones,    rzc Cstddddddddtdddd dd d td d dddddg}t|}t|dks.J|jtdks8J|jdks?J|dtdddddddksRJ|ddus\J|dtdddd dd d ksoJ|dtd d dddddksJdSN r!rlrmrnrIrprqrr.9rrrrrr*r+)rMr#r,r3r5 timestampr4r?rrrrtest_sequence_timestamps&   rrv)NUTCz Etc/GMT-1zEurope/BudapestrXr\c sttdfdd}fdd}tdddd d d d jtd ddddddddtddddd dddtdd dddddg}j|d|dd|dj|djg}tjd}tj ||d}t |dksJ|j |ksJ|j dksJ| d}tt||} || ksJtt |D]} || ||| ksJqdS)NpytzcsDgd}d|d}|durdStt|}t||S)Nr\rFr+)r^decimalDecimalrrrc)dtrdretsrfrrrg%s  zEtest_sequence_timestamp_with_timezone..expected_integer_valuecsd|durdSdkr|jdd}ndkr |j|jddd}dur*|jddS|S)NrrrhrZritzinfo)rjrb astimezonerv)rrrvrXrrexpected_datetime_value/s zFtest_sequence_timestamp_with_timezone..expected_datetime_valuer|r}r~rrlrmrnr!rrri US/EasternrrIrprq Europe/Moscowrrrr+rrxrHr)rV importorskiprMutclocalizervrr#rr,r3r5r4rrrr7rAr?) rvrXrgrrutcdatar'r9rqr(irrr%test_sequence_timestamp_with_timezones@     rcstdddl}|ddtddddd d d |jtd d ddddd|dtdd ddd dd|dtdddddddg}dd|D}|durb||fdd|D}tj d|d}tj ||d}| |ksxJdS)NrrPYARROW_IGNORE_TIMEZONErr|r}r~rrlrmrnrr!rrirrrIrprqrrrrcSsg|]}|jddqS)Nr)rjrRrrrrr ~szEtest_pyarrow_ignore_timezone_environment_variable..csg|]}|qSr)fromutcrrrrr rorrrH) rVrrsetenvrMrrrvr#rr,r7) monkeypatchrvrrr(r'r9rrr1test_pyarrow_ignore_timezone_environment_variablees*      rcCstdddl}tddddddd |jtd d d d ddd d|dtdd ddddd|dtdddd dddg}tjdddtjdddtjdddtjdddtjdddg}t ||D]\}}|g|}t |}|j |ks~JqidS)Nrrr|r}r~rrlrmrnrr!rrirrrIrprqrrrrrrr) rVrrrMrrrvr#rzipr,r5)rrr(rr prependedr9rrr/test_sequence_timestamp_with_timezone_inferences2          rcCstdddl}tddddddd tjd d d d ddd tjjddtjdd ddddd|jdddtjdddd ddd|jdddg}tjdddtjdddtjdddtjdddtjdddg}t ||D]\}}|g|}t |}|j |ks~JqidS)Nzoneinforr|r}r~rrlrmrnrr!rrirrrIrprqr)r[rrrrrrr) rVrrrMrvrZoneInfor#rrr,r5)rrr(rrrr9rrr8test_sequence_timestamp_with_zoneinfo_timezone_inferences2         rc Cstdddl}ddl}|jd|dddtddd d d d d |jtddddddddg}|d |j|j|d|d |jdg}t |}|j t j dddks[J|d}dd|D}||ksoJdS)Nrrl{FK|rr[)rxrXr|r}r~rrlrmrnrr!rrir*rrrcSs$g|]}|rt|dndqS)i@BN)rcrrrrrr s$zStest_sequence_timestamp_from_mixed_builtin_and_pandas_datetimes..)rVrrpandas TimestamprvrMrrrr#r,r5rrr7)rpdrrr9rqr(rrr?test_sequence_timestamp_from_mixed_builtin_and_pandas_datetimess,    rcCstdddg}tjtddtj|tddWdn1s$wYtj|tdd}||ks;Jttj d d }tjddd d |d g}tjtddtj|tddWdn1skwYtj|tdd}|dtdddksJdS)NirrIz out of boundsr^r[rHrr)rsrorlrr) rMrVrWrr#r,rr7rvru)rr9rxrrr0test_sequence_timestamp_out_of_bounds_nanoseconds"rcCsttddddddddttdddd dd d ttd d dddddg}t|}t|dks7J|jtdksAJ|jdksHJ|d tdddddddks[J|d duseJ|d tdddd dd d ksxJ|d td d dddddksJdSr{) rv datetime64rMr#r,r3r5rr4r?rrrrtest_sequence_numpy_timestamps&   rc@ eZdZdS)MyDateNrrrrrrrr rc@r) MyDatetimeNrrrrrrrrc@r) MyTimedeltaNrrrrrrrrc Cstdddg}t}tj||d}t|dksJ|j|ks!J|dtdddks0Jt ddddddd g}t d }t d }t d }tj||d}t|dksYJ|j|ks`J|dtdddddddkssJtj||d}t|dksJ|j|ksJ|dtddddddd ksJtj||d}t|dksJ|j|ksJ|dtddddddd ksJt dddg}t d }t d }t d }t|}t|dksJ|j|ksJ|dt dddksJtj||d}t|dksJ|j|ksJ|dt ddks%Jtj||d}t|dks5J|j|ks=J|dt dddksMJtj||d}t|dks]J|j|kseJ|dt dddksuJdS)Nr|r}r~rHr!rrlrmrnrrZrix{iiri)rr#rOr,r3r5r?rMrNrrrdurationru) r date_typearr_daterrZrarr_sarr_msarr_usrrrtest_datetime_subclassingsf            $rz)pandas required for nanosecond conversion)rc Cstdddddddgtdddddddgg}|D]0}td}tj||d }t|dks/J|j|ks6J|d tdddddddksIJqdS) Nr|r}r~r!rlrmrnr[rHr)rMrr#rr,r3r5r?)inputsrr[arr_nsrrr#test_sequence_timestamp_nanosecondsWs  rc Csdg}td}td}td}td}tj||d}t|dks&J|j|ks-Jt|ddks7Jt|dd ksAJtj||d}t|dksPJ|j|ksWJt|dd kscJt|dd ksmJtj||d}t|dks|J|j|ksJt|dd ksJt|dd ksJtj||d}t|dksJ|j|ksJt|ddksJt|ddksJt} Gddd} |t t fD]!} t | tjd| g| dWdn1swYqdS)Nr!rrZrr[rHrz0z1970-01-01 00:00:01z,datetime.datetime(1970, 1, 1, 0, 0, 0, 1000)z1970-01-01 00:00:00.001000z)datetime.datetime(1970, 1, 1, 0, 0, 0, 1)z1970-01-01 00:00:00.000001z*Timestamp('1970-01-01 00:00:00.000000001')z1970-01-01 00:00:00.000000001c@r)z?test_sequence_timestamp_from_int_with_unit..CustomClassNrrrrr CustomClassrr) r#rr,r3r5reprrr?r`rOdate64rVrW) rrrZrr[rrrr expected_excrr'rrr*test_sequence_timestamp_from_int_with_unithsR      rcCstddd}tddd}|d|g}t|}t|dks J|jtdks*J|jdks1J|d|ks;J|ddusEJ|d|ksOJdS Nr*ir!drir+rr) rMrur#r,r3r5rr4r?td1td2rr9rrrtest_sequence_durations  rcCstddd}tddd}t|dt|g}t|}t|dks&J|jtdks0J|j dks7J|d |ksAJ|d dusKJ|d |ksUJdSr) rMrurv timedelta64r#r,r3r5rr4r?rrrr test_sequence_duration_np_scalars rcCstdddg}tddtdddtdddtdddd}t|}tj||d}t|dks6J|j|ks=J|d||ksIJdS) Nr+irir\rHr!r)rMrur#rr,r3r5r?)rXrr(r'rrrr test_sequence_duration_with_units      rcCsRdg}t|}tj||d}t|dksJ|j|ksJ|djdks'JdS)NrrHr!r)r#rr,r3r5r)rXrr'r9rrr)test_sequence_duration_from_int_with_units  rcCsrtddd}tdd}|dg||gg}t|}t|dks"J|jttdks/J||ks7JdS)Nr!rirr*r) rMrur#r,r3r5rrr7rrrr#test_sequence_duration_nested_listss  rcCs~tddd}tdd}|dg||gg}tj||tdd}t|dks)J|j|tdks5J||ks=JdS)Nr!rirrZrHr*)rMrur#r,rr3r5r7)rrrrr9rrr6test_sequence_duration_nested_lists_with_explicit_types rcCstddd}tdd}t|dgt|t|gg}t|}t|dks+J|jtt dks8J| |dg||ggksFJtjt|dgddtt|t|gg}t|}t|dkslJ|jtt dksyJ| |dg||ggksJdS)Nr!rirr*rztimedelta64[us]rt) rMrurvrr#r,r3r5rrr7rrrr)test_sequence_duration_nested_lists_numpys      rcCsgd}t|}|jtksJ||ksJdgdgdg}t|}|jttks2J||ks:Jdggddgg}t|}|jttksTJ||ks\Jdddgggddgdgg}t|}|jtttks~J||ksJtjtjf}t |tdddggWdn1swYt |tddggWdn1swYt |tdgdgddgggWddS1swYdS)Nrr!r*)r*r+r) r#r,r5rr7rrArrowTypeErrorrVrW)rr9 exceptionsrrrtest_sequence_nesting_levels s4        "rcCsDgd}ttjt|WddS1swYdS)N)rr!r)rVrWr#rr,rGrrrtest_sequence_mixed_types_fails+s "rcCsZddddidddg}t}tttj||dWddS1s&wYdS) Nz-10z-5rr!0510rH)r#rrVrWr`r,)rr5rrr3test_sequence_mixed_types_with_specified_type_fails1s  "rcCsXtdtdg}tjtjtjtjfD]}tj||dddd}||ks)JqdS)Nz1234.183z8094.234r}r+ precisionscalerH rrr# decimal32 decimal64 decimal128 decimal256r,r7rr5r9rrrtest_sequence_decimal9 rcCTtdtdg}tjtjtjfD]}tj||dddd}||ks'JqdS)Nz1234234983.183z 80943244.234r~r+rrHrrr#rrrr,r7rrrr*test_sequence_decimal_different_precisions@s rcCsRtdtdg}tjtjtjfD]}tj||ddd}||ks&JqdS)N 12342349838094324rF)rrHrrrrrtest_sequence_decimal_no_scaleIs rcCr)Nz -1234.234983z -8.094324rFrrrHrrrrrtest_sequence_decimal_negativePs rcCsXtdtdg}tjtjtjtjfD]}tj||dddd}||ks)JqdS)Nz -.4234983z.0103943r}rrHrrrrr#test_sequence_decimal_no_whole_partWrrcCsPtdtdg}tjtjfD]}tj||dddd}||ks%JqdS)Nz-394029506937548693.42983z32358695912932.01033rlrrrH)rrr#rrr,r7rrrr#test_sequence_decimal_large_integer^srcCsRgd}dd|D}tjtjfD]}tj||dddd}||ks&JqdS)N)rr!lc&E(ONcSsg|]}t|qSr)rrrQrrrr hroz7test_sequence_decimal_from_integers..rrrH)r#rrr,r7)rr(r5r9rrr#test_sequence_decimal_from_integersfs rcCsFtjtddttdgWddS1swYdS)Nzprecision out of ranger^P11111111111111111111111111111111111111111111111111111111111111111111111111111111)rVrWrr#r,rrrrrr(test_sequence_decimal_too_high_precisionns"rc Cstdtddftdtddftdtdd ftd tddftd tddftd td dftdtddftdtddftdtddff D]%\}}t|g|ksjJt|g}|j|kswJ|d|ksJq\dS)N1.234rr+12300rrz12300.0rr!1.23E+4z123E+2z123E+4r}z0.0123z0.01230z1.230E-2rrr#rrr,r5r7)rrr9rrrtest_sequence_decimal_inferts  rc Cs"tdtdgtddftdtdgtddftdtdgtddftd td gtd d ftd tdgtd d ftdtdgtddftdtdgtddfg}|D]!\}}t||kszJt|}|j|ksJ||ksJqmdS)Nrz3.456rr+z456.7rz123.4z4.567123e24567e3r}r123e44567e2z0.123z0.04567rz0.001z1.01E5 r)casesrrr9rrr!test_sequence_decimal_infer_mixeds8         rc CsTtdtddtddtddgtddtddgftdtddtddtdd gtddtdd gftd tddtddtdd gtddtdd gffD]E\}}}|D]}tj|g|d }|j|kszJ|d|ksJqi|D]}tt tj|g|d Wdn1swYqqbdS) Nrrr+rr*rrrrrH) rrr#rr,r5r7rVrWr)rtyps wrong_typsrr9rrr test_sequence_decimal_given_types0    rcCs*ttd}td}||sJdS)Nr+r>)r#r,rAr-r.rrrtest_range_typess rcCsRttd}t|dksJ|jdksJ|jtksJ|gks'JdSr2)r#r,rAr3r4r5r6r7r8rrrtest_empty_ranger;rcCs tjjggd}|jtgksJt|dksJ|gks"Jtjgdtd}tjgdt d}tjgdt d}tj|||ggd}ddd dd dd dd d ddg}|}||ksnJ||ft t tj|gddgWddS1swYdS)N)namesrNr*r+rHrNrTFN)intsstrsboolsrTr*Fr+rrr )r# StructArray from_arraysr5structr3r7r,rrrrVrWr)r9rr r r(pylistrrrtest_structarrays(    "rc Csttdttdttdtg}tjg|d}|gks)Jddddd d d dg}tj||d}||ksDJddd didd d g}tj||d}ddddddddddd ddg}||ksqJdS)NrrrrHrrTrrrFrrr r#r fieldrrrr,r7)r'r9rr(rrrtest_struct_from_dictss.     rc Csttdttdttdtg}tjg|d}|gks)Jddddd d g}tj||d}|dddd ddd d gksLJdS) NrrrrHrr)rrrF)rrrr)r'r9rrrr!test_struct_from_dicts_bytes_keyss  rc Cs8ttdttdttdtg}ddg}dddd d d d d g}tj||d }tjt |t d}||dd<tj||d }| |ksOJ| |sVJgd}dddd dd dd d g}tj||d }| |ksvJdD]!}t jtddtj|g|d Wdn1swYqxdS)NrrrrrT)rrFrrTrrrFrHrt))rrNN)rNF))rrr)rrTN(?i)tuple sizer^)r#r rrrrr,rvrwr3rxr7r-rVrWr)r'rr(r9data_as_ndarrayr0tuprrrtest_struct_from_tupless:     rc Csttdttdttdtg}gdgddg}tj||d}|ddd d d d d d dgks=Jttdttdttdtg}gdgdg}tj||d}t t  |Wdn1szwYttdttdttdtg}ggdddggdg}dddd ddd d dddd dd d d g}tj||d}||ksJdS)Nrrr)rrrrrT)rrrrrFrHrrTrrrF)r)rr)rT)r)rr)rF)rr*)rbaz)rr!rr)djuliar*r r!) r#r rrrrr,r7rVrWr)r'rr9r(rrrtest_struct_from_list_of_pairs8sX         r$c CsPttdttdttdtg}ggdg}d}tjt|dtj ||dWdn1s;wYd}t t dt d d g}|D]V}t | t|t|j}tjt|dtj |ggd g|dWdn1s}wYtjt|dtj gd |gg|dWdn1swYqOdS) Nrrr)rr)rNz0The expected field name is `b` but `c` was givenr^rHzKCould not convert {} with type {}: was expecting tuple of (key, value) pairz unknown-keyr)rr)rN)r#r rrrrrVrWrr,rfreescapeformatrr5rr`)r'rmsgtemplaterkey_value_pairrrr%test_struct_from_list_of_pairs_errorsosRr+c Csttdttdttdtg}dddddg}tttj ||d WddS1s:wYdS) NrrrrrrFrrH) r#r rrrrrVrWr`r,)r'rrrrtest_struct_from_mixed_sequences  "r,cCsttdttdttdtg}dddddd d dg}t|}t|j|| |ks9Jddd didd d g}ddddddddddd ddg}t|}t j t |t d }||dd<t|}t|j|| |ks~J||sJttdttdtttdtgtdtg}ddgdddd dd ddd dd d g}t|}| |ksJtig}|jtgksJ| igksJttjtjftdddigWddS1swYdS)NrrrrrTrrrFrr rtaaab)r-r.r!r*)r#r rrrrr,r)r5r7rvrwr3rxr-rrVrWrr)rrr9r(rr0rrr test_struct_from_dicts_inferencesZ             $r/c Csgd}gd}gd}gd}||||g}tj|gd}tjtj|ddtj|ddt|tj|ddggd}tttj|Wdn1sTwY||s`JdS) Nrrrr})rr r  int_nonnullrrHutf8)r#r r r,rVrWrr-)rr r  ints_nonnullarraysrr(rrr#test_structarray_from_arrays_coerces(     r4cCstddtdtdg}tt t|}Wdn1s#wYtj|dd}|jt ddks:J| |ddddgksJJtj|t dddd }| td dddgkseJdS) NrrTrrr+r*rFrz1.2340) rrrrVrWr`r#r,r5rr7)rqr,rrr$test_decimal_array_with_none_and_nans   "r5c Csddddddgdddgdd dd d dd ddgd ddgg}dd|D}tj|tttd}||ks@Jd|d<d|d<tj|tttd}||ks_Jdd igigdddgfD]%}tjtddtj|gtdddWdn1swYqldd dgdddiigfD]&}tjtj ddtj|gtdddWdn1swYqdS)Nrr!r[rrr*rr+drerfgr}cSsg|] }dd|DqS)cSsg|] }|d|dfqSr6r)rRr"rrrr  sz2test_map_from_dicts...r)rRentryrrrr  rz'test_map_from_dicts..rHr)rlrmz Invalid Mapr^i4rr[r) r#r,map_r"rr7rVrWrr)rr(r9r;rrrtest_map_from_dictss2  r>c Csddgdggddgg}tj|tttd}||ks#Jd|d<tj|tttd}||ks>Jdgd gd gfD]%}tjtd d tj|gtd d dWdn1sfwYqFdS)N)rr!)rr*)rr+))r7r)r8r)r9N)r:r}rHr!)rr)rrTrr^r<) r#r,r=r"rr7rVrWr)r(r9r;rrrtest_map_from_tuples#s r?cCstjttd}tjgd|d}t|jtjsJ|j|s%Jtjgdtd}tjddgtd}|j |sCJ|j|sKJdS)Nr)FFTFTrH)rrr!rr!FT) r# dictionaryrrr,r0r5DictionaryTyper-indices)rrexpected_indicesexpected_dictionaryrrrtest_dictionary_from_boolean:srFrcCstjt|d}tjgd|d}t|jtjsJ|j|s#Jtjgdtd}tjgd|d}|j|s?J|j|sGJdS)Nr@)r!r*r!r!r*r+rH)rr!rrr!r*r}) r#rArr,r0r5rBr-rCrrrrDrErrrtest_dictionary_from_integersFs rHinput_index_typecCs8tj|td}tjtd|d}|j|sJdS)Nr@rFrH)r#rArr,rAr5r-)rIrr9rrrtest_dictionary_index_type^s rJcCstjttd}tjtd|d}ttt}|j|s&Jtjtd|d}ttt}|j|sAJdS)Nr@rH) r#rArrr,rAr5r-r)rrr(rrr"test_dictionary_is_always_adaptivemsrMcCsttfD]@}tt|}tjgd|d}t|jtjs$Jtjgdtd}tjgd|d}|j |s@J|j |sHJqtttd}tjgd|d}t|jtjsfJtjgdtd}tjgdtdd}|j |sJ|j |sJdS) N)rbbrrOcccrH)rr!r*r!r*r+)rNrrOrPr+)aaarQbbbrPrR)rrr!r*r!)rQrRrP) r#r"rrArr,r0r5rBrCr-rGrrrtest_dictionary_from_strings|srS)rXr() milliseconds) microsecondsie;cCst|}tjdg|d}tj||d}||sJ|g}|dkr   rncCsdggd}t|}t|tjsJ|jdksJt|ddks%Jt|ddks0J|ddggdks?JdS)Nxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxi r*rir!)r#r,r0rfrgr3rhr7)valsr9rrr!test_auto_chunking_list_of_binary s  "rqcCstjddd}|gd}tj|ttd}t|tjs Jt|dks(Jtjddd}|gd}tj|ttd}t|tj sHJ|j dksOJt| ddksZJt| d d kseJ| d }|d}t|tj svJtj|td}|j |ksJdS) N)rrtr}rHrr*rr!)rvonesr#r,rrr0r1r3rfrgrh ListScalarrq)r rr9rhscalarr(rrrtest_auto_chunking_list_like s"   rvcCsvttt}dgd}|gd}tj||d}t|tjs#Jt|ddks.Jt|ddks9JdS)N)r!r!rrrrHrr}r!)r#r=rr,r0rfr3rh)r'r rr9rrrtest_auto_chunking_map_type) s  rw)r'charrSr`cCs|d}|d}ttdttdttd|g}dd|dgd }|dd|dtj||d }t|tjsAJ|dd|dtj||d }t|tj sYJ|j d ks`Jt | d d kskJt | ddksvJ| dd  dd|dksJdS)Nrarbboolinteger string-likeTr!)ryrzr{rcrHr*rre)r#r rrrappendr,r0r1rfrgr3rhr?)r'rxrirj struct_typerr9rrrtest_nested_auto_chunking6 s, r~cCdgd}tj|td}t|tjsJt|dksJt|jdks'Jtdt }tj||td}t|tjs?Jt|dksGJt|jdksPJtj|t d}t|tjsaJt|dksiJt|jdksrJdS)NsaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaairHr!rr5 r#r,rr0rfr3chunksrvzerosryr"itemsr9rrrr$test_array_from_pylist_data_overflowW s  rcCr)NrlrHr!rrrrrr&test_array_from_pylist_offset_overflowm s  r)r scalar_datarr)r!NN)rrNrri)r!rir}rHrQrR)r!r*r!)rr*)r!rrcCsht||tkr tdtj||d}t||}||s"Jtj|||d}||s2JdS)Nz&The elements in the set get reordered.rH)r5r&rVskipr#r,r-)r{rrrexpectrrrr!test_array_accepts_pyarrow_scalar sb rcCs|tdtdtdg}tjtjdd t|Wdn1s(wY|dtddg}tjtjdd t|Wdn1sOwY|tdtddg}tjtjdd t|Wdn1sywY|tdtddg}tjtjdd t|Wdn1swYtjtjddtjtdgtd Wdn1swYtjtjd dtjtdgt d WddS1swYdS) Nr!rrz'cannot mix scalars with different typesr^zIpyarrow scalars cannot be mixed with other Python scalar values currentlyz0.1z=Cannot append scalar of type string to builder for type int32rHz;Cannot append scalar of type int64 to builder for type null) r#rurVrWrr,rvrrr6)r{sequencerrr(test_array_accepts_pyarrow_scalar_errors sD     "r)rirMrrLr.r% hypothesishrVnumpyrvrWpyarrow.pandas_compatrpyarrowr#pyarrow.tests.strategiestests strategiespastrrrrrrr r int_type_pairsrnp_str_int_types pa_int_typesr rr r)r1r:rBrErGrJrNrYrZrardrgrkrsryrzSEQUENCE_TYPESr|ITERABLE_TYPESCOLLECTIONS_TYPESmark parametrizeparametrize_with_iterable_typesparametrize_with_sequence_types"parametrize_with_collections_typesrrrrrrrrrrrrrrrrrrrrrrrrrxfailrrrrrrrrrrrrrrrrrrAssertionErrorrrr6rNtodayrOrrytime32time64nowrrurr"r(rr)r r rrrrr+rr'r, large_memoryr3slowr8r:r>rBrFrHrJrPrTrVrzrrrr timezone_datarrrrrrrrr have_pandasrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr$r+r,r/r4r5r>r?rFrHrJrMrSr[r^given all_arraysr_rnrqrvrwr~rrru MonthDayNanomonth_day_nano_intervalr*rArrrrrrs~                               "                                                 : C     A  5         "                 " 7 .  4        & )     $ * (     "($* & (& .        `