
    IR-e                         d dl Zd dlZd dlmZ d dlmZmZ d dl	m
Z
  G d de          Zd Zej                            d e                      d	             ZdS )
    N)units)Model
fix_inputs)Polynomial1Dc                   ,     e Zd ZdZdZ fdZd Z xZS )_ExampleModel   c                     dt           j        i| _        dt           j        t           j        z  i| _        t                                                       d S )Nxy)um_input_unitss_return_unitssuper__init__)self	__class__s    Ilib/python3.11/site-packages/astropy/modeling/tests/test_units_mapping.pyr   z_ExampleModel.__init__   sB     !#J!139-    c                 F    |t          j        dt           j                  z  S )Nr	   )r   Quantityr   )r   inputs     r   evaluatez_ExampleModel.evaluate   s    qz!QS))))r   )__name__
__module____qualname__n_inputs	n_outputsr   r   __classcell__)r   s   @r   r   r   
   sQ        HI    
* * * * * * *r   r   c                     t                      t                      z  } t                      t                      z   }t          d          }dt          j        t          j        z  i|_        dt          j        t          j        z  i|_        t                      |z  }t          | ddi          }t          | ddi          }| ||||g}t          j        d          t          j        d          ddt          j        d          idt          j        d          idt          j        d          idt          j        d          ig}t          j        d	          t          j        d	          d
dt          j        d	          idt          j        d	          it          j        d	          t          j        d	          d
t          j        d	          t          j        d	          d
g}t          j
        |||gt                    j        S )Nr	   r   r   x0r   r   )r#   x1r$   zm / s)y0y1)dtype)r   r   r   r   r   r   r   r   UnitnparrayobjectT)	m1m2pm3m4m5modelsinput_unitsreturn_unitss	            r   _models_with_unitsr6      s   	=??	*B	=??	*BQA139%ANAC!#I&AO	1	B	Bq		"	"B	BA		B"b"b!Fvc{{!&++..	afSkk	afSkk	qvc{{	qvc{{K vgafWoo66	afWoo	afWoovgafWoo66vgafWoo66L 8V[,7vFFFHHr   modelr4   r5   c                 :    | j         |k    sJ | j        |k    sJ dS )z,Test input_units on various compound models.N)r4   r5   r7   s      r   test_input_unitsr:   4   s2     ++++------r   )numpyr)   pytestastropyr   r   astropy.modeling.corer   r   astropy.modeling.modelsr   r   r6   markparametrizer:    r   r   <module>rC      s               3 3 3 3 3 3 3 3 0 0 0 0 0 0
* 
* 
* 
* 
*E 
* 
* 
*I I I: ACUCUCWCWXX. . YX. . .r   