ó šßÈ[c@sƒddlZddlZddlZddlZddlmZddlmZ dfd„ƒYZ dfd „ƒYZ d „Z dS( iÿÿÿÿNi(tTimei(t broadcast_totTestManipulationcBszeZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „ZRS( sEManipulation of Time objects, ensuring attributes are done correctly.c CsMtjddƒ}tjdddƒ}t|dd…tjf|dddd ƒ|_t|dd…tjf|dddd d d ƒ|_t|dd…tjf|dddd d tjt|ƒƒtjt|ƒƒfƒ|_t|dd…tjf|dddd d tjt|ƒƒtjt|ƒƒfƒ|_dS(NiPÃiZÃgg+‡ÙÎ÷ï?gš™™™™™É?tformattmjdtscaletutctlocationt45dt50d(RR (tnptarangeRtnewaxistt0tt1tlentt2(tselfRtfrac((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pytsetups2, ,0,cCsÙ|jjƒ}|j|jjfks-t‚tj|j|jjjƒkƒsWt‚tj|j|jjƒsxt‚|j dkst‚|j jƒ}|j|j jfksºt‚tj|j|j jjƒkƒsät‚tj|j|j jƒst‚|j |j j ks t‚|j jƒ}|j|j jfksMt‚tj|j|j jjƒkƒswt‚tj|j|j jƒs˜t‚|j j|jks³t‚tj|j |j j ƒ sÕt‚dS(N( R traveltshapetsizetAssertionErrorR talltjd1tmay_share_memoryRtNoneRR(Rtt0_raveltt1_raveltt2_ravel((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_ravels *!*!*!cCsy|jjƒ}|j|jjfks-t‚|jdksBt‚tj|j |jj ƒ sdt‚|j jƒ}|j|j jfks‘t‚tj|j |j j ƒ s³t‚|j|j jk sÎt‚|j|j jksét‚|j jƒ}|j|j jfkst‚tj|j |j j ƒ s8t‚|jj|jksSt‚tj|j|j jƒ sut‚dS(N( R tflattenRRRRRR RRRR(Rt t0_flattent t1_flattent t2_flatten((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_flatten0s"""cCse|jjƒ}|jdks$t‚tj|j|jjjƒkƒsNt‚tj|j|jjƒsot‚|jdks„t‚|j jƒ}|jdks¨t‚tj|j|j jjƒkƒsÒt‚tj|j|j jƒsót‚|j|j jkst‚|j jƒ}|jdks2t‚tj|j|j jjƒkƒs\t‚tj|j|j jƒs}t‚|jj|jks˜t‚tj|j|j jƒs¹t‚|j j }|jdksÚt‚tj|j|j jj kƒst‚tj|j|j jƒs"t‚|jj|jjks@t‚tj|j|j jƒsat‚dS(Nii (ii (ii (ii (ii ( R t transposeRRR RRRRRRRtT(Rt t0_transposet t1_transposet t2_transposett2_T((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyttest_transposeAs,*!*!*!! '!cCs½|jjƒ}|jdks$t‚tj|j|jjjƒkƒsNt‚|jdksct‚tj |j|jjƒs„t‚|j jƒ}|jdks¨t‚tj|j|j jjƒkƒsÒt‚|j|j jksít‚tj |j|j jƒst‚|j jƒ}|jdks2t‚tj|j|j jjƒkƒs\t‚|jj|jkswt‚tj |j|j jƒs˜t‚tj |j|j jƒs¹t‚dS(Ni(i(i(i( R tdiagonalRRR RRRRRRR(Rt t0_diagonalt t1_diagonalt t2_diagonal((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_diagonalZs *!*!*!cCsá|jjddƒ}|jdks*t‚tj|j|jjjddƒkƒsZt‚tj|j|jjƒs{t‚|jdkst‚|j jddƒ}|jdksºt‚tj|j|j jjddƒkƒsêt‚tj|j|j jƒs t‚|j|j jks&t‚|j jddƒ}|jdksPt‚tj|j|j jjddƒkƒs€t‚tj|j|j jƒs¡t‚|jj|jks¼t‚tj|j|j jƒsÝt‚dS(Niiii (ii (ii (ii ( R tswapaxesRRR RRRRRRR(Rt t0_swapaxest t1_swapaxest t2_swapaxes((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_swapaxesls 0!0!0!cCs|jjdddƒ}|jdks-t‚tj|j|jjjjdddƒkƒsct‚tj|j|jjjjdddƒkƒs™t‚tj |j|jjƒsºt‚tj |j|jjƒsÛt‚|j dksðt‚|j jdddƒ}|jdkst‚tj|j|j jjdddƒkƒsPt‚tj |j|j jƒsqt‚|j |j j ksŒt‚|j jdddƒ}|jdks¹t‚tj|j|j jjdddƒkƒsìt‚tj |j|j jƒs t‚|j j|jks(t‚tj |j |j j ƒsIt‚|j jdddƒ}|jdksvt‚tj|j|j jjdddƒkƒs©t‚tj |j|j jƒsÊt‚|j j|jksåt‚tj |j |j j ƒ st‚|j jddƒj}|jdks4t‚tj |j|j jƒsUt‚|j j|jkspt‚tj |j |j j ƒs‘t‚|jddƒ}|jd ks¸t‚tj |j|j jƒ sÚt‚|j j|jksõt‚tj |j |j ƒ st‚dS( Niii (iii(iii(iii(iii(ii (i i(R treshapeRRR RRt_timetjd2RRRRRR&(Rt t0_reshapet t1_reshapet t2_reshapet t2_reshape2t t2_reshape_ttt2_reshape_t_reshape((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_reshape~sH66!!3!3!!3!"!!"  cCsÂ|jjƒ}d|_|jdks-t‚tj|j|jjjjdddƒkƒsct‚tj|j |jjj jdddƒkƒs™t‚|j dks®t‚|j }t jtƒd |_WdQX|j|j jksñt‚|jj|j jkst‚|j j|j jks-t‚|jjƒ}d |_|jd ksZt‚tj|j|jjjdddƒkƒst‚|j jd ks¥t‚d |j_|jjdksÉt‚|jjjdksät‚|jj jdksÿt‚|jj jdkst‚|jj jdks5t‚|jj}t jtƒd|j_WdQX|jjj|ks~t‚|jj j|ks™t‚|jj j|ks´t‚|jƒdS(Niii iii2(iii(iii(i i(iii(iii((iii(iii(iii(iii(iii(iii(i2(R tcopyRRR RRR7R6R8RRR&tpytesttraisestAttributeErrorRRtstridesR(RR9t t0_reshape_tR:toldshape((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyttest_shape_setting§s< 66  3  cCs|jjdddddƒjƒ}|jdks9t‚tj|j|jjjdddƒkƒslt‚tj|j|jjƒst‚|j dks¢t‚|j jdddddƒjƒ}|jdksÛt‚tj|j|j jjdddƒkƒst‚tj|j|j jƒs/t‚|j |j j ksJt‚|j jdddddddƒjƒ}|jdks‰t‚tj|j|j jjdddƒkƒs¼t‚tj|j|j jƒsÝt‚|j j|jksøt‚tj|j |j j ƒst‚dS(Niii(iii(iii(iii( R R6tsqueezeRRR RRRRRRR(Rt t0_squeezet t1_squeezet t2_squeeze((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_squeezeÑs $3!$3!*3!cCsA|jdd…tjdd…f}|jdks:t‚tj|j|jjdd…tjdd…fkƒszt‚tj|j|jjƒs›t‚|jdks°t‚|j dd…dd…tjf}|jdksêt‚tj|j|j jdd…dd…tjfkƒs*t‚tj|j|j jƒsKt‚|j|j jksft‚|j dd…dd…tjf}|jdks t‚tj|j|j jdd…dd…tjfkƒsàt‚tj|j|j jƒst‚|jj|jkst‚tj|j|j jƒs=t‚dS(Ni ii(i ii(i ii(i ii( R R R RRRRRRRRR(Rt t0_adddimt t1_adddimt t2_adddim((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyttest_add_dimensionãs %@!%@!%@!cCsÕ|jjd ƒ}|jd ks't‚tj|j|jjjjd ƒkƒsWt‚|jdkslt‚|j jdddƒ}|jdks™t‚tj|j|j jjdddƒkƒsÌt‚|j|j jksçt‚|j jddd ƒ}|jdkst‚tj|j|j jjddd ƒkƒsGt‚|jj|jksbt‚|j jdƒ}|jdks‰t‚tj|j|j jjdƒkƒs¶t‚|jj|jksÑt‚dS(Niiitaxisii iiii(ii(i(ii(ii(i i(ii(iii(ii(iii(ii(i(ii( R ttakeRRR RRR7RRRR(Rtt0_takett1_takett2_takett2_take2((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_takeõs 033-cCsÆ|jjtddƒ}|jdks-t‚tj|j|jjkƒsQt‚tj|j|jjƒsrt‚|j dks‡t‚|j jtddƒ}|jd ks´t‚tj|j|j jkƒsØt‚tj|j|j jƒsùt‚|j |j j kst‚|j jtdd ƒ}|jd ksAt‚tj|j|j jkƒset‚tj|j|j jƒs†t‚|j j|jks¡t‚tj|j |j j ƒsÂt‚dS( sTest using a callable method.Rii iN(ii i(ii i(ii i(ii i(ii i(ii i( R t_applytnp_broadcast_toRRR RRRRRRR(Rt t0_broadcastt t1_broadcastt t2_broadcast((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyttest_broadcasts $!$!$!(t__name__t __module__t__doc__RRR$R+R0R5R?RGRLRPRWR](((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyRs      ) *   tTestArithmeticcBsáeZdZiidd6idd6idd6idd6fZdZd„Zejj d e j eeƒƒd „ƒZ ejj d e j eeƒƒd „ƒZ d „Zd „Zd„Zd„Zd„Zd„Zd„ZRS(s/Arithmetic on Time objects, using both doubles.RQiiitmintmaxtsortcCs®tjdddƒjdddƒ}tjddddd gƒ}t||d d d dƒ|_tjdddddgƒ}t||d d d dƒ|_|||_dS(NiPÃi´Ãi iiigš™™™™™¹?gV瞯Òlib/python2.7/site-packages/astropy/time/tests/test_methods.pyR s $skw, funccCsÁt|jd|ƒ|}t|jd|ƒ|}ttd|ƒ|j|}tj||kƒsrt‚tj||kƒst‚|j|jks¥t‚|j|jks½t‚dS(s  Test that np.argfunc(jd, **kw) is the same as t0.argfunc(**kw) where jd is a similarly shaped array with the same ordinal properties but all integer values. Also test the same for t1 which has the same integral values as jd. targN(tgetattrR RR RfRRR(Rtkwtfunctt0vtt1vtjdv((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_argfuncs*scCsmt|j|ƒ|}tt|ƒ|j|}tj|j|kƒsQt‚|j|jksit‚dS(s• Test that np.func(jd, **kw) is the same as t1.func(**kw) where jd is a similarly shaped array and the same integral values. N(RhRR RfRtvalueRR(RRiRjRlRm((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_funcs:scCs|jjƒdkst‚tj|jjddƒdkƒsEt‚tj|jjddƒdkƒsot‚tj|jjddƒdkƒs™t‚dS(NiRQii(R targminRR R(R((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_argminEs**cCs§|jjƒ|jjdks%t‚tj|jjddƒdkƒsOt‚tj|jjddƒdkƒsyt‚tj|jjddƒdkƒs£t‚dS(NiRQiiii(R targmaxRRR R(R((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_argmaxKs%**c CsXtj|jjƒtjdddddgƒkƒs<t‚tj|jjddƒtjdƒjdddƒkƒs~t‚tj|jjddƒtjdƒjddƒkƒs½t‚tj|jjddƒtjdddddgƒkƒsÿt‚tj|jjddƒtjdƒjd dƒdd…d fj ƒkƒsTt‚dS( NiiiiiRQii2iÿÿÿÿ(iiiii( R RR targsortReRR R6RR(R((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_argsortQs <B?BcCs©|jjƒ|jdks"t‚tj|jjdƒ|jdkƒsPt‚tj|jjdƒ|jdd…dfkƒsŠt‚tj|jjdƒ|jdd…dd…dfkƒsÍt‚|jjdƒjdksît‚|jjddtƒjdkst‚|jjdƒjd ks6t‚|jjddtƒjd ks]t‚|jjdƒjd ks~t‚|jjddtƒjd ks¥t‚dS( Niiiitkeepdims(iii(ii(iii(ii(iii(ii(iii(R RbRR RRtTrue(R((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyttest_minYs".:C!'!'!cCs|jjƒ|jd ks"t‚tj|jjdƒ|jdkƒsPt‚tj|jjdƒ|jdd…dfkƒsŠt‚tj|jjdƒ|jdd…dd…dfkƒsÍt‚|jjdƒjd ksît‚|jjdd tƒjd kst‚dS( NiÿÿÿÿiþÿÿÿiiiiiiRw(iÿÿÿÿiÿÿÿÿiþÿÿÿ(ii(iii(R RcRR RRRx(R((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyttest_maxes ".:C!cCsÀ|jjƒ|jjƒ|jjƒks1t‚tj|jjdƒ|jjdƒ|jjdƒkƒstt‚|jjdƒjdks•t‚|jjddtƒjdks¼t‚dS(NiiRwi(ii(iii( R tptpRcRbRR RRRx(R((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyttest_ptpms1C!c Cs¼tj|jjƒ|jdd…dd…dfkƒs@t‚tj|jjdƒ|jkƒsjt‚tj|jjdƒ|jkƒs”t‚tj|jjdƒ|jdd…dd…dfkƒs×t‚tj|jjdƒ|jdd…dd…d fjƒkƒs t‚tj|jjdƒdd…dd…df|jjdƒkƒslt‚tj|jjdƒdd…dd…df|jjdƒkƒs¸t‚dS( Niiiiiiÿÿÿÿ(iiiii(iiiii(iiiii( R RR RdRRRRbRc(R((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyt test_sortss@**C4LN(RbRcRd(R^R_R`Rtkwargst functionsRRAtmarkt parametrizet itertoolstproductRnRpRrRtRvRyRzR|R}(((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyRas1 **     cCs¾tdddddƒ}|j}tj|ƒ}t|jƒtjksNt‚|j ƒ}t|jƒtjksxt‚|j ƒ}t|jƒtjks¢t‚|j|jksºt‚dS(Ng€5è@RttaiRR( Rtut1R@tdeepcopyttypet delta_ut1_utcR tndarrayRR R(tttt_ut1t t_ut1_copyt t_ut1_flattent t_ut1_ravel((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyttest_regressions   ( R‚R@RAtnumpyR tRtutils.compat.numpyRRYRRaR(((s>lib/python2.7/site-packages/astropy/time/tests/test_methods.pyts    ÿd