
    IR-e                     x    d dl Zd dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
mZmZ d Zd ZdZ G d	 d
          ZdS )    N)given)basic_indices)assert_equal)check_broadcastsimplify_basic_indexunbroadcastc                  h   t          dd          dk    sJ t          ddd          dk    sJ t          j        t                    5  t          dd           d d d            n# 1 swxY w Y   t          j        t                    5  t          ddd           d d d            d S # 1 swxY w Y   d S )N)
         )r
   r   )   r   r   r   )r   r   r
   r   )r
      )r   r   r   r   )r   pytestraises
ValueError     ?lib/python3.11/site-packages/astropy/utils/tests/test_shapes.pytest_check_broadcastr      s7   7D))W44447D,77=HHHH	z	"	" ' '&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 
z	"	" 5 5|4445 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s$   A$$A(+A(B''B+.B+c                  ^   t          j        g d          } t          j        | d          }t          |          }|j        dk    sJ t           j                            ||            t          j        d          } t          j        | d          }t          |          }|j        dk    sJ d S )N)r   r   r   )r   r   r   r   )r      )r   r   r   )nparraybroadcast_tor   shapetestingr   ones)xyzs      r   test_unbroadcastr"      s    
A
9%%AAA7d????JAq!!!
A
9%%AAA7fr   )      r   Z   c                   N    e Zd Zd Z e ee                    d             ZdS )TestSimplifyBasicIndexc                 p    t           | _        t          j                            t                     | _        d S )N)
TEST_SHAPEr   r   randomdata)selfs    r   setup_classz"TestSimplifyBasicIndex.setup_class*   s$    
I$$Z00			r   c                    t          || j                  }t          | j        |         | j        |                    t	          |t
                    sJ t          |          t          | j                  k    sJ t          |          D ]\  }}t	          |t          t          f          sJ t	          |t                    r	|dk    sJ At	          |j
        t                    sJ |j
        dk    sJ |j
        t          |         k     sJ |j        At	          |j        t                    sJ |j        dk    sJ |j        t          |         k    sJ t	          |j        t                    sJ d S )N)r   r   )r   r   r   r+   
isinstancetuplelen	enumeratesliceintstartr)   stopstep)r,   index	new_indexidimidxs        r   test_indexingz$TestSimplifyBasicIndex.test_indexing.   sj   (djAAA	TYu%ty';<<<)U+++++9~~TZ0000"9-- 	1 	1ID#cE3<00000#s## 
1axxxxx!#)S11111yA~~~~y:d#333338'%ch444448q====8z$'77777!#(C000000	1 	1r   N)__name__
__module____qualname__r-   r   r   r)   r<   r   r   r   r'   r'   &   sS        1 1 1 U==$$%%1 1 &%1 1 1r   r'   )numpyr   r   
hypothesisr   hypothesis.extra.numpyr   numpy.testingr   astropy.utils.shapesr   r   r   r   r"   r)   r'   r   r   r   <module>rE      s               0 0 0 0 0 0 & & & & & & S S S S S S S S S S5 5 5
 
 
 
1 1 1 1 1 1 1 1 1 1r   