ó šßÈ[c@`s]dZddlmZmZmZmZddlmZddlm Z ddl Z ddl Z ddl Z ddl ZddlmZddlmZdd lmZmZejejejgZejejejgZej ej!ej"gZ#ej$ej%ej&ej'd ej(fZ)d e*fd „ƒYZ+d „Z,d„Z-d„Z.d„Z/de*fd„ƒYZ0de*fd„ƒYZ1de*fd„ƒYZ2d„Z3d„Z4de*fd„ƒYZ5d„Z6d„Z7de*fd„ƒYZ8de*fd „ƒYZ9d!e*fd"„ƒYZ:d#e*fd$„ƒYZ;d%e*fd&„ƒYZ<d'e*fd(„ƒYZ=dS()u/ Test the Logarithmic Units and Quantities i(tabsolute_importtunicode_literalstdivisiontprint_functioni(tsix(tzipN(tassert_allclose(tassert_quantity_allclose(tunitst constantsitTestLogUnitCreationcB`s½eZd„Zejjdeeeƒƒd„ƒZ ejjdeƒd„ƒZ ejjde j ee ƒƒd„ƒZd„Zejjde j eejge ƒƒd „ƒZRS( cC`sgtjjtjƒdks!t‚tjjtjƒdksBt‚tjjtjƒdksct‚dS(u-Check logarithmic units are set up correctly.gš™™™™™¹?gÀiüÿÿÿN(tutdBttotdextAssertionErrortmag(tself((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyttest_logarithmic_units!s!!ulu_unit, lu_clscC`sCt|tjƒst‚t|ƒs*t‚|j|ks?t‚dS(N(t isinstanceR tUnitBaseRtcallablet_function_unit_class(Rtlu_unittlu_cls((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyttest_callable_units'sulu_unitcC`s7|ƒ}||jkst‚|j|ks3t‚dS(N(t_default_function_unitR(RRtlu((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyt.test_equality_to_normal_unit_for_dimensionless-s ulu_unit, physical_unitcC`s=||ƒ}|j|ks!t‚|j|jks9t‚dS(uuCreate a LogUnit subclass using the callable unit and physical unit, and do basic check that output is right.N(t physical_unitRt function_unitR(RRRtlu1((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyttest_call_units3s cC`sPtjtƒtjgƒWdQXtjtƒtjtjƒƒWdQXdS(N(tpytesttraisest TypeErrorR Rt ValueError(R((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyttest_call_invalid_unit<sulu_cls, physical_unitcC`s¶||ƒ}|j|ks!t‚|j|jks9t‚||dd|jƒ}|j|ksgt‚|jtjd|jƒksŒt‚tjtƒ||tj ƒWdQXdS(ujCreate a LogUnit subclass object for given physical unit, and do basic check that output is right.RiN( RRRRR tUnitR!R"R$tm(RRRRtlu2((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyttest_subclass_creationBs  %(t__name__t __module__RR!tmarkt parametrizeRtlu_unitst lu_subclassesRRt itertoolstproductt pu_sampleR R%R tLogUnitR)(((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyR s '  cC`sÀtdtjjdtjtjdtjtjƒtdtjjdtjtjdtjtj ƒtdtj jt j ƒtdtj jt j dtjdt jdƒdS(Ngš™™™™5Àgð?igÍÌÌÌÌLHÀig@g$@(RR tSTmagtphysicaltergtcmtstAAtABmagtHztM_boltctL_bol0tm_boltnptpitpc(((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyttest_predefined_magnitudesTs''cC`s|tjdƒtjkst‚tjdƒtjks<t‚tjdƒtjksZt‚tjdƒtjksxt‚dS(NuSTuABuBolubol(R RR4RR:R<R?(((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyt test_predefined_reinitialisation^scC`s¶tjjƒ¡tjtjjƒƒtjks7t‚tjtjjƒƒtjks^t‚tjtjjƒƒtjks…t‚tjtj jƒƒtj ks¬t‚WdQXdS(uEnsure roundtripping; see #5015N( R tmagnitude_zero_pointstenableR&R4t to_stringRR:R<R?(((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyt test_predefined_string_roundtripes '''cC`s~tjtjƒ}tjtjƒ}tjtjdƒ}||}||ksVt‚||ksht‚||kszt‚dS(u+Check __ne__ works (regresssion for #5342).iN(R RtJyRR(RR(tlu3tlu4((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyttest_inequalityns tTestLogUnitStringscB`seZd„ZRS(cC`s¨tjtjƒ}t|ƒdks*t‚t|ƒdksBt‚|jdƒdks]t‚tjt ƒ|jdƒWdQXtj ƒ}t|ƒdks¤t‚t|ƒdks¼t‚|jƒdksÔt‚tj tjd d tjƒ}t|ƒd ks t‚t|ƒd ks#t‚|jƒd ks;t‚tjtj ƒ}|jdƒd ksht‚|jdƒdksƒt‚|j ƒ|jdƒks¤t‚dS(u:Do some spot checks that str, repr, etc. work as expected.umag(Jy)uUnit("mag(Jy)")ugenericufitsNudexuUnit("dex(1)")udex(1)Riu 2 mag(Jy)uMagUnit("Jy", unit="2 mag")umag(ct)ulatexu3$\mathrm{mag}$$\mathrm{\left( \mathrm{ct} \right)}$(R RRItstrRtreprRGR!R"R$RtMagUnittctt _repr_latex_(RRR(RJRK((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyttest_str{s$ (R*R+RS(((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyRMystTestLogUnitConversioncB`seZejjdejeeƒƒd„ƒZ ejjdeƒd„ƒZ ejjdejeeeƒƒd„ƒZ d„Z d„Z RS(ulu_unit, physical_unitcC`s±||ƒ}|j|ƒs!t‚|j|dƒdks?t‚|j|ƒsTt‚|j|dƒdksrt‚tjd|ƒ}|j|ƒsšt‚|j|dƒdks¸t‚|j|ƒsÍt‚t|j|dƒdddƒtjdddƒ}t|j||j||ƒƒ|ddƒtj}|j|ƒ sKt‚t j tj ƒ|j|ƒWd QX|j|ƒ s‡t‚t j tj ƒ|j|ƒWd QXd S( ufCheck various LogUnit subclasses are equivalent and convertible to their non-log counterparts.ggð?g @gÀ?tatolgV瞯Ò"%  "#upowerigà?iic C`s´tjtjddƒtjƒ}|dkrPtj||dkƒsŸt‚nO|dkr~tj|||kƒsŸt‚n!tjtj ƒ ||WdQXtjtjdƒƒ}||}|dkr|j tj ksåt‚tj|j dkƒs°t‚nª|dkr0tj||kƒs°t‚n€t |t|ƒƒ sLt‚|j |j j|kskt‚tjtjƒƒ.tjtj ƒ|jtj ƒWdQXWdQXdS(uÝCheck that raising LogQuantities to some power is only possible when the physical unit is dimensionless, and that conversion is turned off when the resulting logarithmic unit (say, mag**2) is incompatible.gð?g@iiNg$@(R R£R@RuRIR°RR!R"RYRyR^R[RRrRRsRtR (RR€RŸRÅR|((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyRzs&" " "  ! cC`sDtjtjddƒtjƒ}tjtƒ ||WdQXdS(Ngð?g@(R R£R@RuRIR!R"R#(RRŸ((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyttest_error_on_lq_as_poweršs"uothercC`s“tjtjddƒtjƒ}d|}tjtjƒ ||WdQXtjtjƒ ||WdQXtjtjƒ ||WdQXdS(Ngð?g$@g®Gáz®ó?(R R£R@RuRIR!R"RY(RR‚RŸR¤((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyRƒŸs" g®Gáz®ó?g¸…ëQ¸@gš™™™™™ @g= ×£p=@g®Gáz®@g…ëQ¸@g@cC`s×tjtjddƒtjƒ}|jt|jdtjƒdtj ƒƒ}||}t |j |j |ƒ||}t |j |j |ƒ||}t |j |j |ƒ||}t |j ||j ƒdS(uœCheck that addition/subtraction with quantities with magnitude or MagUnit units works, and that it changes the physical units appropriately.gð?g$@u physical_unitRºN( R R£R@RuRIR R…RyR^RtRR5(RR‚RŸtother_physicaltlq_sftlq_srtlq_dftlq_dr((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyR‹¬s"     cC`søtjtjƒ}tjtjddƒ|ƒ}tjtjƒ||7}WdQXtj |j tjddƒkƒs}t ‚|j |ks’t ‚tjtjƒ||8}WdQXtj |j tjddƒkƒsßt ‚|j |ksôt ‚dS(Ngð?g$@( R RRIR£R@RuR!R"RYR°R[RRy(RR‚RRÏ((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyt-test_inplace_addition_subtraction_unit_checksÅs**cC`s­tjtjddƒtjƒ}|jt|jdtjƒdtj ƒƒ}|j ƒ}||7}t |j |j |ƒ|j ƒ}||8}t |j |j |ƒdS(u¤Check that inplace addition/subtraction with quantities with magnitude or MagUnit units works, and that it changes the physical units appropriately.gð?g$@u physical_unitRºN( R R£R@RuRIR R…RyR^RttcopyRR5(RR‚RŸRÕRÖRØ((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pyt!test_inplace_addition_subtractionÕs"     cC`sÌtjdddtjdtjdƒ}tj|ƒ}dtj}d|}||}|jjtj tj tj ƒs…t ‚tj |j|jdtjdtjddƒdksÈt ‚d S( u@For fun, a more complicated example of addition and subtraction.uDMgð?g@g$@ig@gY@gV瞯ÒRþt __future__RRRRtexternRtextern.six.movesRR“R0R!tnumpyR@t numpy.testingRt tests.helperRtRR R R=RRtdecibelR.tDexUnitRPt DecibelUnitR/R¯R£tDecibelR·R^R'RXR8RIR2RvR RCRDRHRLRMRTRqR–R›RœRÂRÃRÄRÎRÒRâRìRú(((sClib/python2.7/site-packages/astropy/units/tests/test_logarithmic.pytsB"    )5  Z  T   ®)/