IR-eM dZddlZddlZddlmZddlmZddlm Z ddl m Z gej dfej dzd fejejz d fejejzd fej ejd zzd fej ej z d fejdfejej dzejzejzz dfejej dzejzejzz dfejej dzejzejzz dfejdfejdfejjejz d fejejz dfejejz dfejejz ej dzz dfejej dzzdfejej zdfej ej!z dfej"dd fej dzdfejdfej dfej#dfej ejdzz dfejd fej$d!fej%d"fej&d#fej'd#fejj(d#fejj)d#fejj)ej z d$fejd%fej*dfej+d&fej,dfej-ej dzz d'fej dzej-z d(fej%ej dzz d)fej-ej zejz d*fej-ej dzzejz dfej ejz dfej ejdzz d+fej$ej ejzz d,fej dzejz d-fej d.zd/fej.dfejd0fej/d1fej0d2fej1d3fej2d4fejej zd5fej.ej dzz dfej/ej z d6fejej dzz d7fejej dzz d8fej2ej z d9fej3dfejd:fej.ej z d;fej4ej z dfej6d?fej7d@fej,ej#z dAfej5ej dzz dBfej8j9dCfej8j:dDfej;jdfej;jejz dEfej<j=dFfej<j>dGfej<j?dHfej<j@dIfejjAejej zejj(zz dJfejjBej'z dKfejjBej'z ej$z dLfej*ej dMzzejd.zzdNfej,ej dzz dNfej dzej%z dOfej ej1z dPfej1ej z dQfej.ej dzzdRfej*ejz dRfejCd.zejdSzzdTfej ejdzz dUfej ejdVzz dWfej ejdXzz dYfej ejdZzz d[fej'ej z d\fejj(ej z d\fejj)ejjDz d\fejjEej$z d]fej%ejFz ejz d^fejjGejjDzejdzzd_fej%ejz d`fejjHej'z ej%z dafej%ej-z dbfejjIejJzdcfejjDdzejz ddfejejz defej+d.zdffejKd fZLejMNdgeLdhZOej jPZQejjPZRej ejz jPZSej dzjPZTej d.zjPZUejKjPZVej+jPZWej-ej zejz jPZXejMNdidjej dfdkdXej zdfeQeQfej+dlfdmgdnZYejMNdod eZfdpeZfdqhe[fgdrZ\dsZ]geQeQdtdufeQeTdtdvfeQddtdufdeQdte^feVeVdtdufeXd*dtdufeWdldtdufej d zjPd dtdufej d zjPej dwzjPdtdvfeQeQdxdvfeSeRdxdufeWeVdxdufeQej dte^feQeQdyeTfeSeRdyeQfeSeRdzeQfeQeRd{eSfeTeQd{eQfeQeTd|eQfeVeVdyeVfeVeVd{eVfeQdd}eTfeTd~d}eQfeVdVd}eVfej eQdye^fdeQdye^fej eRd{e^fdeQd{e^feQej dyeTfeQej dzeTfeSejdyeQfeQddyeQfeQddzeQfeQejd{eSfeTdd{eTfeRej d|eSfeQdd|eUfeQdd}eTfeQddye^feQddze^feQdd{e^feQd.d|e^feQddyeTfeQddzeTfeTdd{eQfeQdd|eQfZ_ejMNde_dZ`ej dhfej-ej zejz ddhfej+hdfgZaejMNdeadZbdZcdZdej ej fej ej fejej-ej zejdMzzfejejzej dzfej&ej'fej&ejj)fej&ejj(fej'ejj(fej dzejedzfg ZfejMNdefdZgejMNdefdZhej ejfej dzej dzfejej*fejejj(fgZiejMNdeidZjeQdfeSdfeWdfej'jPd#fej*ej&z ej-z jPdfgZkeQdfeSdfeWdfej'jPdfej*ej&z ej-z jPdfgZlejMNdekdZmejMNdeldZndZoejMNdepddd.gdZqdZrddgZsejMNdesdZtGddZuejMNdgddZvdZwejMNdgddZxdZydS)zC Unit tests for the handling of physical types in `astropy.units`. N)units)hbar)physical)AstropyDeprecationWarninglengthvolumespeedunknown dimensionlesszangular momentumzspectral flux density wavzphoton flux density wavzphoton flux densityz data quantitypowerzelectrical currentzelectrical current densityz magnetic fluxenergyz angular speedareatimeanglez solid angle acceleration frequencymasszamount of substance temperaturetemperature_gradientforcepressurez mass densityzspecific volumezmolar concentrationzmomentum/impulsezangular accelerationzdynamic viscosityzkinematic viscosity wavenumberzelectrical chargezelectrical potentialzelectrical resistancezelectrical conductancezelectrical capacitancezelectrical dipole momentzelectrical field strengthzelectrical flux densityzelectrical charge density permittivityzmagnetic flux densityzmagnetic field strengthzelectromagnetic field strength inductancezluminous intensityz luminous fluxzluminous emittance/illuminancezradiant intensity luminancezspectral flux densityz photon flux bandwidthzelectrical charge (ESU)zelectrical current (ESU)zelectrical current (EMU)zelectrical charge (EMU)zthermal conductivityz heat capacityzspecific heat capacityz energy fluxz molar volumezelectrical resistivityzelectrical conductivityzmagnetic momentzvolumetric ratejerksnapcracklepopztemperature gradientzspecific energyz reaction ratezmoment of inertiazcatalytic activityzmolar heat capacitymolalityabsementzvolumetric flow ratezfrequency driftcompressibilityzunit, physical_typecL|j|ksJ|d|d|jddS)z Test that the `physical_type` attribute of `u.Unit` objects provides the expected physical type for various units. Many of these tests are used to test backwards compatibility. z&.physical_type was expected to return z, but instead returned .N physical_type)unitr1s Alib/python3.11/site-packages/astropy/units/tests/test_physical.pytest_physical_type_namesr4}s\   . . .  K K  K K373E K K K / . . . .z0physical_type_representation, physical_type_name)?r )workr7energy_density)r8r8cvtj|}t|tjsJ||ksJdS)z/Test different ways of getting a physical type.N)rget_physical_type isinstance PhysicalType)physical_type_representationphysical_type_namer1s r3test_getting_physical_typer?sF./KLLM mX%: ; ;;;; . . . . . . .r5zargument, exceptionznot a name of a physical typez'this set cannot be made into a Quantityctj|5tj|ddddS#1swxYwYdS)zo Test that `get_physical_type` raises appropriate exceptions when provided with invalid arguments. N)pytestraisesrr:)argument exceptions r3%test_getting_physical_type_exceptionsrEs y ! !--"8,,,------------------s 7;;ctjt5tjtjjtjddddS#1swxYwYdS)z Test that `PhysicalType` instances cannot be cast into `Quantity` objects. A failure in this test could be related to failures in subsequent tests. N)rArB TypeErroruQuantitymr1r5r3)test_physical_type_cannot_become_quantityrLs y ! !++ 13$ac***++++++++++++++++++s/AAA__eq__TF __ne____mul____rmul__ __truediv__ __rtruediv____pow__g?g @r6 zleft, right, operator, expectedcDt||||ksJdS)zg Test that `PhysicalType` dunder methods that require another argument behave as intended. N)getattr)leftrightoperatorexpecteds r3test_physical_type_operationsr\s/ #74 " "5 ) )X 5 5 5 5 5 5r5impulsemomentum>stressrenergy densityzunit, expected_setc>t|j}||ksJdS)zs r3 z0test_physical_type_iteration..sQQQ2D-QQQr5)r`rr_N)r)physical_type_namess r3test_physical_type_iterationrk s4QQQQQ "J"J"J J J J J J Jr5c2dtvsJdtvsJdS)ze Test that `in` works as expected for `PhysicalType` objects with one or multiple names. rrN)rrrKr5r3test_physical_type_inrms, v      ! ! ! ! ! !r5z unit1, unit2cV|j|jkdusJ|j|jkdusJdS)zi Test that `physical.PhysicalType` instances for units of the same dimensionality are equal. TFNr0)unit1unit2s r3$test_physical_type_instance_equalityrr)sB  5#6 64 ? ? ? ?  5#6 65 @ @ @ @ @ @r5ctj|}tj|}||ksJ|dkr||usJdSdS)z Test that `get_physical_type` retrieves the same `PhysicalType` instances for equivalent physical types, except for unknown types which are not cataloged. r N)rr:rprqphysical_type1physical_type2s r3'test_get_physical_type_equivalent_pairsrw3s_/66N/66N ^ + + + +""////#"//r5ctj|d}tj|d}||kdusJ||kdusJdS)zy Test that `physical.PhysicalType` instances for units with different dimensionality are considered unequal. ptype1ptype2TFN)rr<rts r3&test_physical_type_instance_inequalityr|IsZ *5(;;N*5(;;N n , 5 5 5 5 n , 6 6 6 6 6 6r5zspeed/velocityzenergy density/pressure/stressz'specific entropy/specific heat capacityzPhysicalType('length')z#PhysicalType({'speed', 'velocity'})z6PhysicalType({'energy density', 'pressure', 'stress'})zPhysicalType('temperature')ztdi}|tdksJdS)z>Test that a `PhysicalType` instance can be used as a dict key.*N)r) dictionarys r3test_physical_type_hashrws'"J f  # # # # # #r5 multiplicandrc~tjt5t|zddddS#1swxYwYdS)zv Test that multiplication of a physical type returns `NotImplemented` when attempted for an invalid type. N)rArBrGr)rs r3!test_physical_type_multiplicationr}s y ! !s 266ctjdd}|j}t|tjsJ|dksJdS)zU Test basic functionality for the physical type of an unrecognized unit. parrotsilent) parse_strictr N)rHUnitr1r;rr<)unrecognized_unitr1s r3$test_unrecognized_unit_physical_typersQ xh???%3M mX%: ; ;;;; I % % % % % %r5)r)z valid inputr invalid_inputctjdz}tjt5t j||dddn #1swxYwY|jdksJdS)z Test that `PhysicalType` cannot be instantiated when one of the supplied names is not a string, while making sure that the physical type for the unit remains unknown. WNr )rHsrArB ValueErrorrr<r1)r obscure_units r3test_invalid_physical_typesrs37L z " ";;lM:::;;;;;;;;;;;;;;;  % 2 2 2 2 2 2sA  AAcjeZdZejdzZejdzZdZdZ dZ dZ e dZ dZd S) TestDefPhysTypecrctjt5tj|jddddn #1swxYwYdtjvsJdS)z.Test that a unit cannot be defined as unknown.r N)rArBrrdef_physical_type weird_unit_unit_physical_mappingselfs r3,test_attempt_to_define_unknown_physical_typezAEH EHB*J*ctjt5tj|jdddddS#1swxYwYdS)z Test that a physical type name already in use cannot be assigned for another unit (excluding `"unknown"`). rN)rArBrrrrrs r3test_redundant_physical_typez,TestDefPhysType.test_redundant_physical_types ]: & & B B  &t A A A B B B B B B B B B B B B B B B B B BsAA Act|jD]}tj|=tj|=|jdksJdS)z-Reset the physical type of unit to "unknown".r N)listr1rr_physical_unit_mapping_get_physical_type_id)r2names r3_undef_physical_typez$TestDefPhysType._undef_physical_types]+,, 6 6D/55  +D,F,F,H,H I!Y......r5c|j|jfD]J}tj|}|dkr|||jdksJd|dKdS)z Remove the definitions of the physical types that were added using `def_physical_unit` for testing purposes. r zthe physical type for z., which was added fortesting, was not deleted.N)r strange_unitrr:rr1)rr2r1s r3teardown_methodzTestDefPhysType.teardown_methods _d&78  D$6t<>>>>>r5cpttttjsJttttjsJtjD]D}tj|}tt|sJtt||usJEtj td5tj ddddS#1swxYwYdS)Nzhas no attributer) rbdirr issupersetrkeys__all__hasattrrWrArBrnot_a_valid_physical_type_name)pnamers r3!test_physical_types_module_accessrsJ s8}}   ( ()L)Q)Q)S)S T TTTT s8}}   ( ()9 : ::::499;;113E:x'''''x''500000 ~-? @ @ @00//000000000000000000s D++D/2D/)z__doc__rrAastropyrrHastropy.constantsr astropy.unitsrastropy.utils.exceptionsrrJcmkmhbarnMpcrr2ergAAphotonHzbytebitimperialmiweekCTNradmsrsrgmolKdeg_Cdeg_Fdeg_RJPaWkgAVOhmSFWbHcdlmlx astrophysJyRmisccgsFranklin statampereBiot abcoulombbtucalyrftCalorieLlbfkcalinchhrdimensionless_unscaledunit_physical_type_pairsmark parametrizer4r1rrr rrr rr^r?rrGrErLNotImplementedoperation_parametersr\unit_with_physical_type_setrerkrmpcequivalent_unit_pairsrrrwnonequivalent_unit_pairsr|physical_type_with_expected_str physical_type_with_expected_reprrrrrrrinvalid_inputsrrrrrrrKr5r3r1s= """""""""""">>>>>>kS(OkT1WhkTACZkVae^X k S136\9 k S13Y k Y"#kUadAgmad"#%@AkXq13%&(ABkXq13%&(=>kV_kUOkZ]QVW%kUQS['kS13Y$%k S13Yq67!k"S136\?##k$S13Y%k&UQT\?#'k( QVAYY )k*S!VV+k,S&M-k.UG/k0T=1k2S136\>"3k4T;5k6S&M7k8U !"9k:S-;k<Wm=k>Z}%?k@Z}%AkBZ34CkDS'NEkFS(OGkHT:IkJS'NKkLTACF]N#MkNS!Vad]%&OkPUQS!V^*+QkRTACZ!#)*SkTTACF]QS,-UkVUQS[/"WkXUQS!V^+,YkZSAC!#I+,[k\S!Vac\()]k^S"Wl_k`S akbS ckdS !ekfU #$gkhS "#ikjS "#kklS13Y*+mknS136\/0okpS13Y+,qkrS136\,-sktS136\./ukvS13YwkxT?ykzS !"{k|S13Y)*}k~S13Y01k@S,AkBT  CkDT?EkFT +,GkHS14Z$%IkJTACF]K KkL[^,-MkN[]M"OkPVZ!QkRVZ!#{#SkTU^./UkVU12WkXUZ+,YkZU_/0[k\Z^qsQSy1:#3346LM]k^Z^ag/_k`Z^ag#%=>akbS137]QS"Wm,ckdS136\=!ekfS!Vae^^$gkhS13Y()ikjS13Y)*kklS136\$%mknS13Y!"okpT2Xr ,-qkrS136\6sktS136\6ukvS136\9wkxS136\5ykzWqs]*+{k|Z34}k~Z %'=>k@Z!#01AkBUQS[13(CkDZ^ajm#ac1f,.ABEkFUQS[&'GkHZ_qw&(=>IkJUQT\:KkLZ_qtZ(MkNZ]A34OkPTACZ"#QkRT2X !SkT/Uk\.0HII  JI   s qs! Qc2g $ (6 4  D13J  +6 h QS(   ,  //  / J (*5 3 4i@---+++/ VXt$/ T8U#/ Xx&/vx0 / M8T2 / !8T2 /40/c1fY$7/c1fac1f3XuE/ VXu%/ D(D!/}h-/ QS(N+/ VY%/ D)V$/  D*f%!/" T=%(#/$ 6=&)%/& T>6*'/(M9m<)/*M=-@+/, Q 4 -/. 3 6"//0Ay-01/2S&)^,3/4 &)^,5/6S$ ~.7/8 &-09/: QS)T";/< QS*d#=/> ACF#?/@ Q 6"A/B Q F#C/D QS-'E/F 1mT"G/H 13&I/J S.*-K/L Q 4 M/N RN+O/P Q N+Q/R S-0S/T R0U/V Xy$'W/X Xz4(Y/Z 8]F+[/\ V^V,]/d:??AA@?A)>?? 0 0@? 0S!#JS"Wac2gS!#JVQZ  )ABB77CB7 X  /0WM*cACi!$%'PQ # %& 12 GHW9: qsQT (F $ 68WXX..YX. "$D000 $$$ $$&&"a)<==>=&&&,-.99 3 3:9 3TTTTTTTTnWWW-- - @ @ @'E'E'EFFGF 0 0 0 0 0r5