U e{j.b@szdZddlZddlmZddlZddlZddlmZddl 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`ddZaddZbddZcddZdddZeddZfddZgddZhddZiddZjddZkddZlddZmddZnddZoddZpddZqddZrddZsddZtej9:ddddddddddddddddddddddddgddɄZudS)z'Regression tests for the units package.N)Fraction)assert_allclose) constantsunits)utilscCs0ttjtjksttdtj}|tdtjgdgksBtt||ksTttd|tdtjgdgkszttjdd|dtd}|tdtjgdgksttdtjksttd tjksttjtjksttd |ksttdtdggkstttjks,tdS) N$@gY@foo namespaceg@mz10 m)uUnitrAssertionError CompositeUnitlocalsZdimensionless_unscaledone)Z ten_meterr r]/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/units/tests/test_units.pytest_initialisations&rcCsjtjtdd}t|jdts$ttjtdd}t|jdtsHttjd}t|jdtsftdS)Nr rr UUUUUU?)rrr isinstancepowersrfloatxrrrtest_invalid_power$s  r cCstjtjkrtdSNrrsrrrrrtest_invalid_compare0sr$cCstjtjddkstdS)Nr i)rh_get_converterr#rrrrr test_convert4sr'c CsZttjtjtjdW5QRXttjtjtjtjdW5QRXdSNr )pytestraisesr UnitsErrorcmtor#rrrrrtest_convert_fail8sr.cCstjtjtjtjddks&ttjtjtjdks@ttjtjtjtjdks`ttjdtjddks|tdS)Nr $r rMbP?) rr,r#r%r&rrHzr-rrrrtest_composite?s& r3cCsttjdkstdS)Nr,)strrr,rrrrrtest_strHsr5cCsttjdkstdS)Nz Unit("cm"))reprrr,rrrrr test_reprLsr7cCstjjtjksttjjjdks$ttjjjtjgks:ttjjdkrVtjjtjgksZtttjjjdtjjjtj gksttj dt d}|j|ksttj ddtj t d}|jjdkst|jjtj gkst| jd kst| jtjgkstdS) N@@?gHzlT6+@blar blabla ri)rrZ representsrkmscalebasesRyrZeVdef_unitrhr decomposer#)r:r;rrrtest_representsPs rDcCs|ttjtjdttjtjdttjtjdttjtjdttj tj dttj tj ddS)Nr1r0ư>g(벥U>g-DT!@g-DT!)@) rrkpcr-MpcyrMyrZAUpccycleradspatsrrrrrtest_units_conversion`s rOcCs0tjtjtddtjtjtjddS)Nr r )rrFrHrrIAAergrrrrtest_units_manipulationisrScCstjtjkstdSr!)rr@rCrrrrrtest_decomposeosrTcCs4dtjdtj}|jjdks$tt|jddS)zh Issue #1150: Test for conversion of dimensionless quantities to the SI system r9 dimensionlessr1N)rrFrGunit physical_typerrsiZtestunitrrrtest_dimensionless_to_sissrZcCs4dtjdtj}|jjdks$tt|jddS)zi Issue #1150: Test for conversion of dimensionless quantities to the CGS system r9rUr1N)rrr=rVrWrrcgsrYrrrtest_dimensionless_to_cgssr\c Cs.tjtjddtjdddW5QRXdS)NFOOmatchwarn parse_strict)r)warnsr UnitsWarningrrrrrtest_unknown_unitsrec Csxtjtjddtddks(tW5QRXtttjdddW5QRXtjddd}|dksttdS) Nz8'm/s/kg' contains multiple slashes, which is discouragedr^zm/s/kgz m / (kg s)Zvounitformatzkg(3/10) * m(5/2) / szkg(3/10) m(5/2) / s) r)rcrrdr to_stringrr* ValueErrorrrrrtest_multiple_soliduss  rjc Cs0tjddd}t|tjst|jdks,ttjddd}||ksFt||sTttjddd}||ksnt||r|t|dkst|tjkst|dkst|dtjfkstt t | |W5QRX| d}| d}t t tjdddW5QRXt t tdW5QRXdS)Nr]silentraZBARlatexr[strict)rrrZUnrecognizedUnitrname is_equivalentrr)r*rir&rh TypeError)rVunit2Zunit3_rrrtest_unknown_unit3s*         rsc Cs*ttdddgtjW5QRXdS)Nabc)r)r*rprrrrrrtest_invalid_scales rwcCs"tjdddd}|jdkstdS)Nz 10+22/cm2cdsrk)rgrbgMD)rrr>rrVrrrtest_cds_powersrzc Csdtjdtjdtd}dtks&tt|dtjksDtW5QRXdtjks`tdS)Nr rr )rrArrradd_enabled_unitsget_current_unit_registryregistry)r rrr test_registers  r~cCs$tjtj}|tjtjd}dSr()rr,r#Zin_unitsrJhour)Z speed_unitrrrrr test_in_unitss rcCstjtjtdkstdSr()rrrrrrrrtest_null_unitsrcCs,tjddksttjddks(tdS)Nr FrJT)rrrorrrrrtest_unrecognized_equivalencysrc Cs8tjtjddtjtjtjdW5QRXdS)Nzlength.+ are not convertibler^r )r)r*rr+rQr-r%r#rrrrtest_convertible_exceptionsrc Cs.tjtjddtjtjW5QRXdS)Nz(length. and .+time.+ are not convertibler^)r)r*rr+rr-r#rrrrtest_convertible_exception2src Cs4Gddd}ttt|W5QRXdS)Nc@s eZdZdS)ztest_invalid_type..AN)__name__ __module__ __qualname__rrrrAsr)r)r*rprr)rrrrtest_invalid_types rcCsptjtjtjsttjjtjjd}|djdtjksBttjjtjjd}|djdtjksltdS)z Issue #599 rrN) rrNrorLrcomposer[r?__dict__resultsrrrtest_steradians rcCspddlm}ddlm}|jjj|jd}|jt j t j t j gksDt |jtddddgks^t |jd kslt d S) z From issue #576 r)er[r?rr ?r9N)astropy.constantsr astropy.unitsr[ZesurVrCr?_basesrr,gr#r_powersr_scale)rr[drrrtest_decompose_basess   rcCs<tjtjtj}|}t|djtjtjhks8tdS)Nr) rcdrNZWbrsetrZlmr)complexcomposedrrrtest_complex_compose srcCs,tjjtd}tdd|Ds(tdS)N)Z equivalenciescss|]}tjg|jkVqdSr!)rr2r?.0rrrr sz%test_equiv_compose..)rrrZspectralanyr)rrrrtest_equiv_composesrc Cs*ttjtjjgdW5QRXdSNr)r)r*rr+rrrrrrtest_empty_composesrcCst|Sr!)r4ryrrr _unit_as_strsrrV)key)ZidscCs^|}d}|D]<}t|jr:|jd|krPd}qRqt|jdkrd}qRq|sZtdS)NFrT)rCrlenr?r)rVZ composed_listfoundrrrrtest_compose_roundtrip,s  rcs8tj}fdd|Ds"t|djks4tdS)Ncsg|]}|qSrrorryrr Nsz*test_compose_cgs_to_si..r) to_systemrrXr)rVrXrryrtest_compose_cgs_to_siIs rcshztj}Wn*tjk r:tjjkr4nYn*Xfdd|DsRt|djksdtdS)Ncsg|]}|qSrrrryrrrksz*test_compose_si_to_cgs..r)rrr[r+rrCr?r)rVr[rryrtest_compose_si_to_cgs^srcCs@tjjtjk sttjtjtjdtjtjjjkstdS)Nr rr)rZParr[rZBarrrrrr test_to_cgs}s"rcCs0ddlm}tjj|jdjd|jks,tdS)Nrrr) rr[rrrCr?rr,rrrrrtest_decompose_to_cgss rcCsxtjtjdtj}|jtjtjtjgd}t|dks>t|djtjtjtjgks\t|dj dddgksttdS)Nr rr r) rrr#rrNrrrr)rVresultrrrtest_compose_issue_579s rc Cs6tjjtjfd}|djdtjks*t|djdksr=r#rJrIrr)r*r+rrrrtest_compose_prefix_unitsrcCs0tjtj}t|jtjtjgddks,tdS)Nrr )rrr#rrrrryrrrtest_self_composes rc Cs2tj}ttj|jtjgdW5QRXdSr)rrr)r*r+rrryrrrtest_compose_failedsrcCstjdtjd}|}|D]}||ks tq |jtjd}|D]}||ksLtqL|jtjd}|D]}||ksxtqxdS)Nr?r)rrrHrrCrr[rX)rZfactoredrVrrrtest_compose_fractional_powerssrcCstj}t|djdks t|djdtjks8ttjd}|djdtjtjfksdttj }|djdtj kstdS)Nrr r) rlrrr?rr#r2ZBqr@rCrrrrtest_compose_best_unit_firsts rcCsRtjtjdtjdtjdtjd}|jtjjd}t |dksNt dS)Nrg@rr rr ) rrr#aurHrNrr[r?rr)newrrrrtest_compose_no_duplicatess.rcCs.dtjtjd}|tjtjddS)z Issue #672 l@=7M.cr N)rZM_pr,r-ZM_sunrJ)sigmarrr test_long_intsrcCsNdD]D}dD]:}dD]0}tjdddg|||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)48r r r)ZdtypeN)nparrayrrr-r,)ZendianZntypebyterrrrtest_endian_independences rcCsdtjjjtjkstdS)z Issue #863 r N)rdegreerXrVrLrrrrrtest_radian_basesrcCs ttdrtttdstdS)NasZ attosecond)hasattrrrrrrr test_no_assrcCsftjjddgksttjjddgks(ttjjdgks:tttjjttjjttjjBksbtdS)Nctcount)rrnamesrZ short_namesZ long_namesrphrrrrtest_no_duplicates_in_namessrc Cs|ttj}t|}|tjks$ttjdgddid}dtjksJtt|}t|}||k sjt|j dgkszt| ddkstdtjkstt |g8t|}dtjkstt|}||kstW5QRXt gldtjkstt|}||k s t|j dgks2t| ddksFtdtjksZtW5QRXdtjksxtdS)Nr bazrrf) pickledumpsrrloadsrIrreducibleUnitr|r}rZget_format_namer{)potherZnew_unitZ new_unit_copyrrr test_picklings0         rcCs8ttj}td}|tjks"tttj|ks4tdS)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)hashrrrrr)Zhash_mrVrrrtest_pickle_between_sessions-s   rr rrrfZm_per_sc Cst|}|jdk sttt|}||k s2t|jdks@tt||ksPtt|gLtt|}t|tj r||kstn ||k stt||kstW5QRXdS)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_hashrrrrrr{rr)rVZ unit_hashZ unit_copyZunit_refrrr'test_pickle_does_not_keep_memoized_hashCs    rcCs"tjddd}tt|dS)z Issue #2047 ZasdfrkraN)rrrrr)rtrrrtest_pickle_unrecognized_unitasrc Cs*tttjdtjdW5QRXdS)Nrr )r)r*rirrArrrrrtest_duplicate_defineis rcCs.ddlm}|}t|jt|jks*tdS)Nr)r|)Zastropy.units.corer|rZ all_unitsZnon_prefix_unitsr)r|r}rrrtest_all_unitsns rcCstjtjdkstdS)Nrl)rrZ _repr_latex_rhrrrrrtest_repr_latexusrcCs@tjdtjdtjksttjddtjtjkstdSzACheck that multiplication with strings produces the correct unit.rNrr,rru1usrrr$test_unit_multiplication_with_stringsrcCsBtj}d}||t||ks$t|||t|ks>tdSrrrrrrtest_unit_division_by_stringsrcCs$tjtjjtjtjjks tdS)z See #1616.N)rrZJyr?rrrrrtest_sorted_basessrcCsDtjtjksttjtjks ttjtjks0ttjtjks@tdS)z See #1543N)rZMbitZMbrZmegabitZMbyteMBZmegabyterrrr test_megabitsrcCs:td}td}ttd||gddgdks6tdS)z See #1576znrad/szHz(1/2)r rznrad / (Hz(1/2) s)N)rrr4rr)Zunit1rqrrr#test_composite_unit_get_format_names  rcCs ddlm}|tjtjddS)Nr!assert_follows_unicode_guidelines)Z roundtrip)Zastropy.tests.helperrrrrrrrrtest_unicode_policys rc Cs<dD]2\}}tjtd|dt|W5QRXqdS)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 r^)r)r*rirr)searchmatchesrrrtest_suggestionss rc Csftjtjdd}td}W5QRX|tjtjdtjdtjdksRt t |dksbt dS)z See #1911.zmultiple slashesr^zerg /s /cm**2 /angstromrrr N) r)rcrrdrrRr#r,Zangstromrr)wrrrrtest_fits_hst_units,rcCs$tjtjksttjtjks tdS)zBRegression test for https://github.com/astropy/astropy/issues/3753N)rZfbarnZ femtobarnrZpbarnZpicobarnrrrrtest_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"ttj d }t |j d tsBt|j d jdksXt|j d jd ksnttjtddtjtdd}t |j d tst|j d td dksttjdd}t |j d tst|j d tddkstdS)z See #2069eAr9gQ@r r rrJrgbeF?rg۶m۶m?gg]tEݿN)rZMsunr=r#rGrvrr[rr-rrrrrr numerator denominatorr,ZTeV)rZtHvcrZv1Zv2Zv3Zv4rrrtest_fractional_powerss4 & $          rcCs8tjd}t|jdst|dtjks4tdS)Nrimagr )rmagrrCr>r)Zsqrt_magrrr test_sqrt_mags rcCs.tjjtdgdd}tjj|gddS)Nr#rr)rr#rr)Zcomposite_unitrrrtest_composite_composesrcCstjtjstdSr!)rrrobitrrrrrtest_data_quantities srcCs tjdkrttjdkstdSr!)rrrrrrrtest_compare_with_nonesrcCs8td}t|tst|jdks&t|jdks4tdS)Ng?r r )rZvalidate_powerrrrr r )fracrrr#test_validate_power_detect_fractions rc Csdtjdtj}dtj}d}dtjtjd}d}d}d }d tj}dtj|ttj}|| ||||tj |d} || d } | j j d kst | tjdS) Ng(\?r gv@iBg?g& .>rrgaa?gyy?.Aglength)rr,rZsolMassZsolRadrvrsqrtpirrVrWrr-) kappaZr_0qZrho_0ybetartZm_ht1t2rrrr'test_complex_fractional_rounding_errors s  ( r#cCsTtjdtdd}t|jdts(t|jdjdks|stq|jdkrR|stq|jdkrf|stq|jdkr~|dkstq|jdkr|dkstq|jd kr|d kstqd 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) astrophysZlyrrJZbarnrKZNorMZvoxZYesN)rr)rZ_iter_unit_summaryrrnr)r)summaryrVrrprefixesrrrtest_unit_summary_prefixesHs          r,cCs\tjdtjd}|d}|jtjtjgks2t|jddgksDt|tjtjksXtdS)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?rr)Zm2s2Zspmrrrtest_raise_to_negative_powercs r-z name, symbol, multiplying_factor)ZquettaQgꌠ9Y>)F)ZronnaRgqىE)ZyottaYgyCxD)ZzettaZgPKD)ZexaEgNgmC)ZpetaPg4&k C)ZteraTgmB)Zgigarr)megarr)Zkilokr8)Zdecadar)Zdecirg?)Zcentirvg{Gz?)Zmillirr1)microrrE)Znanong& .>)Zpicorg-q=)ZfemtorgV瞯<)ZattortgC]r2<)ZzeptozgO ;)ZyoctorgW:)Zrontorg<6S:)ZquectorgKH9cCsjdtj}||d}||d}t||s:tt||sJt|j|j}t||sftdS)Nr rZgram)rrr-iscloservalue)rnsymbolZmultiplying_factorbaseZquantity_from_symbolZquantity_from_nameZ value_ratiorrrtest_si_prefixesos  r@)v__doc__r fractionsrnumpyrr)Z numpy.testingrZastropyrrvrrrrrr r$r'r.r3r5r7rDrOrSrTrZr\rerjrsrwrzr~rrrrrrrrrrrrrZCOMPOSE_ROUNDTRIPrvaluesvalrZUnitBaseZ PrefixUnitaddmarkZ parametrizesortedrZCOMPOSE_CGS_TO_SIr[Zdeg_CrZCOMPOSE_SI_TO_CGSrXrrrrrrrrrrrrrrrrrrrrrr#rrrrrrrrrrrrrrrrrrrrrrrr#r%r(r,r-r@rrrrs:          !               '&   (