IR-e)jddlZddlZddlZddlmZmZddlmZ ddl m Z m Z m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!ddl"m#Z#ddl$m%Z%ej&'dd d d gfd d d gfd ddgfdddgfdddgfgdZ(ej&'ddd d gfdd d gfdddgfdddgfdddgfgdZ)ej&'ddd fdd fddfddfddfgdZ*d Z+d!Z,d"Z-d#Z.d$Z/ej&'d%ededzedzededz edzededzedzededz edzededzedzed&d'd(ed)d*d+zgd,Z0d-Z1d.Z2d/Z3d0Z4ej&'d1ed&d'ed'ed&d'gd2Z5d3Z6d4Z7d5Z8d6Z9Gd7d8e Z:Gd9d:e Z;d;Zd>Z?d?Z@d@ZAej&Be% dABdCZCdDZDej&Be% dABdEZEej&'d1ed*ed*ed*gdFZFdGZGej&'dHdIdJdKdLgdMZHdNZIdOZJej&'dHdPdQdRdSgdTZKdUZLdVZMdWZNdXZOdYZPej&Be% dABdZZQej&Be% dABd[ZRej&Be% dABd\ZSej&Be% dABd]ZTdS)^N)assert_allcloseassert_array_equal) CompoundModelModelModelDefinitionError)LevMarLSQFitter) Chebyshev1D Chebyshev2DConst1D Gaussian1D Gaussian2DIdentity Legendre1D Legendre2DLinear1DMapping Polynomial1D Polynomial2D Rotation2DScaleShift Tabular1D fix_inputs) Parameter) HAS_SCIPY)exprresultc ||zSNxys Dlib/python3.11/site-packages/astropy/modeling/tests/test_compound.pyr%' a!e@c ||z Srr r!s r$r%r%(r&r'gc ||zSrr r!s r$r%r%)r&r'g@c ||z Srr r!s r$r%r%*r&r'gUUUUUU?c ||zSrr r!s r$r%r%+ adr'g @c|tddtdd}|dd}t||dS)Nr0r0n_models)r3rF)model_set_axis)r rrrsouts r$test_model_setr8$sY WVa ( ( ('&1*E*E*EFFA !Ae $ $ $CsF#####r'c ||zSrr r!s r$r%r%7r&r'c ||z Srr r!s r$r%r%8r&r'c ||zSrr r!s r$r%r%9r&r'c ||z Srr r!s r$r%r%:r&r'c ||zSrr r!s r$r%r%;r-r'cd}tjt|5|tddtddddddS#1swxYwYdS) ziCheck that creating model sets with components whose _n_models are different raise a value error z+Both operands must have equal values for .*matchr/r0r1r3N)pytestraises ValueErrorr )rrMESSAGEs r$!test_model_set_raises_value_errorrF4s=G z 1 1 1BB WVa ( ( ('!a*@*@*@AAABBBBBBBBBBBBBBBBBBs+AAAc ||zSrr r!s r$r%r%Jr&r'c ||z Srr r!s r$r%r%Kr&r'c ||zSrr r!s r$r%r%Lr&r'c ||z Srr r!s r$r%r%Mr&r'c ||zSrr r!s r$r%r%Nr-r'c|tdtd}t|tsJ|jdksJ|jdksJ|d}||ksJt|t sJdS)z Like test_two_model_class_arithmetic_1d, but creates a new model from two model *instances* with fixed parameters. r0r3rArN)r isinstancern_inputs n_outputsfloatr5s r$%test_two_model_instance_arithmetic_1drQGs WQZZ$$A a ' '''' :???? ;!     !A$$C &==== c5 ! !!!!!!r'ctdtdz}t|tsJ|jdksJ|jdksJ|ddksJtdtdz}t|tsJ|jdksJ|jdksJ|ddksJt |gdgddS) z\ Shift and Scale are two of the simplest models to test model composition with. r0r3rA"@r()rAr0r3)r(g@rSN)rrrMrrNrOr)S1S2s r$ test_simple_two_model_compose_1drVcs qE!HH B b- ( (((( ;!     <1     2a55C<<<< qE!HH B b- ( (((( ;!     <1     2a55C<<<<rr)))}}ooo66666r'ctdtdz}t|tsJ|jdksJ|jdksJt |ddddtdtdz}t |ddd d||z}t |ddd dd S) z7 A simple example consisting of two rotations. -r0rrA)rg|=atolZ)rrYN)rrMrrNrOr)r1r2r3s r$ test_simple_two_model_compose_2dr`ys B*R.. (B b- ( (((( ;!     <1    BBq!HHgE2222 B*R.. (BBBq!HHgE2222 bBBBq!HHgE222222r'cttz}|jdksJ|tz}|jdksJ||z}|jdksJ||z }|jdksJdS)z\ Test that CompoundModel.n_submodels properly returns the number of components. r0r3N)r n_submodels)g2g3g5g7s r$test_n_submodelsris  $B >Q     jll B >Q     bB >Q     bB >Q      r'ctddd}tdddddd}||z}|dksJ||z|z}|dksJ|||zz}|d ksJ||z|z}|d ksJ|||zz|z}|d ksJ||z||zz}|d ksJ||z||zz}|d ksJ||z}|dksJ|||zz}|dksJ||z|z}|dksJ||z|z}|dksJ|||zz}|dksJ||z|z}|dksJ|||zz}|dksJdS)z\ Test that the expression strings from compound models are formatted correctly. rAr0r3rbz [0] + [1]z[0] + [1] + [2]z[0] + [1] * [2]z[0] * [1] + [2]z[0] + [1] * [2] + [3]z([0] + [1]) * ([2] + [3])z[0] * [1] + [2] * [3]z [0] ** [1]z[0] + [1] ** [2]z([0] + [1]) ** [2]z[0] + [1] | [2]z[0] + ([1] | [2])z[0] & [1] | [2]z[0] & ([1] | [2])N)r r _format_expression)GG2Ms r$test_expression_formattingrqs 1aA Aq!Q1 % %B AA   ! ![ 0 0 0 0 A A   ! !%6 6 6 6 6 AE A   ! !%6 6 6 6 6 A A   ! !%6 6 6 6 6 AE A A   ! !%< < < < < Q1q5A   ! !%@ @ @ @ @ Q1q5A   ! !%< < < < < 1A   ! !\ 1 1 1 1 AqDA   ! !%7 7 7 7 7 Q1 A   ! !%9 9 9 9 9 A A   ! !%6 6 6 6 6 QU A   ! !%8 8 8 8 8 A A   ! !%6 6 6 6 6 QU A   ! !%8 8 8 8 8 8 8r'ctdtdztdtdzztdz}t|j|ddddS)z Test basic inversion of compound models in the limited sense supported for models made from compositions and joins only. r0r3r\rrArrAN)rrrrinverse)ts r$test_basic_compound_inverservsd qE!HH qE!HH!45 2FAIAIqqAww'00000r'modelrAr0r3rkrbrlcd}tjt|5|jddddS#1swxYwYdS)zJ Ensure inverses aren't supported in cases where it shouldn't be. z3No analytical or user-supplied inverse transform .*r?N)rBrCNotImplementedErrorrt)rwrEs r$!test_compound_unsupported_inverserzs EG *' : : : s 377ctddd\}}tdtdz}|dd\}}t||f||ftdtdz}|ddd\}}}|tdz}|ddd\}}}t|||f|||fdS)zx Tests a couple basic examples of the Mapping model--specifically examples that merely permute the outputs. r\rAr0)rArr3r0rrAN)rrrr) r"r#rsx_primey_primemzmsz_primes r$test_mapping_basic_permutationsrs :b>>!Q  DAq B'&// )Br!QxxGWQFWg./// 2q!Aa1ajjGAq! WY   B "1a GWgQ1I':;;;;;r'c tdtdz}tdtdz}|tgdz|z}td|j|ddd d d S) z;Tests inverting a compound model that includes a `Mapping`.g333333(@gffffff*@g,@g.@r|)rrAr0rrAr0g:0yE>rZN)rrrrrt)rs1rs2rs r$test_mapping_inversers T  U4[[ (C T  U4[[ (C giii   3&AIyqy!!Aq!**5EBBBBBBr'ctd}td}td}||z|z}t|ddddgdS)z Test a case where an Identity (or Mapping) model is the first in a chain of composite models and thus is responsible for handling input broadcasting properly. Regression test for https://github.com/astropy/astropy/pull/3362 rAr\)angler0g?N)rrrr)ident1shiftrotationrws r$test_identity_inputrs\a[[F !HHE###H UNX %EEE!QKK$-----r'c`d}tjt|5tdt dddzdddn #1swxYwYd}tjt|5tdt dddzddddS#1swxYwYdS) zq Test that certain operators do not work with models whose inputs/outputs do not match up correctly. zUnsupported operands for |:.*r?r\rAr皙?Nz6Both operands must match numbers of inputs and outputs)rBrCrrr )rEs r$test_invalid_operandsr.s0 /G +7 ; ; ;//2Aq#....///////////////HG +7 ; ; ;//2Aq#....//////////////////s#"A  AA4"B##B'*B'polycgd|_td}||z}tjddddf\}}|||}||||}t ||dS)zY Tests that polynomials are scaled when used in compound models. Issue #3699 rAr0r3rkrAr0r3N%) parametersrnpmgridr)rrrwr"r#result_compoundrs r$!test_compound_with_polynomials_2dr=s )((DO !HHE 5LE 8CRC"H DAqeAqkkO U441::  FFO,,,,,r'c tddddd}tddddd}t|ddi}t|d|ddt|gd|gdgd t|d di}t|d|dd||z}t|d d d }t|dd|dd dd t|dd d}t|dd|ddd dt|dd|ddd ddS)NrArr0g??gɿ333333?)rrAr3)rrrr"r皙?)rAr3g)x0r0)y0y1)r rr)g1resg1_1sg1_2gg1sgg1_1sgg1_2s r$test_fix_inputsrLs Aq!Q " "B CdC - -B rAq6 " "EEE!HHbbAhh'''EE)))$$bbIII&>&>??? rC8 $ $EEE#JJ1c +++ r'C -- . .FFF1aLL##aa"5"5666 DS11 2 2FFF1aLL##dAsA"6"6777FFaA&&&D!S!(<(<=====r'c|tddddd}d}tjt|5t |ddddddn #1swxYwYtjt|5t |ddidddn #1swxYwYtjt|5t |t jddidddn #1swxYwYtjt|5t |t jddidddn #1swxYwYtjt|5t |ddidddn #1swxYwYd }tjt|5t |d dddn #1swxYwYd }tjt|5td ||dddn #1swxYwYd }tjt|5t |ddi}|dddddn #1swxYwYtjt|5t |t jddi}|dddddn #1swxYwYtjt|5t |t jddi}|ddddddS#1swxYwYdS)NrArr0z4Substitution key .* not among possible input choicesr?)rrr3wz9Expected a dictionary for second argument of "fix_inputs"rsz.*Illegal operator: ', '#'.*#z,Too many input arguments - expected 1, got 2)r#) r rBrCrDrrint32int64rr)rrErs r$test_fix_inputs_invalidr\s Aq!Q " "BEG z 1 1 1((2aA'''((((((((((((((( z 1 1 121v z 1 1 1))2 Q'((())))))))))))))) z 1 1 1))2 Q'((())))))))))))))) z 1 1 1!!2Qx   !!!!!!!!!!!!!!!KG z 1 1 12v.G +7 ; ; ;##c2r"""###############>G z 1 1 1aV$$ A  z 1 1 1bhqkk1-.. A  z 1 1 1bhqkk1-.. A sAAA7BBB<%C--C14C1%EE E*F  F F 1GGG6HHH< I((I,/I,2K  KK22L11L58L5cftddddd}tddddd}t|jd||z}d|_|jdksJt|ddd}|jdksJt ddd}||z}d |_t|ddi}|jd ksJdS) NrAr)gg@r)r)gg@g333333g333333@)gg@)rr0)rr r3)rArk)rlr)r r bounding_boxrr )rrersgggfs r$!test_fix_inputs_with_bounding_boxrs Aq!Q " "B Aq!Q " "BBO%?@@@ r'CKC  S S S S S CQ1 & &B ?8 8 8 8 8 B1  B RA%AN A1v  B ?f $ $ $ $ $ $r'ctdddtdz}t|dtsJt|dtsJ|jdksJ|djdcxkr |jksnJ|djdcxkr |jksnJ|djdcxkr |j ksnJ|djdcxkr |j ksnJ|d}d|_ |jdksJd|_|j dksJtdddd }tdd }||z}|dj d ksJ|dj d ksJ|d j d ksJ|d j d ksJ|d}d |_ |jd ksJd |_|jd ksJt|dtsJt|dtsJd}t#jt&|5|ddddn #1swxYwYd}t#jt&|5|ddddn #1swxYwYt)dddddd }t+|ddi}|d j d ksJtdd }tdd }tdd }|||zz} t-| dddddS)z*Test indexing on compound model instances.rArrr0) amplitude_0mean_0stddev_0 amplitude_1*r3rnamepi90irYzlist index out of ranger?Nz%No component with name 'foobar' foundfoobarrkrbg?Ag@Bg@Cg ףp= @)r r rM param_names amplitudermeanrstddevrrrrc0_1c0c1c1_1rBrC IndexErrorr rr) rconstrrrrErrrrps r$test_indexing_on_instancers 1a *A adJ ' '''' adG $ $$$$ =P P P P P Q4>Q / / / /!- / / / / / / Q49 % % % %QX % % % % % % Q4;# + + + + + + + + + + Q4>Q / / / /!- / / / / / / aDEAM ?b EO =C     1a%%%AQS!!!A AA Q49     Q49     S6;#     S6;#     Q4D AF 7e    DG 6T>>>> ae\ * **** aeZ ( (((((G z 1 1 1 "7G z 1 1 1 (  1aAqs+++A1q!fA S6;#     #A#A#A AE AJAc#gJqMM#455555s$ H..H25H2 I++I/2I/cZeZdZedddZeddZedZdS)_ConstraintsTestArr)defaultminmaxTrfixedc ||fSrr )rrs r$evaluatez_ConstraintsTestA.evaluates t|r'N)__name__ __module__ __qualname__rrr staticmethodrr r'r$rrsW YqaS 1 1 1F 9Qd + + +D\r'rc>eZdZeddZedZdS)_ConstraintsTestBrTrc|Srr )rs r$rz_ConstraintsTestB.evaluates r'N)rrrrrrrr r'r$rrsA 9Qd + + +D\r'rctddidditddiz}d|jvsJ|jddksJ|jjdksJd|jvsJ|jddusJ|jjdusJd |jvsJ|jd dusJ|jjdusJ|j|d jusJd |j_|d jjd ksJ|d jdd ksJd |j_|d jjd ksJ|d jdd ksJd |dj_|jjd usJ|djjd usJ|jdd ksJd |_d|d j_|jdd ksJd|_|jddksJdS)z Various tests for copying of constraint values between compound models and their members. Regression test for https://github.com/astropy/astropy/issues/3481 r)rrrT)boundsr)rrrmean_1r)rg?)rrFrA)rrN) r rrrrrrrsync_constraints)rws r$test_inherit_constraintsrsK x264. I I IJtnMMM E  % % % % < #x / / / / > H , , , , u{ " " " " ;x D ( ( ( ( <  % % % % u{ " " " " ;x D ( ( ( ( <  % % % % >U1X_ , , , ,%EN 8? !X - - - - 8?8 $ 0 0 0 0&EN 8? !Z / / / / 8?8 $ 2 2 2 2E!HM <  & & & & 8= % ' ' ' ' < #z 1 1 1 1"E%E!HO < #z 1 1 1 1!E < #x / / / / / /r'ctddd}td}td}||z}||_||z}t |d||jzdd}t jt|5||zjdddn #1swxYwYt jt|5||zjddddS#1swxYwYdS)z Test that a compound model with a custom inverse has that inverse applied when the inverse of another model, of which it is a component, is computed. Regression test for https://github.com/astropy/astropy/issues/3542 rAr0)rrzTNo analytical or user-supplied inverse transform has been implemented for this modelr?N)rrrrtrrBrCry)rscalermodel1model2rEs r$test_compound_custom_inversers aA & & &D !HHE !HHE E\FFNV^FFNN1%%(<u}(>> AF 6S==== $B 6T>>>> 7d??????r'ctddd}tddd}||z}d}tjt|5|ddddn #1swxYwYd|_|djdksJd|_d}tjt|5|dddddS#1swxYwYdS)NrAr0z#No component with name 'bozo' foundr?bozoz0Multiple components found using 'bozo' as name.*)r rBrCrrr)rrerrEs r$test_name_indexr gsQ Aq!  B Aq!  B RA4G z 1 1 1 & BG V9>Q    BGAG z 1 1 1 & s$ AA!A# B99B=B=zrequires scipy)reasonc tgdfgdd}td}td}tjdd}||z|z}|||\}}|jdksJ|jdksJd S) zK Issue #7411 - evaluate should not change the shape of the output. )rArbrc) F)points lookup_table bounds_errorr0rAr)r3rkN)rrrrarangereshapendim)rurotrr"rwx1rs r$test_tabular_in_compoundrws )))LLLuUUUA Q--CQA " f%%A ECKE U1a[[FB 7a<<<< 7a<<<<<@g5B܀@g_P@g]@g(O@gVZ@g w@ghh=@r) r rrrarraynanisnanrsum)rr#r"valcomparemaskval2s r$test_bounding_boxr%s z!S#q!Q777A'AN 8AbD!B$J DAq CA CA !AqD ) ) )Ch RVRVRV VRVRVRVRV 5 RVRVRV VRVRVRVRV 5 RVRVRV VRVRVRVRV 5 RVRVRV VRVRVRVRV 5 VRVRVRVRV VRVRVRVRV 5 VRVRVRVRV VRVRVRVRV 5 VRVRVRVRV VRVRVRVRV 5 VRVRVRVRV VRVRVRVRV 5 VRVRVRVRV VRVRVRVRV 5 VRVRVRVRV VRVRVRVRV 5' G4 HSMM>DCIwt}--- 1QUAET 2 2 2D 8D>>    3 & & & & & &r'ctjdtjz}tjdtjz}t ||}|dtjzddtjzksJdS)NrbrATrr)rrupixAAr)rltrus r$test_bounding_box_with_unitsr+sk Yq\\AE !F 1 B&"A 1QY$ / / /3: = = = = = =r'cgd|_td}||z}tjddd}||}|||}t |||jdksJdS)zY Tests that polynomials are offset when used in compound models. Issue #3699 rr3rbr)c0_0c1_0c2_0c3_0c4_0c5_0offset_1N)rrrlinspacerr)rrrwr"rrs r$!test_compound_with_polynomials_1dr6s )((DO !HHE 5LE B2AeAhhO U4477^^FFO,,,  !      r'ctddtddz}tddtddz}||z}|ddd ksJ|dtd d }|ddd ksJ|ddd ksJ|d ddksJ|dtdtdz}|ddd ksJ|dddksJ|dddksJ||z}|ddksJ|dtd d }|ddksJt t ddddi}||z}t |dd|dt d}t |ddd}tjt|5|dtd}dddn #1swxYwYtddd } tddgd!} | | z}t|ddgdd"}tjt|5|dtdd#}dddn #1swxYwY|dtdddgd$}t|ddgd%td}td}tdd&} || z} td'| _|| z}|d&td }|ddd(ksJ|dd)dksJdS)*z- Replace a model in a Compound model r0shift2rr3scale3scale2shift3rA) rcrkscale4)rrcrrc)rAr0)rcr>r\rotate)rrr)rA)rYrBz"No submodels found named not_therer? not_thereNr)degreer2rr1z6New and old models must have equal values for n_models)rEr)rErr2)r0rkS3g)r0rr) rrreplace_submodelrtrrrrBrCrDrr) rTrUrrm3Rm4m5rEPSrFS23s r$test_replace_submodelrOs qx 5#:#:#: :B qx 5#:#:#: :B RA 1Q77f      HeAH&=&=&= > >B 2a88w     1Q77f     ::b!   & & & &  HeAhhq&9 : :B 1Q77f     2a88w     ::b!   & & & & RA 1Q442::::  HeAH&=&=&= > >B 2a55B;;;; :BX666A??A aBBBqEE7###  Xz# 7 7BBBqEE8$$$4G z 1 1 177    U1XX 6 6777777777777777 A777A q!fq!!!A AAqq!Qyy&)))GG z 1 1 1FF    A!(D(D(D E EFFFFFFFFFFFFFFF  FLq!fq$Q$Q$Q R RBrr1a&zz6*** qB qB qt   B r'C++CK SA  D%(( + +B 2a88v     ::a  v % % % % % %s$$I  II7&K))K-0K-rc ||zSrr rrs r$r%r% rBwr'c ||z Srr rQs r$r%r%rRr'c ||zSrr rQs r$r%r%rRr'c ||z Srr rQs r$r%r%rRr'c btjddd}tjgd}tjgd}td}t ddd}|||}t |j|g||R||j|g|R|j|g|RdS) t Tests that compound evaluate function produces the same result as the models with the operator applied r-rbrr)rArrr0rAN)rr5rrr rrrr"p1p2rrcompounds r$test_compound_evaluater\s B2A $$$ % %B +++  B !__F 1a FtFF##H!&b&2&&& _V_Q $ $ $ $ofoa&="&=&=&=>>r'cDtjddd}tjgd}tjdg}tddd}t d}||z}t |j|g||R|j|g|R|j|g|RzdS) z Tests that compound evaluate function produces the same result as the models with the power operator applied r-rbrrArrr3r0rANrr5rr r rrr"rYrZrrr[s r$test_compound_evaluate_powerrb5s B2A +++  B 1#B 1a F QZZFv~H!&b&2&&&B?6?1#:r#:#:#::r'c(tjddd}tjddd}tdddddd}td}td}tddddddtdztdz}t ||||dz |dzdddddddd |||dddddd||dz d||dzdgdS)Nr-rbrrArr0r)rr5r rrr)r"r#rrrHrs r$#test_compound_evaluate_double_shiftrdHs B2A B2A Aq!Q1 % %B qB qB1aAq!$$uQxx/%((:A 1aRRAq!Q1a@@ KK1aAq!Q / / KKB " " KKB " " r'c ||zSrr rQs r$r%r%]rRr'c ||z Srr rQs r$r%r%^rRr'c ||zSrr rQs r$r%r%_rRr'c ||z Srr rQs r$r%r%`rRr'c tjddd}tjgd}tjgd}tddd}tddd}|||}t |j|g|R|d|d|dd ||j|g|R|j|g|Rd S) rWr-rbrr_)r3rrr0rArrrrN)rr5rr rrrXs r$"test_compound_evaluate_named_paramrkZs B2A +++  B --- B 1a F 1a FtFF##H!RbRRbeBqEBqERRR _V_Q $ $ $ $ofoa&="&=&=&=>>r'c ntjddd}tjgd}tjdg}tddd}t d}||z}t |j|g|R|d|d|dd |j|g|R|j|g|Rzd S) r^r-rbrr_r3r0rArrjNr`ras r$'test_compound_evaluate_name_param_powerrmvs B2A +++  B 1#B 1a F QZZFv~H!RbRRbeBqEBqERRRB?6?1#:r#:#:#::r'cLtjddd}tjgd}tjdg}t}t }||z}t |j||g||R|j|g|R|||gdS)rWr-rbr)rArrr3N)rr5rr rrrras r$test_compound_evaluate_andros B2A --- B 1#B \\F WWFH!Q((R(((  R &//!R"8"89r'c tjddd}tjdg}tjdg}tjdg}tjdg}td}t ddd}||z}t |||||||||||||d S) rWr-rbrrr3rrrAN)rr5rrr rr)r"rY p2_amplitudep2_meanp2_stdrrr[s r$test_compound_evaluate_orrts B2A 3%B8QC==LhsmmG Xse__F 3ZZF 1c " "FH!Rw??2.. gvNNr'ctjddddf\}}gd}tdddd}t|d|d zi}t |j||g|R|j|d z|g|RdS) z Tests that compound evaluate function produces the same result as the models fix_inputs operator is applied when using the keyword Nrr3rrrArrrAr0rrr"rbrrr rrrr#r" model_paramsrwr[s r$,test_compound_evaluate_fix_inputs_by_keywordrzs 8CRC"H DAq)))L q!Q $ $E%#q1u..H!Q....q1ua/,///r'ctjddddf\}}gd}tdddd}t|d|dzi}t |j||g|R|j|dz|g|RdS) z Tests that compound evaluate function produces the same result as the models fix_inputs operator is applied when using the input index NrrvrAr0rrrbrwrxs r$-test_compound_evaluate_fix_inputs_by_positionr|s 8CRC"H DAq)))L q!Q $ $E%!QU,,H!Q....q1ua/,///r'c@t}tjdddtjz}tjdddtjztjztjz }tdtjztjz tjz dtjztjz }tdtjztjz d tjz}||z}||||}||}||}|j|jcxkr*tjtjztjz ksnJt|||j D]&} t|| t|| ksJ'd S z* Regression test for issue #12320 rrAerbrr)slope interceptrg$@N rrr5r'r6rkgrunitrrgetattr fitterr"r#rrtruthfit unfit_output fit_outputrs r$3test_fit_multiplied_compound_model_with_mixed_unitsrsj   F Aq#$A Ar3!#%,qs2A AC!# +sQSy13 G G GB ad QS(D14K @ @ @B GE &1  C588LQJ    E E E EAD13J4D E E E E E EL*---!::ud##wsD'9'999999::r'cdt}tjdddtjz}tjdddtjztjztjztjz }tdtjztjz tjz dtjztjz }tdtjztjz d tjz}tdtjztjz d tjz}||z|z}||||}||}||} |j| jcxkr7tjtjztjztjz ksnJt|| |j D]&} t|| t|| ksJ'tjdddtjz}tjdddtjztjztjztjztjz }tdtjztjz tjz dtjztjz }tdtjztjz d tjz}tdtjztjz d tjz}tdtjztjz d tjz} ||z} || z} | | z}||||}||}||} |j| jcxkrDtjtjztjztjztjz ksnJt|| |j D]&} t|| t|| ksJ'd Sr~rrr"r#rrrHrrrrrrJm11m22s r$=test_fit_multiplied_recursive_compound_model_with_mixed_unitsrsh   F Aq#$A Ar3!#%+ad2QS8A AC!# +sQSy13 G G GB ad QS(D14K @ @ @B ac AC4!#: > > >B GbLE &1  C588LQJ    K K K KAD13J4Dqs4J K K K K K KL*---!::ud##wsD'9'999999 Aq#$A Ar3!#%+ad2QT9AC?A AC!# +sQSy13 G G GB ad QS(D14K @ @ @B ac AC4!#: > > >B ad QS(D14K @ @ @B r'C r'C #IE &1  C588LQJ    R R R RAD14K!#4E4Kac4Q R R R R R RL*---!::ud##wsD'9'999999::r'c@t}tjdddtjz}tjdddtjztjztjz }tdtjztjztjz dtjztjz}tdtjztjz d tjz}||z }||||}||}||}|j|jcxkr*tjtjztjz ksnJt|||j D]&} t|| t|| ksJ'd Sr~ rrr5r'r6rrrrrrrrs r$0test_fit_divided_compound_model_with_mixed_unitsr sj   F Aq#$A Ar3!$&,qs2A AD13,ad QS8H I I IB ac AC4!#: > > >B GE &1  C588LQJ    E E E EAD13J4D E E E E E EL*---!::ud##wsD'9'999999::r'cdt}tjdddtjz}tjdddtjztjztjztjz }tdtjztjztjz dtjztjz}tdtjztjz d tjz}tdtjztjz d tjz}||z |z}||||}||}||} |j| jcxkr7tjtjztjztjz ksnJt|| |j D]&} t|| t|| ksJ'tjdddtjz}tjdddtjztjztjztjztjz }tdtjztjztjz dtjztjz}tdtjztjz d tjz}tdtjztjz d tjz}tdtjztjz d tjz} ||z } || z} | | z}||||}||}||} |j| jcxkrDtjtjztjztjztjz ksnJt|| |j D]&} t|| t|| ksJ'd Sr~rrs r$8test_fit_mixed_recursive_compound_model_with_mixed_unitsr9sh   F Aq#$A Ar3!$&,qs2QS8A AD13,ad QS8H I I IB ac AC4!#: > > >B ac AC4!#: > > >B GbLE &1  C588LQJ    K K K KAD13J4Dqs4J K K K K K KL*---!::ud##wsD'9'999999 Aq#$A Ar3!$&-3ac9AC?A AD13,ad QS8H I I IB ac AC4!#: > > >B ac AC4!#: > > >B ad QS(D14K @ @ @B r'C r'C #IE &1  C588LQJ    R R R RAD14K!#4E4Kac4Q R R R R R RL*---!::ud##wsD'9'999999::r')UrnumpyrrB numpy.testingrr astropy.unitsunitsr'astropy.modeling.corerrrastropy.modeling.fittingrastropy.modeling.modelsr r r r r rrrrrrrrrrrrastropy.modeling.parametersr"astropy.utils.compat.optional_depsrmark parametrizer8rFrQrVr`rirqrvrzrrrrrrrrrrrrrrrrr skipifrr%r+r6rOr\rbrdrkrmrortrzr|rrrrr r'r$rs  ========LLLLLLLLLL444444&211111888888  c3Z(  dD\*  c3Z(  i34  S#J'   $$  $   c3Z(  dD\*  c3Z(  i34  S#J'   BB  B  S!  T"  S!  Y'  C   ""  "$777,333*&898989v111  a5588eeAhh& a5588eeAhh& a5588eeAhh& a5588eeAhh& aEE!HHuuQxx' 1ajjAq111     <<<* C C C . . . / / /++a"3"3\\!__jjQRTUFVFV!WXX - -YX - > > > &&&R%%%&<6<6<6~-0-0-0`   B!!!"           M*:;;<;"&'&'&'T M*:;;>><;>++a..**Q--a!QRRSR.A&A&A&H  (&$  &&&*&& M*:;;::<;:0 M*:;;,:,:<;,:^ M*:;;::<;:0 M*:;;,:,:<;,:,:,:r'