IR-e0dZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZmZddlmZddlmZd Zd Zd ZGd d e ZGddZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dS)zA Tests that relate to evaluating models with quantity parameters N)assert_allclose)units)Model) Gaussian1D Pix2Sky_TANScaleShift)assert_quantity_allclose) UnitsErrorzT{}: Units of input 'x', {}.*, could not be converted to required input units of {}.*ctddd}tdtjzdtjzdtjz}t |dtjz|dtjzt jtt ddd5|ddddn #1swxYwYt |d|dtjzt|d tj zj |d t jtt dd d 5|d tj zdddn #1swxYwYt jtd5|d tjzddddS#1swxYwYdS)zn Test evaluation of a single model with Quantity parameters that do not explicitly require units. g?rzm matchNrgMb@??smzACan only apply 'subtract' function to dimensionless quantities .*)ruJrr pytestraisesr MESSAGEformatrkmvaluer)ggqs Qlib/python3.11/site-packages/astropy/modeling/tests/test_quantities_evaluation.pytest_evaluate_with_quantitiesr sk 1aA AGQWcACi 0 0BRRAC[[!!A$$*555 z b$)O)O P P P 1RRUUAAaDD13J///BBv}%%+QQsVV444 z c3)O)O P P P 1qs7  R    !ac' s6; CCC6FF"FG''G+.G+ctdtjzdtjzdtjz}t jt tddd5|dtj zd d d n #1swxYwYt|dtj zd tj i |d tjzd S) zN We now make sure that equivalencies are correctly taken into account r rPHznmrNx equivalenciesg3eZdZdZdZdZdZdZdZdZ dZ d S) TestInputUnitsc,t|_dSN)r.model)r3methods r setup_methodzTestInputUnits.setup_methodas ]] r5cFt|dddt|dtjzddtjzt|dtjzddtjzdS)Nr)r r@rrdegr3s r test_evaluatezTestInputUnits.test_evaluatedsz Aq!1!12666 AGQ!7!7acBBB AIq!9!92:FFFFFr5cdtji|j_t |dtjzddtjzt |dtjzddtjzt |dtjzdtjzdtjztjztjtt ddd  5|dtjzddddn #1swxYwYtjtt dd d  5|ddddddS#1swxYwYdS) Nr'rrF r#r.rrHrr) rrHr@ _input_unitsr radrrrr rrrIs rtest_input_unitszTestInputUnits.test_input_unitsjs#&,  AIq!9!92:FFF AIq!9!91qu9EEE AIq13w!?!?QUQSQQQ ]:W^^M3PU-V-V W W W # # JJq13w " " " # # # # # # # # # # # # # # #]:W^^M2u-U-U V V V   JJq!                     s$"$EEEF66F:=F:cndtji|j_d|j_t |dtjzddtjzt |dtjzddtjztjtt dd d  5|dtj zddddn #1swxYwYt |ddd dS) Nr'TrrFrLr#rMr.rrHr ) rrHr@rN _input_units_allow_dimensionlessr rOrrr rrrrIs r$test_input_units_allow_dimensionlessz3TestInputUnits.test_input_units_allow_dimensionlessws+#&, 6: 3 AIq!9!92:FFF AIq!9!91qu9EEE ]:W^^M3PU-V-V W W W # # JJq13w " " " # # # # # # # # # # # # # # # !Aq!1!1155555s$DD  D cdtji|j_d|j_t |dtjzddtjz|t jtjzd}t |dtjz|j tjusJdS)Nr'TrrFrLr#ih) rrHr@rN_input_units_strictr nppirOunitr3results rtest_input_units_strictz&TestInputUnits.test_input_units_stricts#&, )- & AIq!9!92:FFFBEAEM1-- qu555{ae######r5c 6dtji|j_t jt tddd5|dtj zddddn #1swxYwYdtj i|j_ t|dtj zdddtj z tjtj zdS)Nr'r.r$micronrrr()rr^r@rNrrr rrr$r+input_units_equivalenciesr torIs rtest_input_units_equivalenciesz-TestInputUnits.test_input_units_equivalenciess#&/  ] gnn]E8LL    % % JJq15y! $ $ $ % % % % % % % % % % % % % % % 14QZ\\/B , JJq15y! $ $ QUqxqz||DD D     s$A>>BBcdtji|j_dtji|j_|dtjzd}t |dtjz|jtjusJdS)NzrrFrLrrHr@rNrO _return_unitsr rYrZs rtest_return_unitsz TestInputUnits.test_return_unitssl#&, $'<  AIq)) ae444{ae######r5c dtji|j_tj|j_|dtjzd}t |dtjz|jtjusJdS)Nr'rrFrLrdrZs rtest_return_units_scalarz'TestInputUnits.test_return_units_scalarsj$', #$5  AIq)) ae444{ae######r5N) r6r7r8rBrJrPrTr\rarfrhr;r5rr=r=`s###GGG    6 6 6$$$    $$$ $ $ $ $ $r5r=ctdtjz}tdtjz}||z}|dtjzdtjz}t |ddtjzdtjzzt |ddtjzdtjzzdS)- Test units to first model in chain. r"rGrr Nr rrH arcsecondr arcsecs1s2csouts rtest_and_input_unitsrss rAEz  B rAEz  B bB "R!+ rAK/ 0 0CSVR!%Z"qx-%?@@@SVR!%Z"qx-%?@@@@@r5c tdtjz}tdtjz}||z}|dtjz}t |dtjzdtjzzdSrjr"rGNrkrns rtest_plus_input_unitsrvp rAEz  B rAEz  B bB "R!+   CS"qu*rAH}"<=====r5c tdtjz}tdtjz}||z}|dtjz}t |dtjzdtjzzdSrurkrns rtest_compound_input_unitsryrwr5cLtdtjz}tdtjz}||z}tjt t ddd5|dtjzddddS#1swxYwYdS)z: Test incompatible units to first model in chain. r"r pixrHrN) r rrHrrr rrr{rorprqs rtest_compound_input_units_failr} rAEz  B rAEz  B bB z)N)N O O O 2:3BB BcLtdtjz}tdtjz}||z}t jt tddd5|dtjzddddS#1swxYwYdS)z1 Test incompatible model units in chain. r"r r{rHrN) r rr{rHrrr rrr|s r%test_compound_incompatible_units_failrr~rcJtdtjz}tdtjz}||z}|dtjzdtjdtjztjz i}t |dtjzdS)zo Check that equivalencies work when passed to evaluate, for a chained model (which has one input). r"r'rr(Nr rrHr{ pixel_scaler rns rtest_compound_pipe_equiv_callrs rAEz  B rAEz  B bB "R!%ZQ]3;;N-O-O'P Q Q QCS"qu*-----r5c tdtjz}tdtjz}||z}|dtjzdtjztjdtjztjz tjdtjztjz d}t |ddtjzt |ddtjzdS) zk Check that equivalencies work when passed to evaluate, for a composite model with two inputs. r"rx0x1r(rrEr Nrrns rtest_compound_and_equiv_callrs rAEz  B rAEz  B bB " QU  QU -ae ae 344-ae ae 344     CSVR!%Z000SVR!%Z00000r5ctdtjz}dtjdtjztjz i|_tdtjz}tdtjz}||z}|jdtjdtjztjz iksJ|dtjz}t |dtjz||z}|jJ|dtjz}t |dtjz||z}|jdtjdtjztjz iksJ|d}|dtjzdtjz}t |dtjztj tt d d d  5|dtjzdtjz}ddddS#1swxYwYdS) zF Test setting input_units_equivalencies on one of the models. r"r'rrNrGr TestModelr r{rHr) r rrHrr{r_r renamerrr rr)rorpsprqrrs r'test_compound_input_units_equivalenciesrs' rAEz  B$'sQU{QU7J)K)K#LB rAEz  B rAEz  B bB 'CsQU{QU?R1S1S+T T T T T "R!%Z..CS"qu*--- bB ' / / / "R!%Z..CS"qu*--- bB 'D!-ae ae@S2T2T+U U U U U ;  B "R!%Zae $ $CS"qu*--- z)N)N O O O))baeR!%Z(())))))))))))))))))s'II IcGddt}|d}td}||z}|dtjz}t|dtjz|jtjusJ||z}|dtjz}t|dtjz|jtjusJ||z}|dtjzdtjz}t|dtjz|djtjusJ|djtjusJd S) z? Test setting input_units_strict on one of the models. c eZdZdejiZdS)6test_compound_input_units_strict..ScaleDegreesr'Nr6r7r8rrH input_unitsr;r5r ScaleDegreesrAAEl r5rr#r"(rGrr N)rrrmr rYrHrrorprqrrs r test_compound_input_units_strictr<sL #####u### aB qB bB "R!(]  CS"qx-000 8qu     bB "R!(]  CS"qx-000 8qu     bB "R!(]BM * *CS"qx-000 q6;!%     q6;!( " " " " " "r5cGddt}|d}d|_td}||z}|d}|d}t|dtjz|dtjz}t|dtjztjtt ddd  5|dtj z}d d d n #1swxYwYd |_||z}|d}tjtt dd d  5|d}d d d n #1swxYwYd|_||z}|d}|d}t|dtjz|dtjz}t|dtjztjtt ddd  5|dtj z}d d d n #1swxYwYd |_||z}tjtt dd d  5|d}d d d n #1swxYwYd|_|d}d|_|d}d |_||z}|d}|ddtjz}t|ddtjzt|ddtjztjtt dd d  5|dd}d d d d S#1swxYwYd S)zL Test setting input_units_allow_dimensionless on one of the models. c eZdZdejiZdS)Ctest_compound_input_units_allow_dimensionless..ScaleDegreesr'Nrr;r5rrr`rr5rr#Trr"rrrHrNFrrrGr ) rrSrr ronermrrr rrrrs r-test_compound_input_units_allow_dimensionlessr[s #####u### aB*.B' qB bB ;  B "R&&CS"qu*--- "R!(]  CS"qx-000 zU)S)S T T Tbacll+0B' bB ;  B zE)R)R S S Sbff+/B' bB ;  B "R&&CS"qu*--- "R!(]  CS"qx-000 zU)S)S T T Tbacll+0B' bB zE)R)R S S Sbff+/B' aB*.B' aB*/B' bB ;  B "Rah  CSVR!%Z000SVR!(]333 zE)R)R S S SbRjjsZ!DD  D ' E??FF=I""I&)I&. KK  K * OO OcGddt}t|z}t|dtjzdtjzdtjzdS)zn Test that return_units on the first model in the chain is respected for the input to the second. cXeZdZdZdZfdZedZedZdZ xZ S)-test_compound_return_units..PassModelr#c:tj|i|dSr?)super__init__)r3argskwargs __class__s rrz6test_compound_return_units..PassModel.__init__s% EGG d -f - - - - -r5c4tjtjdS)z Input units.rrrHrIs rrz9test_compound_return_units..PassModel.input_units%qu-- -r5c4tjtjdS)z Output units.rrrIs r return_unitsz:test_compound_return_units..PassModel.return_unitsrr5c|j|jfSr?)r)r3r'ys rr4z6test_compound_return_units..PassModel.evaluates7AG# #r5) r6r7r8r9r:rpropertyrrr4 __classcell__)rs@r PassModelrs  . . . . .  . .  .  . .  . $ $ $ $ $ $ $r5rr)rZN)rrr rrH)rrqs rtest_compound_return_unitsrsv $$$$$E$$$(  $BRRAE 1qu955wGGGGGr5)$__doc__numpyrWr numpy.testingrastropyrrastropy.modeling.corerastropy.modeling.modelsrrrr astropy.tests.helperr astropy.unitsr rr r,r.r=rsrvryr}rrrrrrrr;r5rrs ))))))''''''IIIIIIIIIIII999999$$$$$$  $$$T"%O$O$O$O$O$O$O$O$d A A A > > > > > >       . . .111,)))B###>CCCLHHHHHr5