U e{js(@sdZddlZddlZddlmZddlmZddl m Z m Z ddl m Z mZmZmZmZddlmZmZddlmZdd lmZGd d d e Zd d ZddZddZddZddZddZddZ ddZ!ej"#dej$dfddej$ffdd Z%d!d"Z&d#d$Z'd%d&Z(d'd(Z)dS))zF Tests that relate to using quantities/units on parameters of models. N) coordinates)units)Fittable1DModelInputParameterError)Const1D Gaussian1D Pix2Sky_TANRotateNative2Celestial Rotation2D) ParameterParameterDefinitionError)assert_quantity_allclose) UnitsErrorc@seZdZeddZdS) BaseTestModelcCs|S)N)xarrp/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/modeling/tests/test_quantities_parameters.pyevaluateszBaseTestModel.evaluateN)__name__ __module__ __qualname__ staticmethodrrrrrrsrcCstdtjdtjdtj}|jjdks.t|jjtjks@t|jjdksPt|jjtjksbt|j jdksrt|j jtjkstdS)z Basic tests for initializing general models (that do not require units) with parameters that have units attached. 皙??N) ruJm amplitudevalueAssertionErrorunitmeanstddevgrrrtest_parameter_quantitysr'cCstdtjdtjdtj}dtj|_t|jdtjdtj|_t|jdtjdtj |_ t|j dtj dtj |_t|jdtj dtj |_t|jdtj dS)az Make sure that parameters that start off as quantities can be set to any other quantity, regardless of whether the units of the new quantity are compatible with the original ones. We basically leave it up to the evaluate method to raise errors if there are issues with incompatible units, and we don't check for consistency at the parameter level. rrN) rrrrZkJrr kmr#mmr$sJyr%rrrtest_parameter_set_quantity,s      r/c Cs:tdtjdd}d}tjt|d d|_W5QRXdS)ab Check that parameters that have been set to a quantity that are then set to a value with no units raise an exception. We do this because setting a parameter to a value with no units is ambiguous if units were set before: if a parameter is 1 * u.Jy and the parameter is then set to 4, does this mean 2 without units, or 2 * u.Jy? rr)rzVThe .* parameter should be given as a .* because it was originally initialized as a .*matchr*N)rrr.pytestraisesrrr&MESSAGErrrtest_parameter_lose_unitsMs r6cCs.tddd}dtj|_t|jdtjdS)z On the other hand, if starting from a parameter with no units, we should be able to add units since this is unambiguous. rr)rr*N)rrr.rr r%rrrtest_parameter_add_units`s  r7c Csltddtjdtj}d}tjt|dtj|j_W5QRXd}tjt|dtj |j _W5QRXdS)z Test that changing the unit on a parameter does not work. This is an ambiguous operation because it's not clear if it means that the value should be converted or if the unit should be changed without conversion. rrzNCannot attach units to parameters that were not initially specified with unitsr0zYCannot change the unit attribute directly, instead change the parameter to a new quantityN) rrrr2r3 ValueErrorr.rr"cmr#r4rrrtest_parameter_change_unitlsr:c Cstdtjdtjdtj}dtj|_d|j_t|jdtj|jjdksTt|jjtjksftd}t j t |ddtj|j_W5QRXdS) zH Test that changing the value on a parameter works as expected. rrr*r(zZThe .value property on parameters should be set to unitless values, not Quantity objects.*r0r)N) rrr.rrr r r!r"r2r3 TypeErrorr4rrrtest_parameter_set_values r<c Cstdtjdtjdtj}t|jjdtjdtj|j_|jjdksPt |jj tjksbt dtj |j_|jjdkst |jj tj kst d}t j t|dd|j_W5QRXdS) zK Test that the quantity property of Parameters behaves as expected rrr(z:The .quantity attribute should be set to a Quantity objectr0r)N)rrr.rr rZquantityZmJyr r!r"r-r2r3r;r4rrr test_parameter_quantity_propertys r>c Cs2d}tjt|dGdddt}W5QRXdS)NzNparameter default 1.0 m does not have units equivalent to the required unit Jyr0c@s eZdZedejejdZdS)z1test_parameter_default_units_match..TestCrdefaultr"N)rrrr rrr.rrrrrTestCsrA)r2r3r r)r5rArrr"test_parameter_default_units_matchsrB)r"r@rrc sbGfdddt}|jjtjks(t|jjdks8t|}|jjtjksPt|jj|jjkrldksrnt|dtj}|jjtjkst|jjdkst|jjdkst|dtj}|jjtjkst|jjdkst|jjdkst|dtj }|jjtj kst|jjdks&t|jjdks8td}t j t |d|dW5QRXdS)zG Test that default quantities are correctly taken into account cseZdZedZdS)z*test_parameter_defaults..TestModelr?N)rrrr rrr?rr TestModelsrCrg@z'.* requires a Quantity for parameter .*r0N) rrr"rrr!r@r Zpcr.r2r3r)r"r@rCrr5rr?rtest_parameter_defaultss*"rDc CsJtdtjdtjdtj}|jdtjdtjksrBmarkZ parametrizerrDrGrJrQrTrrrrs0    !   1#)