U ,d@sdZddlZdddddddd d d d d dddZedejZddZddZ ddddddddddd d! Z d"d#Z d$d%Z d&d'Z d(d)Zd*d+Zd,d-Zd.d/ZdS)0z Helper functions for np.timedelta64 and np.datetime64. For now, multiples-of-units (for example timedeltas expressed in tens of seconds) are not supported. N )YMWDhmsmsusnsZpsfsasZnatcCst|dkt|dkkS)zB Whether the *src* and *dest* units are of the same kind. rDATETIME_UNITSsrcdestr!?/tmp/pip-unpacked-wheel-eu7e0c37/numba/np/npdatetime_helpers.py same_kind(sr#cCsXt|}t|}||krdS|dkr(dS||kr4dS|dkr@dS|dkrT|dkrTdSdS)NTrFrrrr!r!r"can_cast_timedelta_units/sr$)rr )rr)r)r<)rr&)r)r r')r r')r r')r r')r r') rrrrrrrr r r r cCsb|dkr dS|}d}||krNzt|\}}Wntk rBYdSX||9}q||krZ|SdSdS)z Return an integer multiplier allowing to convert from *big_unit_code* to *small_unit_code*. None is returned if the conversion is not possible through a simple integer multiplication. rrN)_factorsKeyError)Z big_unit_codeZsmall_unit_codecfactorZmultr!r!r"_get_conversion_multiplierUs  r,cCstt|t|S)zl Return an integer multiplier allowing to convert from timedeltas of *src_unit* to *dest_unit*. )r,r)Zsrc_unitZ dest_unitr!r!r"get_timedelta_conversion_factorosr-cCst|}t|}|dks |dkr*|ddfS|dkrJ|dkrJtd||fd\}}|dkr|dkrpd}d }d}q|dkrd}d }d}n2|dkr|dkrd}d }d}n|dkrd}d }d}||krt||}|d k st||f||||fSt||}|d k s t||f||||fSd S)a Compute a possible conversion for combining *datetime_unit* and *timedelta_unit* (presumably for adding or subtracting). Return (result unit, integer datetime multiplier, integer timedelta multiplier). RuntimeError is raised if the combination is impossible. rrrz1cannot combine datetime64(%r) and timedelta64(%r))rrrri:ii ii@N)r RuntimeErrorr,AssertionError) datetime_unittimedelta_unit dt_unit_code td_unit_codeZ dt_factorZ td_factorr+r!r!r"!get_datetime_timedelta_conversionxsD   r4cCsPt|}t|}|dkr|S|dkr(|S|dkr<|dkrsL  4