
    HR-e,                     Z    d dl Z d dlZd dlmZ ddlmZ d
dZd
dZ G d d	e          Z	dS )    N)fits   )FitsTestCaseFc                    |                      t          j                  }|                     t          j                  }d}|j        j        D ]}|}||j        j        vrW|                                }||j        j        vr5|                                }||j        j        vrt          d| d          |rBt          j	        
                    d| d           t          j	        
                    d           ||         j        ||         j        k    r'|dz  }|rt          j	        
                    d           |r>t          j	        
                    d	           t          j	        
                    d
           t          j        ||                                         ||                                         k              \  }|j        dk    r4|dz  }|r+t          j	        
                    d|j         d| d           |rt          j	        
                    d	           |dk    r#|rt          j	        
                    d           dS |r"t          j	        
                    | d           dS )zY
    Compare the values field-by-field in two sets of numpy arrays or
    recarrays.
    r   zfield name z not found in array 1z    testing field: 'z'
z        shape...........r   zshapes differ
zOK
z        elements........z	
        z elements in field z differ
zAll tests passed
Tz differences found
F)viewnpndarraydtypenameslowerupper
ValueErrorsysstdoutwriteshapewhereravelsize)	arr1inarr2inverbosearr1arr2nfailn2n1ws	            Elib/python3.11/site-packages/astropy/io/fits/tests/test_structured.pycompare_arraysr       s]    ;;rz""D;;rz""DEj - -TZ%%%B)))XXZZTZ---$%L2%L%L%LMMM 	9J;B;;;<<<J78888>T"X^++QJE 4
  !2333 =
  (((
  !;<<<8DHNN,,R0@0@@AADQvzz
 J$$MQVMMMMM    -J$$V,,,zz 	3J1222t 	=J;;;<<<u    c                     t          j        dg d          }t           j                            d           g d|d<   g d|d<   t           j                            |d         j                  |d<   |S )	N   ))f1i4)f2S6)f3z>2f8#   r   r#      r$   )helloworldbyebyer&   r(   )r   zerosrandomseedr   )r   sts     r   get_test_datar3   ?   sn    	!AAA	B	BBINN2yyBtH+++BtHy4//BtHIr!   c                       e Zd Zd ZdS )TestStructuredc                 F   |                      d          }t          j        |dd          \  }}t          j        |dd          \  }}t                      }|                     d          }t          j        ||d           t          j        ||           t          j        ||           |j        j        sJ t          j
        |d         g d	k              sJ t          j        |dd          \  }}	t          j        |dd          \  }
}t          j        |d
d          \  }}t          ||d          sJ t          ||
d          sJ t          ||d          sJ t          j        |ddt          j                  \  }}t          ||d          sJ d S )Nzstddata.fitsr   T)extheader   z	test.fits)	overwriter$   r*   r#   )r   )r7   r8   r   )datar   getdatar3   tempwritetoappendr
   isnativer   allr    r	   )selffnamedata1h1data2h2r2   outfile
data1checkh1check
data2checkh2checkstchecksthcheckdataviewcheck
hviewchecks                   r   test_structuredzTestStructured.test_structuredK   s   		.))LAd;;;	rLAd;;;	r__))K((Wet4444GU###GR   x    vbh)))+,,,,,"l7$GGG
G"l7$GGG
G LaEEEeZ>>>>>>eZ>>>>>>b'4888888 %)L4bj%
 %
 %
!z e]DAAAAAAAAr!   N)__name__
__module____qualname__rQ    r!   r   r5   r5   J   s(        B B B B Br!   r5   )F)
r   numpyr   
astropy.ior   conftestr   r    r3   r5   rU   r!   r   <module>rY      s    


           " " " " " "0 0 0 0f   B B B B B\ B B B B Br!   