
    sh                         S SK rS SKrS SKJr  S SKJrJr  S SKJ	r	  S r
S r " S S\	5      r\S	:X  a  \R                  " 5         gg)
    N)njit)typeserrors)TestCasec                     X@XU2'   g )N )astartstopstepscalars        o/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/tests/test_storeslice.pysetitem_slicer   	   s    eo    c                     US-  n[         R                  " U5      U SU& SX'   [         R                  " X-
  S-
  5      XS-   S & g )N   r   iA     )nparange)obsnPointscenters      r   usecaser      sE    \FIIf%C&MCK7#3a#78C!r   c                        \ rS rSrS rS rSrg)TestStoreSlice   c                    Sn[         R                  " U5      nUR                  5       n[        [        R
                  S S  [        R                  45      " [        5      nU" X!5        [        X15        U R                  X#5        g )N
   )	r   zeroscopyr   r   float64intpr   assertPreciseEqual)selfnobs_gotobs_expectedcfuncs        r   test_usecaseTestStoreSlice.test_usecase   s^    ((1+||~emmA&

34W=g 6r   c           
      d   Sn[         R                  S S  [         R                  [         R                  [         R                  [         R                  4n[        U5      " [        5      n[        R
                  " U[        R                  S9nSSS[        U5      SS4SS[        U5      -  SS4SS	S
SSS4
nU Ht  u  pgp[        R
                  " U[        R                  S9n[        R
                  " U[        R                  S9n
U" XFXxU	5        [        XXxU	5        U R                  XJ5        Mv     [        R
                  " U[        R                  S9nU R                  [        5       nU" USSSS5        S S S 5        U R                  [        WR                  5      S5        g ! , (       d  f       N4= f)Nr   )dtype)r      r      )r   r-   r.   r   M   )r0      X   )r0   r2   i'  )r2   r0      r4   )r2   r0   r4   )   r   r   r4   )r8   r   r0   r4   r3   r-   r   r4   zslice step cannot be zero)r   int64r   r   r   r   lenr#   assertRaises
ValueErrorassertEqualstr	exception)r$   r%   argtysr(   r	   testsr
   r   r   r   bcms               r   test_array_slice_setitem'TestStoreSlice.test_array_slice_setitem!   sT   ++a.%++u{{EKK++V]+IIarxx(c!fa$a#a&j!R( # ! !
 */%E		!288,A		!288,A!D/!D7##A) */ IIarxx(z*b!Q1b! +R\\*,GH +*s   &F!!
F/r   N)__name__
__module____qualname____firstlineno__r)   rD   __static_attributes__r   r   r   r   r      s    	7Ir   r   __main__)numpyr   unittestnumbar   
numba.corer   r   numba.tests.supportr   r   r   r   rF   mainr   r   r   <module>rR      sE       $ ( 9+IX +I\ zMMO r   