U
|
ý°dT ã f @ s d Z ddlZddlmZ ddlZddlZddlmZ ddl m
|
Z
|
ddl m Z m Z mZmZmZmZmZmZ ddlmZ ddlmZ ddlZdgd d
|
Zd d Zej d dde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfdde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfdde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfdde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd!de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd"de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd#de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd$de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd%de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd&de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfd'de d¡j d dfde d¡j d dfde d¡j d dfde d¡j d dfg`¡d(d) Z!d*d+ Z"d,d- Z#ej d d.d/e $d/¡j d0 d1fd2d1e $d/¡j d0 d1fd3d4e $d/¡j d0 d5fd6d5e $d/¡j d0 d5fd7d/e $d/¡j d8 d5fd9d1e $d/¡j d8 d5fd:d4e $d/¡j d8 d5fd;d5e $d/¡j d8 d5fg¡d<d= Z%d>d? Z&d@dA Z'dBdC Z(dDdE Z)dFdG Z*ej j dHe +dI¡e ,dI¡ej -¡ ej. /¡ gdJdKdLdMgdNdOdP Z0ej j dHe +dI¡e ,dI¡ej -¡ ej. /¡ gdJdKdLdMgdNdQdR Z1dSdT Z2ej j dHej3ddUe 4dVdW¡e 5d¡gdXdYdZgdNd[d\ Z6d]d^ Z7d_d` Z8dadb Z9dcdd Z:dedf Z;dS )hz>
|
These test the method maybe_promote from core/dtypes/cast.py
|
é N)ÚDecimal)ÚNaT)Ú maybe_promote)Úis_complex_dtypeÚis_datetime64_dtypeÚ is_datetime_or_timedelta_dtypeÚis_float_dtypeÚis_integer_dtypeÚis_object_dtypeÚ is_scalarÚis_timedelta64_dtype)ÚDatetimeTZDtype)Úisnac C s8 t |s tt| |\}}|}||ks*tt|| dS )a|
|
Auxiliary function to unify testing of scalar/array promotion.
|
|
Parameters
|
----------
|
dtype : dtype
|
The value to pass on as the first argument to maybe_promote.
|
fill_value : scalar
|
The value to pass on as the second argument to maybe_promote as
|
a scalar.
|
expected_dtype : dtype
|
The expected dtype returned by maybe_promote (by design this is the
|
same regardless of whether fill_value was passed as a scalar or in an
|
array!).
|
exp_val_for_scalar : scalar
|
The expected value for the (potentially upcast) fill_value returned by
|
maybe_promote.
|
N)r ÚAssertionErrorr Ú _assert_match)ÚdtypeÚ
|
fill_valueÚexpected_dtypeÚexp_val_for_scalarZ result_dtypeÚresult_fill_valueÚexpected_fill_value© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/dtypes/cast/test_promote.pyÚ_check_promote s
|
r c C s t | }t |}t| drD| jj|jjks.t| jj|jjks\tn||ks\|j|jks\t| |k}|tjkrrd}t | ot |}|s|std S )Nr F)
|
ÚtypeÚhasattrr Úkindr ÚitemsizeÚ__name__ÚpdÚNAr )r r Zres_typeZex_typeZ match_valueZ match_missingr r r r = s
|
|
r z!dtype, fill_value, expected_dtype)Úint8é r! r! r" Úint16Úint32Úint64Úobject)r! éÿÿÿÿr! )Úuint8r" r( r( Úuint16Úuint32Úuint64)r( r' r# )r# r" r# )r# r' r# )r) r" r) )r) r' r$ )r$ r" r$ )r$ r' r$ )r* r" r* )r* r' r% )r% r" r% )r% r' r% )r+ r" r+ )r+ r' r& c C s: t | ¡} t |¡}t j|g|dd }t| ||| d S ©N©r r ©Únpr Úarrayr ©r r r r r r r Ú test_maybe_promote_int_with_intX s l
|
|
r2 c C sJ t | ¡}t |¡}t jdg|dd }t j}t |¡}t|||| d S ©Nr" r- r )r/ r r0 Úfloat64r )Úany_int_numpy_dtypeÚfloat_numpy_dtyper Ú
|
fill_dtyper r r r r r Ú!test_maybe_promote_int_with_floatÍ s
|
|
|
r8 c C sR t | ¡}t |¡}t jdg|dd }|}t j|g|dd }t|||| d S r3 r. )r6 r5 r r7 r r r r r r Ú!test_maybe_promote_float_with_intÜ s
|
|
r9 )Úfloat32r" r: r: gñ?r4 )r4 r" r4 )Ú complex64r" r; r; Ú
|
complex128)r< r" r< )r: ù ð? ð?r; yñ? ð?)r4 r= r< )r; r= r; )r< r= r< c C s: t | ¡} t |¡}t j|g|dd }t| ||| d S r, r. r1 r r r Ú#test_maybe_promote_float_with_floatì s
|
|
r> c C sT t t¡}t | ¡}t jdg|dd }|tkr:t t¡n|}|}t|||| d S r3 )r/ r Úboolr0 r& r )Úany_numpy_dtyper r7 r r r r r r Ú test_maybe_promote_bool_with_any s
|
|
rA c C sJ t | ¡}d}|tkr t t¡n|}t j|g|dd }t|||| d S )NTr- r )r/ r r? r& r0 r ©r@ r r r r r r r Ú test_maybe_promote_any_with_bool s
|
|
rC c C sJ t | ¡}t |¡}t jdg|dd }t t j¡}|}t|||| d S r3 )r/ r r0 Úobject_r )Z bytes_dtyper@ r r7 r r r r r r Ú!test_maybe_promote_bytes_with_any) s
|
|
rE c C s@ t | ¡}d}t t j¡}t j|g|dd }t|||| d S )Ns abcr- r )r/ r rD r0 r rB r r r Ú!test_maybe_promote_any_with_bytes7 s
|
|
rF c C sd t | ¡}t |¡}t jdg|dd }t|rD|}t |¡ ¡ }nt t¡}|}t|||| d S r3 ) r/ r r0 r r Ú TimestampÚ to_datetime64r& r )Zdatetime64_dtyper@ r r7 r r r r r r Ú&test_maybe_promote_datetime64_with_anyE s
|
|
|
rI r Únowz pd.Timestampz np.datetime64zdatetime.datetimez datetime.date)Úidsc C sl t | ¡}t|r&|}t |¡ ¡ }nt t¡}|}t|tj krZ|j
|
dkrZt t¡}|}t |||| d S )NÚM) r/ r r r rG rH r& r ÚdatetimeÚdater r )r@ r r r r r r r Ú&test_maybe_promote_any_with_datetime64X s
|
|
|
rO c C sH t | ¡}t|d}tj|g|dd }t t¡}|}t|||| d S )N)Útzr- r )r/ r r r ZSeriesr& r )r@ Ztz_aware_fixturer r r7 r r r r r Ú2test_maybe_promote_any_numpy_dtype_with_datetimetzv s
|
|
|
rQ c C sd t | ¡}t |¡}t jdg|dd }t|rD|}t |¡ ¡ }nt t¡}|}t|||| d S r3 ) r/ r r0 r r Ú TimedeltaÚto_timedelta64r& r )Ztimedelta64_dtyper@ r r7 r r r r r r Ú'test_maybe_promote_timedelta64_with_any s
|
|
|
rT )Údaysé Úhz pd.Timedeltaznp.timedelta64zdatetime.timedeltac C sF t | ¡}t|r&|}t |¡ ¡ }nt t¡}|}t|||| d S )N)r/ r r r rR rS r& r )r@ r Úrequestr r r r r r Ú'test_maybe_promote_any_with_timedelta64¢ s
|
|
rY c C sH t | ¡}t |¡}t jdg|dd }t t¡}|}t|||| d S r3 ©r/ r r0 r& r )Z string_dtyper@ r r7 r r r r r r Ú"test_maybe_promote_string_with_any¶ s
|
|
|
r[ c C s. t | ¡}d}t t¡}|}t|||| d S )NÚabc)r/ r r& r rB r r r Ú"test_maybe_promote_any_with_stringÄ s
|
|
|
r] c C sH t | ¡}t |¡}t jdg|dd }t t¡}|}t|||| d S r3 rZ )Z object_dtyper@ r r7 r r r r r r Ú"test_maybe_promote_object_with_anyÑ s
|
|
|
r^ c C s4 t | ¡}t d¡}t t¡}|}t|||| d S )Nr" )r/ r r Z
|
DateOffsetr& r rB r r r Ú"test_maybe_promote_any_with_objectß s
|
|
|
|
r_ c C s |}t | ¡}t|trV|jdkrF|jdkr:t t j¡}n|}t j}nt t¡}|}n®t|rt|t k rtt j}t j}nt
|
|r|t krt t¡}|}npt |r®|}| dd¡}nV|t krÆt t¡}t }n>t |sÖt|râ|}t j}n"t t¡}|tjkrþtj}nt j}t|||| d S )N)ÚiÚuÚfÚc)r` ra r Úns)r/ r Ú
|
isinstancer r r4 Únanr& r r r
|
r r r r r r r )r@ Z nulls_fixturer r r r r r r Ú*test_maybe_promote_any_numpy_dtype_with_naí s<
|
|
|
|
|
|
|
|
|
rg )N)<Ú__doc__rM Údecimalr Únumpyr/ ZpytestZpandas._libs.tslibsr Zpandas.core.dtypes.castr Zpandas.core.dtypes.commonr r r r r r
|
r r Zpandas.core.dtypes.dtypesr Zpandas.core.dtypes.missingr Zpandasr r r ÚmarkZ parametrizeZiinfoÚmaxÚminr2 r8 r9 Zfinfor> rA rC rE rF rI rG Z
|
datetime64rJ rN ÚtodayrO rQ rT rR Z timedelta64Ú timedeltarY r[ r] r^ r_ rg r r r r Ú<module> sZ (
|
|
þk
|
|
ìþ
|
|
ü
|
ø
|
|
ü
|
ø
|
|
ý
|
|
