a ߙfb^ @s6dZddlZddlmZddlZddlZddlmZddl m Z ddl m Z ddlmZdd Zd d Zd d ZddZddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Z d*d+Z!d,d-Z"d.d/Z#d0d1Z$d2d3Z%d4d5Z&d6d7Z'd8d9Z(d:d;Z)dd?Z+d@dAZ,dBdCZ-dDdEZ.dFdGZ/e0Z1e j23D],Z4e5e4e j6rhe5e4e j7she18e4qhej9j:dHe;e1e/dIe/dJdKdLZj23D]:Z4e5e4e j6re5e4e j7se4e j>j?kre=8e4qej9j:dHe;e=e/dIe/dJdMdNZ@e0ZAe jBj23D]:Z4e5e4e j6r:e5e4e j7s:e4e jBj?kr:eA8e4q:ej9j:dHe;eAe/dIe/dJdOdPZCdQdRZDdSdTZEdUdVZFdWdXZGdYdZZHd[d\ZId]d^ZJd_d`ZKdadbZLdcddZMdedfZNdgdhZOdidjZPdkdlZQdmdnZRdodpZSdqdrZTej9:dHe jUdsgdtduidve Vdwe jWe jXgdxdyZYdzd{ZZd|d}Z[d~dZ\ddZ]ddZ^ddZ_ddZ`ddZaddZbddZcddZdddZeddZfddZgddZhddZiddZjddZkddZlddZmddZnddZoddZpddZqddZrddZsddZtdS)z'Regression tests for the units package.N)Fraction)assert_allcloseunits) constants)utilscCs0ttjtjusJtdtj}|tdtjgdgksBJt||usTJtd|tdtjgdgkszJtjdd|dtd}|tdtjgdgksJtdtjksJtd tjksJtjtjksJtd |ksJtdtdggksJttjks,JdS) N$@gY@foo namespaceg@mz10 m)uUnitr CompositeUnitlocalsZdimensionless_unscaledZone)Z ten_meterr r=lib/python3.9/site-packages/astropy/units/tests/test_units.pytest_initialisations&rcCsjtjtdd}t|jdts$Jtjtdd}t|jdtsHJtjd}t|jdtsfJdS)Nr rr UUUUUU?)rrr isinstancepowersfloatxrrrtest_invalid_power%s  rcCstjtjkrJdSNrrsrrrrtest_invalid_compare1sr"cCstjtjddksJdS)Nr i)rh_get_converterr!rrrr test_convert5sr%cCsttj tjtjdWdn1s20Yttj&tjtjtjdWdn1st0YdSNr )pytestraisesr UnitsErrorcmtor!rrrrrtest_convert_fail9s.r,cCstjtjtjtjddks&Jtjtjtjdks@Jtjtjtjtjdks`Jtjdtjddks|JdS)Nr $r rMbP?)rr*r!r#r$rHzr+rrrrtest_composite@s& r1cCsttjdksJdS)Nr*)strrr*rrrrtest_strIsr3cCsttjdksJdS)Nz Unit("cm"))reprrr*rrrr test_reprMsr5cCstjjtjusJtjjjdks$Jtjjjtjgks:JtjjdkrVtjjtjgksZJttjjjdtjjjtjgksJtj dt d}|j|usJtj ddtj t d}|jjdksJ|jjtj gksJ| jd ksJ| jtj gksJdS) Ng@@?gHzlT6+@blar blabla ri)rrZ representskmscalebasesRyrZeVdef_unitrZhr decomposer!)r7r8rrrtest_representsQs r@cCs|ttjtjdttjtjdttjtjdttjtjdttj tj dttj tj ddS)Nr/r.gư>g(벥U>g-DT!@g-DT!)@) rrkpcr+MpcyrMyrZAUpccycleradspatsrrrrrtest_units_conversionas rJcCs0tjtjtddtjtjtjddS)Nr r )rrArCrrDAAergrrrrtest_units_manipulationjsrNcCstjtjksJdSr)rr=r?rrrrtest_decomposepsrOcCs4dtjdtj}|jjdks$Jt|jddS)zh Issue #1150: Test for conversion of dimensionless quantities to the SI system r6 dimensionlessr/N)rrArBunit physical_typersiZtestunitrrrtest_dimensionless_to_sitsrUcCs4dtjdtj}|jjdks$Jt|jddS)zi Issue #1150: Test for conversion of dimensionless quantities to the CGS system r6rPr/N)rrr:rQrRrcgsrTrrrtest_dimensionless_to_cgssrWcCsBtjtjddtjdddWdn1s40YdS)NFOOmatchwarn parse_strict)r'warnsr UnitsWarningrrrrrtest_unknown_unitsr`cCstjtjdd&tddks(JWdn1s<0YtttjdddWdn1st0Ytjddd}|dksJdS) Nz8'm/s/kg' contains multiple slashes, which is discouragedrYzm/s/kgz m / (kg s)Zvounitformatzkg(3/10) * m(5/2) / szkg(3/10) m(5/2) / s)r'r^rr_r to_stringr( ValueErrorrrrrtest_multiple_soliduss 4 ,recCsptjddd}t|tjsJ|jdks,Jtjddd}||ksFJ||sTJtjddd}||ksnJ||r|J|dksJ|tjksJ|dksJ|dtjfvsJtt | |Wdn1s0Y| d}| d}tt tjdddWdn1s,0Ytt tdWdn1sb0YdS)NrXsilentr\ZBARlatexrVstrict) rrrZUnrecognizedUnitname is_equivalentrr'r(rdr$rc TypeError)rQunit2Zunit3_rrrtest_unknown_unit3s*     (   . rncCs<ttgdtjWdn1s.0YdS)N)abc)r'r(rkrrrrrrtest_invalid_scales rrcCs"tjdddd}|jdksJdS)Nz 10+22/cm2cdsrf)rbr]gMD)rrr;rQrrrtest_cds_powersrucCsxtjdtjdtd}dtvs&Jt|"dtjvsDJWdn1sX0YdtjvstJdS)Nr rr )rr>rradd_enabled_unitsget_current_unit_registryregistry)r rrr test_registers  0rycCs$tjtj}|tjtjd}dSr&)rr*r!Zin_unitsrEZhour)Z speed_unitrmrrr test_in_unitss rzcCstjtjtdksJdSr&)rrrrrrrtest_null_unitsr{cCs,tjddusJtjddus(JdS)Nr FrET)rrrjrrrrtest_unrecognized_equivalencysr|cCsLtjtjdd(tjtjtjdWdn1s>0YdS)Nzlength.+ are not convertiblerYr )r'r(rr)rLr+r#r!rrrrtest_convertible_exceptionsr}cCsBtjtjddtjtjWdn1s40YdS)Nz(length. and .+time.+ are not convertiblerY)r'r(rr)rr+r!rrrrtest_convertible_exception2sr~cCsHGddd}ttt|Wdn1s:0YdS)Nc@s eZdZdS)ztest_invalid_type..AN)__name__ __module__ __qualname__rrrrAsr)r'r(rkrr)rrrrtest_invalid_types rcCsptjtjtjsJtjjtjjd}|djdtjusBJtjjtjjd}|djdtjuslJdS)z Issue #599 rrN)rrIrjrGcomposerVr<__dict__Zresultsrrrtest_steradians rcCspddlm}ddlm}|jjj|jd}|jt j t j t j gksDJ|j tddddgks^J|jd kslJd S) z From issue #576 rrV)er<rr ?r6N) astropy.unitsrVastropy.constantsrZesurQr?r<_basesrr*gr!_powersr_scale)rVrdrrrtest_decompose_basess   rcCs@tjtjtj}|}t|djttjtjgksz%test_equiv_compose..)rrrZspectralany)rrrrtest_equiv_composesrcCs>ttjtjjgdWdn1s00YdSNr)r'r(rr)rrrrrrtest_empty_composesrcCst|Sr)r2rtrrr _unit_as_strsrrQ)key)ZidscCs^|}d}|D]<}t|jr:|jd|urPd}qRqt|jdkrd}qRq|sZJdS)NFrT)r?rlenr<)rQZ composed_listfoundrrrrtest_compose_roundtrip,s  rcs8tj}fdd|Ds"J|djks4JdS)Ncsg|]}|qSrrjrrtrr Irz*test_compose_cgs_to_si..r) to_systemrrS)rQrSrrtrtest_compose_cgs_to_siEs rcsfztj}Wn(tjy8tjjvr2nYn*0fdd|DsPJ|djksbJdS)Ncsg|]}|qSrrrrtrrrcrz*test_compose_si_to_cgs..r)rrrVr)rr?r<)rQrVrrtrtest_compose_si_to_cgsWsrcCs@tjjtjusJtjtjtjdtjtjjjvsJdS)Nr rr)rZParrVrZBarrrrr test_to_cgsus"rcCs0ddlm}tjj|jdjd|jus,JdS)Nrrr)rrVrrr?r<rr*rrrrtest_decompose_to_cgszs rcCsvtjtjdtj}|jtjtjtjgd}t|dks>J|djtjtjtjgks\J|djgdksrJdS)Nr rr r)r ) rrr!rrNrrr)rQresultrrrtest_compose_issue_579s rcCsLtjjtjfd}|djdtjus*J|djdks0YdS) Nrrr6T)rZinclude_prefix_unitsr/gObn]?F) rrrr<r;r:r!rErDrr'r(r)rrrrtest_compose_prefix_unitsrcCs0tjtj}t|jtjtjgddks,JdS)Nrr )rrr!rrrrtrrrtest_self_composes rcCsFtj}ttj |jtjgdWdn1s80YdSr)rrr'r(r)rrrtrrrtest_compose_failedsrcCstjdtjd}|}|D]}||ks Jq |jtjd}|D]}||ksLJqL|jtjd}|D]}||ksxJqxdS)Nr?r)rrrCrr?rVrS)rZfactoredrQrrrtest_compose_fractional_powerssrcCstj}t|djdks J|djdtjus8Jtjd}|djdtjtjfvsdJtj }|djdtjusJdS)Nrr r) rlrrr<r!r0ZBqr=r?rrrrtest_compose_best_unit_firsts rcCsRtjtjdtjdtjdtjd}|jtjjd}t |dksNJdS)Nrg@rr rr ) rrr!ZaurCrIrrVr<r)newrrrrtest_compose_no_duplicatess.rcCs.dtjtjd}|tjtjddS)z Issue #672 l@=7M.cr N)rZM_pr*r+ZM_sunrE)Zsigmarrr test_long_intsrcCsLdD]B}dD]8}dD].}tjgd|||d}tjtj|qq qdS)z Regression test for #744 A logic issue in the units code meant that big endian arrays could not be converted because the dtype is '>f4', not 'float32', and the code was looking for the strings 'float' or 'int'. )<>)if)48)r r r)ZdtypeN)npZarrayrrr+r*)ZendianZntypebyterrrrtest_endian_independences rcCsdtjjjtjksJdS)z Issue #863 r N)rdegreerSrQrGrrrrtest_radian_basesrcCs ttdrJttdsJdS)NasZ attosecond)hasattrrrrrr test_no_assrcCsftjjddgksJtjjddgks(Jtjjdgks:JttjjttjjttjjBksbJdS)Nctcount)rrnamesZ short_namesZ long_namesrZphrrrrtest_no_duplicates_in_namessrcCsttj}t|}|tjus$Jtjdgddid}dtjvsJJt|}t|}||usjJ|jdgkszJ| ddksJdtjvsJt |gBt|}dtjvsJt|}||usJWdn1s0Yt gvdtjvsJt|}||us4J|jdgksFJ| ddksZJdtjvsnJWdn1s0YdtjvsJdS)Nr bazrra) pickledumpsrrloadsIrreducibleUnitrwrxrZget_format_namerv)potherZnew_unitZ new_unit_copyrrr test_picklings0       *  4rcCs8ttj}td}|tjus"Jttj|ks4JdS)zWe cannot really test between sessions easily, so fake it. This test can be changed if the pickle protocol or the code changes enough that it no longer works. sastropy.units.core_recreate_irreducible_unithIrreducibleUnit](mmetereR}(_names](hhe _short_names]ha _long_names]ha_format}__doc__ meter: base unit of length in SIub.N)hashrrrr)Zhash_mrQrrrtest_pickle_between_sessions$s   rr rrraZm_per_scCst|}|jdusJtt|}||us2J|jdus@Jt||ksPJt|gVtt|}t|tjr||usJn ||usJt||ksJWdn1s0YdS)a Tests private attribute since the problem with _hash being pickled and restored only appeared if the unpickling was done in another session, for which the hash no longer was valid, and it is difficult to mimic separate sessions in a simple test. See gh-11872. N) r_hashrrrrrvrr)rQZ unit_hashZ unit_copyZunit_refrrr'test_pickle_does_not_keep_memoized_hash9s    rcCs"tjddd}tt|dS)z Issue #2047 Zasdfrfr\N)rrrrr)rorrrtest_pickle_unrecognized_unitVsrcCs>tt tjdtjdWdn1s00YdS)Nrr )r'r(rdrr>rrrrrtest_duplicate_define^s rcCs.ddlm}|}t|jt|jks*JdS)Nr)rw)Zastropy.units.corerwrZ all_unitsZnon_prefix_units)rwrxrrrtest_all_unitscs rcCstjtjdksJdS)Nrg)rrZ _repr_latex_rcrrrrtest_repr_latexisrcCs@tjdtjdtjksJtjddtjtjksJdSzACheck that multiplication with strings produces the correct unit.rNrr*rZu1usrrr$test_unit_multiplication_with_stringsrcCsBtj}d}||t||ks$J|||t|ks>JdSrrrrrrtest_unit_division_by_stringsrcCs$tjtjjtjtjjks JdS)z See #1616.N)rrZJyr<rrrrtest_sorted_basessrcCsDtjtjusJtjtjus Jtjtjus0Jtjtjus@JdS)z See #1543N)rZMbitZMbZmegabitZMbyteZMBZmegabyterrrr test_megabitsrcCs:td}td}ttd||gddgdks6JdS)z See #1576znrad/szHz(1/2)r rznrad / (Hz(1/2) s)N)rrr2r)Zunit1rlrrr#test_composite_unit_get_format_names   rcCs ddlm}|tjtjddS)Nr!assert_follows_unicode_guidelines)Z roundtrip)Zastropy.tests.helperrrrrrrrrtest_unicode_policys rc CsPdD]F\}}tjtd|dt|Wdq1s@0YqdS)N))Zmicronsmicron)z s/micronsr)Mr)ZmetreZmeter)Z angstromszAngstrom or angstrom)Z milimeterZ millimeter)u ångströmz*Angstrom, angstrom, mAngstrom or mangstrom)ZkevzEV, eV, kV or keVz Did you mean rY)r'r(rdrr)searchZmatchesrrrtest_suggestionss rcCsztjtjdd}td}Wdn1s00Y|tjtjdtjdtjdksfJt |dksvJdS)z See #1911.zmultiple slashesrYzerg /s /cm**2 /angstromNrrr ) r'r^rr_rrMr!r*Zangstromr)wrrrrtest_fits_hst_units(,rcCs$tjtjusJtjtjus JdS)zBRegression test for https://github.com/astropy/astropy/issues/3753N)rZfbarnZ femtobarnZpbarnZpicobarnrrrrtest_barn_prefixessrcCsdtj}ddtjtjtj}dtjtj}tjd|d|jtdd|}| d}tjd|d|tdd|}| d}tjd|d|jd |}| d}tjd|d|d |}| d}t ||t ||t ||tj d }t |j d ts"Jtj d }t |j d tsBJ|j d jdksXJ|j d jd ksnJtjtddtjtdd}t |j d tsJ|j d td dksJtjdd}t |j d tsJ|j d tddksJdS)z See #2069geAr6gQ@r r rrErgbeF?rg۶m۶m?gg]tEݿN)rZMsunr:r!rBrqrrVrr+rrrrr numerator denominatorr*ZTeV)rZtHZvcrZv1Zv2Zv3Zv4rrrtest_fractional_powerss4 & $          rcCs8tjd}t|jdsJ|dtjks4JdS)Nrimagr )rZmagrr?r;)Zsqrt_magrrr test_sqrt_mags rcCs.tjjtdgdd}tjj|gddS)Nr!rr)rr!rr)Zcomposite_unitrrrtest_composite_composesrcCstjtjsJdSr)rrrjbitrrrrtest_data_quantitiessrcCs tjdkrJtjdksJdSr)rrrrrrtest_compare_with_nonesrcCs8td}t|tsJ|jdks&J|jdks4JdS)Ng?rr)rZvalidate_powerrrrr)Zfracrrr#test_validate_power_detect_fraction s rc Csdtjdtj}dtj}d}dtjtjd}d}d}d }d tj}dtj|ttj}|| ||||tj |d} || d } | j j d ksJ| tjdS) Ng(\?r gv@iBg?g& .>rrgaa?gyy?g.AgZlength) rr*rZsolMassZsolRadrqrZsqrtZpirrQrRr+) ZkappaZr_0qZrho_0yZbetaroZm_ht1t2rrrr'test_complex_fractional_rounding_errorss  ( r cCsTtjdtdd}t|jdts(J|jdjdks0Yddlm}|$|jtj vsrJWdn1s0YdS)Nr)rs)imperial) rrsenableZgeoMassrrZfind_equivalent_unitsr Zinchr)rsr rrrtest_enable_unit_groupings0s   2  rcCsddlm}t|jD]}|\}}}}}|jdkr>|sJq|jdkrR|sJq|jdkrf|sJq|jdkr~|dksJq|jdkr|dksJq|jd kr|d ksJqd S) z Test for a few units that the unit summary table correctly reports whether or not that unit supports prefixes. Regression test for https://github.com/astropy/astropy/issues/3835 r) astrophysZlyrrEZbarnrFZNorHZvoxZYesN)rrrZ_iter_unit_summaryrri)rZsummaryrQrmprefixesrrrtest_unit_summary_prefixes;s          rcCs\tjdtjd}|d}|jtjtjgks2J|jddgksDJ|tjtjksXJdS)zTest that order of bases is changed when raising to negative power. Regression test for https://github.com/astropy/astropy/issues/8260 r gr rN)rrr!r<r)Zm2s2Zspmrrrtest_raise_to_negative_powerVs r)u__doc__rZ fractionsrr'ZnumpyrZ numpy.testingrZastropyrrrrqrrrrr"r%r,r1r3r5r@rJrNrOrUrWr`rernrrruryrzr{r|r}r~rrrrrrrrZCOMPOSE_ROUNDTRIPrvaluesvalrZUnitBaseZ PrefixUnitaddZmarkZ parametrizesortedrZCOMPOSE_CGS_TO_SIrVZdeg_CrZCOMPOSE_SI_TO_CGSrSrrrrrrrrrrrrrrrrrrrrrr!rrrrrrrrrrrrrrrrrrrrrrrr r rrrrrrrs           !           '   (