IR-e< ddlZddlZddlZddlmZmZddlmZ ddl m Z ddl m Z ddlmZddlmZmZmZmZmZmZmZmZmZGdd eZeeeeegZeeefZeefZejd ed Z ejd ed Z!ejd ed Z"ejd edZ#ejd edZ$dZ%dZ&dZ'dZ(dZ)dZ*ejd edZ+ejd eeegdZ,ejdee e gejd eeegdZ-ejdee e gejdee j.dzfee j.fede j.dzz fgdZ/ejdee e gejdee j.dzfee j.fede j.dzz fgdZ0ejdee e gejd eeegdZ1ejd ed Z2ejd!e3ed"d#d$fd%Z4ejd ed&Z5d'Z6dS)(N)assert_allcloseassert_array_equal)units)CCDData NDDataArray)NDData) "IncompatibleUncertaintiesExceptionInverseVarianceMissingDataAssociationException NDUncertaintyStdDevUncertaintyUnknownUncertaintyVarianceUncertainty_move_preserved_axes_first_unravel_preserved_axescBeZdZedZdZdZdZdZdZ dS)FakeUncertaintycdS)Nfake)selfs Glib/python3.11/site-packages/astropy/nddata/tests/test_nduncertainty.pyuncertainty_typez FakeUncertainty.uncertainty_type0svcdSNr)rvalues r_data_unit_to_uncertainty_unitz.FakeUncertainty._data_unit_to_uncertainty_unit4strcdSrrrdata final_datas r_propagate_addzFakeUncertainty._propagate_add7 rcdSrrr!s r_propagate_subtractz#FakeUncertainty._propagate_subtract:r%rcdSrrr!s r_propagate_multiplyz#FakeUncertainty._propagate_multiply=r%rcdSrrr!s r_propagate_dividez!FakeUncertainty._propagate_divide@r%rN) __name__ __module__ __qualname__propertyrrr$r'r)r+rrrrr/sx X              rr UncertClassc |gd}t|jtjgd|gdtj}t|jtjgd|jtjusJdSN)unit)rarraynpuadur7)r0 fake_uncerts rtest_init_fake_with_listr=Ws+iii((K{("(999*=*=>>>+iiiae444K{("(999*=*=>>>  qu $ $ $ $ $ $rcttjddd}||}t|j||j|usJ||d}|j|usJ||t j}t|j||j|usJ|jt jusJdS)Nd Fcopyr6)r9arangereshaperr8r:r;r7r0uncertr<s rtest_init_fake_with_ndarrayrGcs Ys^^ # #B + +F+f%%K{(&111  F * * * *+f5111K   & & & &+f15111K{(&111  F * * * *  qu $ $ $ $ $ $rctjdddtjz}||}t |j|j|j|jusJ|jtjusJ||d}|j|jusJ|jtjusJ||tj }t |j|j|j|jusJ|jtj usJdS)Nr@r4FrAr6) r9rCrDr:r;rr8rr7mrEs rtest_init_fake_with_quantityrKts Yr]] " "1a ( (15 0F+f%%K{(&,777  FL 0 0 0 0  qu $ $ $ $+f5111K  FL 0 0 0 0  qu $ $ $ $+f13///K{(&,777  FL 0 0 0 0  qs " " " " " "rc tjddd}||}||}t|j||j|usJ||d}||d}t|j|j|j|jusJtjdddt jz}||}||}t|j|j|j|jusJ|jt jusJ||t j }t|j|j|j|jusJ|jt j usJdS)NrIr3FrAr6) r9rCrDrr8r:r;rr7cm)r0rF fake_uncert1 fake_uncert2s rtest_init_fake_with_fakerPs Yq\\ ! !!Q ' 'F;v&&L;|,,L|)6222  V + + + +;vE222L;|%888L|)<+=>>>  !3 3 3 3 3 Yq\\ ! !!Q ' '!% /F;v&&L;|,,L|)6<888  V\ 1 1 1 1   % % % %;|!$777L|)6<888  V\ 1 1 1 1   $ $ $ $ $ $rc*ddd}||}|j|ksJ||tj}|j|ksJ|jtjusJ||d}|j|ksJt |t |ksJdS)Ng333333@g333333?)rdnoisegainr6FrA)r8r:sr7idrEs r!test_init_fake_with_somethingElserVsc * *F+f%%K   & & & &+f13///K   & & & &  qs " " " "+f5111K   & & & & k??bjj ( ( ( ( ( (rctjddd}t|}t jt 5t|dddn #1swxYwYt|}t jt 5t|ddddS#1swxYwYdS)NrIr3)r9rCrDrpytestraisesr r)rF std_uncertr<s r%test_init_fake_with_StdDevUncertaintyr[s4Yq\\ ! !!Q ' 'F"6**J 9 : :$$ ###$$$$$$$$$$$$$$$"&))K 9 : :''+&&&''''''''''''''''''s$A--A14A1 B==CCctddg}|jdksJtddg}|jdksJtddg}|jdksJt ddg}|jdksJdS)Nr@r4rstdvarivar)rrrrr )r<rZ var_uncert ivar_uncerts rtest_uncertainty_typerbs!2q'**K  '6 1 1 1 1"B7++J  &% / / / /$b!W--J  &% / / / /!2q'**K  '6 1 1 1 1 1 1rcntddg}|jrJtddg}|jsJdS)Nr@r4)rsupports_correlatedr)r<rZs rtest_uncertainty_correlatedresH!2q'**K...."B7++J  ))))))rcddlmddlmd fd }d}d}||t||tddlmfd}fd }||||dS) Nr) defaultdict) get_objectscztD]}t|xxdz cc< |tD]}t|xxdz cc< | tfdDsJdS||z dksJdS)zFunction based on gc.get_objects to determine if any object or a specific object leaks. It requires a function to be given and if any objects survive the function scope it's considered a leak (so don't return anything). r3Nc3BK|]}||z dkVdS)rNr).0kafterbefores r zDtest_for_leak_with_uncertainty..test_leak..s5@@QuQx&)+q0@@@@@@rr)inttypeall)funcspecific_objectsirmrnrgrhs @@r test_leakz1test_for_leak_with_uncertainty..test_leaksS!! ! !A 477OOOq OOOO  C    A $q''NNNa NNNN  #@@@@@%@@@@@ @ @ @ @ @)*V4D-EEJJJJJJrcHttjddSNr?)r r9onesrrrnon_leaker_nddataz9test_for_leak_with_uncertainty..non_leaker_nddatas  rws||rcttjdttjddSNr? uncertainty)r r9ryrrrr leaker_nddataz5test_for_leak_with_uncertainty..leaker_nddatas4rws||):273<<)H)HIIIIIIrrcBtjddSrx)r9ryrsrnon_leaker_nddataarrayz>test_for_leak_with_uncertainty..non_leaker_nddataarrays! BGCLL!!!!!rctjdttjddSr|)r9ryrrsrleaker_nddataarrayz:test_for_leak_with_uncertainty..leaker_nddataarrays7 BGCLL.? .M.MNNNNNNrr) collectionsrggcrhr astropy.nddata.compatr)rvrzrrrrrgrhs @@@rtest_for_leak_with_uncertaintyrs('''''KKKKKKK, JJJI((( ImV$$$211111"""""OOOOOI$k222 I +.....rctdd}td|j}|jjj|jksJ|jjj|jksJdS)Nr3r}r4)r r~ parent_nddatar")ndd1ndd2s rtest_for_stolen_uncertaintyr sg ! # # #D !!1 2 2 2D   ) .$) ; ; ; ;   ) .$) ; ; ; ; ; ;rcttjdtj}t jt j|}tj |j |j |j |j ksJtj t5|jddddS#1swxYwYdS)Nr5r6)rr9ryr:rJpickleloadsdumpstestingrr8r7rXrYr r)r~uncertainty_restoreds rtest_stddevuncertainty_picklers#BGAJJQS999K!< [(A(ABBJ!!+"35I5OPPP  38 8 8 8 8 6 7 7++**++++++++++++++++++s)B>>CCc|gdtj}t|jtjsJ|jjtjsJ|gd}t|jtjsJ|jjtjsJdSr2)r:r; isinstancequantityQuantityr7 is_equivalentdimensionless_unscaled)r0r<fake_uncert_nounits r test_quantityrs+iiiae444K k*AJ 7 7777   $ 2 215 9 9999$YYY// (11: > >>>>  & + 9 9!:R S SSSSSSrcp|ddg}d|_t|jtjsJdS)Nr3electron)r7rr:UnitBase)r0vs r4test_setting_uncertainty_unit_results_in_unit_objectr&s>  QFA AF afaj ) ))))))rNDClassc|dd}|d}||_tjtj5|jdz|_ddddS#1swxYwYdS)Nr3r;r6)r~rXrYr:UnitConversionErrorr7)rr0rrs r:test_changing_unit_to_value_inconsistent_with_parent_failsr/s 715 ! ! !D AAD q, - -BsAA"AzUncertClass, expected_unitr4r3cz|ddgtj}|ddg}||_|j|ksJdSNr3r6r:r;r~r7rr0 expected_unitnddrs r7test_assigning_uncertainty_to_parent_gives_correct_unitr>sQ '1a&qu % % %C QFACO 6] " " " " " "rc~|ddgtj}|ddg|}||_|j|ksJdSrrrs r8test_assigning_uncertainty_with_unit_to_parent_with_unitrRsV '1a&qu % % %C QF///ACO 6] " " " " " "rc|ddgtj}|ddgtj}tjtj5||_ddddS#1swxYwYdSr)r:r;secondrXrYrr~)rr0rrs r8test_assigning_uncertainty_with_bad_unit_to_parent_failsrfs '1a&qu % % %C QF***A q, - -sA&&A*-A*c tjddddtjz}||}||}t |j|j|j|jksJdSNr3 r4rI) r9rCrDr:r; represent_asrr8r7r0rF start_uncert final_uncerts r+test_self_conversion_via_variance_supportedrts Yq"   % %a + +ae 3F;v&&L,,[99L|)<+=>>>   1 1 1 1 1 1 1rzUncertClass,to_variance_funcc |dzS)Nr4rxs rrs 1a4rc|Srrrs rrrs1rc d|z S)Nr3rrs rrrs Arctjddddtjz}||}|t }||}t||j|jt|j|j|j |j ksJdSr) r9rCrDr:r;rrrr8rr7)r0to_variance_funcrFrr`rs r*test_conversion_to_from_variance_supportedr}sYq"   % %a + +ae 3F;v&&L**+>??J**;77L$$\%788*:JKKK|)<+=>>>   1 1 1 1 1 1 1rctjddddtjz}||}t jt5||}ddddS#1swxYwYdSr) r9rCrDr:r;rXrY TypeErrorrrs r/test_self_conversion_via_variance_not_supportedrs Yq"   % %a + +ae 3F;v&&L y ! !>>#00== >>>>>>>>>>>>>>>>>>sA>>BBc d}dtjtj||}t |}|j}|dtjtj|tjksJtj|tjfdtt|D}tj tj |dd|sJtj |ttdt|dz }|jd|dksJt||}tj|tj}tj tj |j|sJdS)N)rIr5r4)r3r4rcg|]}|v| Srr)rkru preserve_axess r z0test_reshape_ndarray_methods..s#III!=2H2H!2H2H2Hrr3)axis)r9rCprodrDrshaper8rangelenrrequalsumtupler) rarr reshaped_arr new_shapeshape_unchanged_axessummed unravelledshape_preservedrs @rtest_reshape_ndarray_methodsrs EM )BGENN # # + +E 2 2C-c=AAL"I Q<2728E??28M3J3J#KLL L L L L8E?? IIIIU3u::..IIIJJ 6"(9QRR=*>?? @ @@@@VLuU1c%jj1n-E-E'F'F G G GF <?il * * * *(fmDDJhuoobh}&=&=>O 6"(:+_== > >>>>>>r)7rnumpyr9rX numpy.testingrrastropyrr:astropy.nddata.ccddatarrrastropy.nddata.nddatar astropy.nddata.nduncertaintyr r r r rrrrrruncertainty_types_to_be_tested)uncertainty_types_with_conversion_support,uncertainty_types_without_conversion_supportmark parametrizer=rGrKrPrVr[rbrerrrrrrr;rrrrziprrrrrrrs  ========******------((((((                      D     m   2 "-) 1@AS/T,(FGG%%HG%(FGG % %HG % (FGG##HG#$(FGG%%HG%2(FGG ) )HG )$ ' ' '222***3/3/3/l<<<+++(FGGTTHGT'):OL*** V['$BCC'):OLDCV['$BCC aeQh' AE" !aeQh,'##DC#V['$BCC aeQh' AE" !aeQh,'##DC#V['$BCC(,=RDC(QRR22SR2"C1 oo6222(TUU>>VU>?????r