๓ š฿ศ[c@`shdZddlmZmZmZmZddlZddlZddl m Z ddl m Z ddl mZdd lmZdd lmZdd l mZydd lmZeZWnek rำeZnXd „Zejjdƒd„ƒZejjdƒd„ƒZejjdƒd„ƒZejjdƒd„ƒZ ejjdƒd„ƒZ!dS(u> Tests that relate to fitting models with quantity parameters i(tabsolute_importtunicode_literalstdivisiontprint_functionNi(t Gaussian1Di(tunits(t UnitsError(tassert_quantity_allclose(tNumpyRNGContext(tfitting(toptimizec C`sŽtdƒ\tjdddƒ}dtjd|ddd ƒ}|tjjd d |jƒ7}WdQX|tj}|tj }||fS( Ni90gภg@iะigเฟgอฬฬฬฬฬ๔?igš™™™™™้?ggš™™™™™ษ?g|ฎGแzไ?( RtnptlinspacetexptrandomtnormaltshapetutmtJy(txty((sMlib/python2.7/site-packages/astropy/modeling/tests/test_quantities_fitting.pyt_fake_gaussian_data s #%  u not HAS_SCIPYcC`s‘tƒ\}}tƒ}tjƒ}||||ƒ}t|jdtjddƒt|jdtj ddƒt|j dtj ddƒdS(Nitrtolgš™™™™™ฉ?gอฬฬฬฬฬ๔?gš™™™™™้?( RRR tLevMarLSQFitterRt amplitudeRRtmeanRtstddev(RRtg_inittfit_gtg((sMlib/python2.7/site-packages/astropy/modeling/tests/test_quantities_fitting.pyttest_fitting_simple/s  cC`sธtƒ\}}tddtjddtjddtjƒ}tjƒ}||||ƒ}t|j dtj ddƒt|j d tj ddƒt|j d tj ddƒdS( NRg๐?RiRiRgš™™™™™ฉ?gอฬฬฬฬฬ๔?gš™™™™™้?(RRRtmJytcmtmmR RRRRRRR(RRRRR((sMlib/python2.7/site-packages/astropy/modeling/tests/test_quantities_fitting.pyt test_fitting_with_initial_values@s0 c C`s๓tddtjddtjddtjƒ}tjƒ}tjt ƒ)}||dddgdd d gƒWd QX|j j d d ks’t ‚tjt ƒ0}||dddgtj dd d gƒWd QX|j j d dks๏t ‚d S(uD Raise an error if the model has units but the data doesn't Rg๐?RiRiiiiiNiu8'cm' (length) and '' (dimensionless) are not convertibleuH'mJy' (spectral flux density) and '' (dimensionless) are not convertible(RRR R!R"R RtpytesttraisesRtvaluetargstAssertionErrorR(RRtexc((sMlib/python2.7/site-packages/astropy/modeling/tests/test_quantities_fitting.pyttest_fitting_missing_data_unitsQs0 (/cC`sAtƒ\}}tddddddƒ}tjƒ}||||ƒ}t|jdtjddƒt|jd tj ddƒt|j d tj ddƒtddddtj ddtj ƒ}tjƒ}||||ƒ}t|jdtjddƒt|jd tj ddƒt|j d tj ddƒd S( u= Proceed if the data has units but the model doesn't Rg๐?RiRiRgš™™™™™ฉ?gอฬฬฬฬฬ๔?gš™™™™™้?N( RRR RRRRRRRR(RRRRR((sMlib/python2.7/site-packages/astropy/modeling/tests/test_quantities_fitting.pyt test_fitting_missing_model_unitsds ) c C`sคtddtjddtjddtjƒ}tjƒ}tjt ƒ7}||dddgtj dd d gtjƒWd QX|j j d d ks t ‚d S(uF Raise an error if the data and model have incompatible units Rg๐?RiRiiiiiNiu5'Hz' (frequency) and 'm' (length) are not convertible(RRRRR!R RR$R%RtHzR&R'R((RRR)((sMlib/python2.7/site-packages/astropy/modeling/tests/test_quantities_fitting.pyttest_fitting_incompatible_units}s 0 6("t__doc__t __future__RRRRtnumpyR R$tmodelsRtRRRt tests.helperRtutilsRR tscipyR tTruet HAS_SCIPYt ImportErrortFalseRtmarktskipifRR#R*R+R-(((sMlib/python2.7/site-packages/astropy/modeling/tests/test_quantities_fitting.pyts("