
    sh                         S SK rS SKJr  S SKJr  S SKJrJr  S SK	r	SS jr
SS jrSS jrSS jrSS	 jrSS
 jrS rS rS rS rS rS r " S S\5      r\S:X  a  \	R0                  " 5         gg)    N)jit)types)TestCasetagc                 L    U R                   UR                   :X  d   eU S U*  XS & g Nshapesrcdestks      l/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/tests/test_overlap.pyarray_overlap1r      s)    99

"""3QBxDH    c                 L    U R                   UR                   :X  d   eXS  US U* & g r   r	   r   s      r   array_overlap2r      s)    99

"""BD1"Ir   c                 f    U R                   UR                   :X  d   eU S S 2US 24   US S 2S U* 24'   g r   r	   r   s      r   array_overlap3r      s5    99

"""ad)D3QB3Kr   c                 f    U R                   UR                   :X  d   eU S S 2S U* 24   US S 2US 24'   g r   r	   r   s      r   array_overlap4r      s5    99

"""QssUD12Jr   c                 ^    U R                   UR                   :X  d   eU SUS 24   USS U* 24'   g N.r	   r   s      r   array_overlap5r      s5    99

"""ABKDSqbSMr   c                 ^    U R                   UR                   :X  d   eU SS U* 24   USUS 24'   g r   r	   r   s      r   array_overlap6r      s5    99

"""s3QB3w<DQRLr   c                 J    U R                   UR                   :X  d   eXS S S2'   g Nr	   r   r   s     r   array_overlap11r!   %   s#    99

"""2Jr   c                 R    U R                   UR                   :X  d   eU S S S2   US S & g r   r	   r    s     r   array_overlap12r#   )   s)    99

"""$B$iDGr   c                 R    U R                   UR                   :X  d   eXS S 2S S S24'   g r   r	   r    s     r   array_overlap13r%   -   s'    99

"""4R4Lr   c                 Z    U R                   UR                   :X  d   eU S S 2S S S24   US S & g r   r	   r    s     r   array_overlap14r'   1   s-    99

"""!DbD&kDGr   c                 N    U R                   UR                   :X  d   eXSS S S24'   g N.r   r	   r    s     r   array_overlap15r*   5   s'    99

"""TrTNr   c                 V    U R                   UR                   :X  d   eU SS S S24   US S & g r)   r	   r    s     r   array_overlap16r,   9   s-    99

"""#dd(mDGr   c                   l    \ rS rSrSS jrS rS rS rS rS r	S r
S	 rS
 rS rS rS rS rS rSrg)TestArrayOverlap>   c                 l  ^  SnS nU 4S jn[        SS9" U5      n[        US5       H  nU4U-  n	[        R                  " SXH-  5      R	                  U	5      n
[        U" U
5      U" U
5      5       H<  u  pU(       a&  [        SU5       H  nU" XX[        US95        M     M2  U" XX0 5        M>     M     g )	N   c              3      #    U R                  SS9v   U R                  SS9v   U S S S2   R                  5       S S S2   nUR                  R                  (       d  UR                  R                  (       a   eUv   g 7f)NC)orderFr   )copyflagsc_contiguousf_contiguous)origas     r   vary_layouts4TestArrayOverlap.check_overlap.<locals>.vary_layoutsC   sh     ))#)&&))#)&&TrT
!$B$'Aww++AGG4H4HHHGs   A<A>c                 L   > U " X"40 UD6  U" X340 UD6  TR                  X#5        g r   )assertPreciseEqual)pyfunccfuncpydestcdestkwargsselfs        r   check-TestArrayOverlap.check_overlap.<locals>.checkJ   s+    6,V,%)&)##F2r   T)nopythonr      )r   )r   rangenparangereshapezipdict)rE   r@   min_ndimhave_k_argumentNr<   rF   rA   ndimr
   r:   rB   rC   r   s   `             r   check_overlapTestArrayOverlap.check_overlap@   s    		3
 T"6*(A&DD4KE99Q(007D "%\$%7d9K!L""1a[fVD1IF ) &; "M 'r   c                 $    U R                  XSS9  g )NT)rP   rQ   )rT   )rE   r@   rP   s      r   check_overlap_with_k%TestArrayOverlap.check_overlap_with_k]   s    6dKr   c                 ,    U R                  [        SS9  g NrI   rP   )rW   r   rE   s    r   test_overlap1TestArrayOverlap.test_overlap1`       !!.1!=r   c                 ,    U R                  [        SS9  g rZ   )rW   r   r\   s    r   test_overlap2TestArrayOverlap.test_overlap2c   r_   r   c                 ,    U R                  [        SS9  g N   r[   )rW   r   r\   s    r   test_overlap3TestArrayOverlap.test_overlap3f   r_   r   c                 ,    U R                  [        SS9  g rd   )rW   r   r\   s    r   test_overlap4TestArrayOverlap.test_overlap4i   r_   r   c                 ,    U R                  [        SS9  g rZ   )rW   r   r\   s    r   test_overlap5TestArrayOverlap.test_overlap5l   r_   r   c                 ,    U R                  [        SS9  g rZ   )rW   r   r\   s    r   test_overlap6TestArrayOverlap.test_overlap6o   r_   r   c                 ,    U R                  [        SS9  g rZ   )rT   r!   r\   s    r   test_overlap11TestArrayOverlap.test_overlap11r       ?Q7r   c                 ,    U R                  [        SS9  g rZ   )rT   r#   r\   s    r   test_overlap12TestArrayOverlap.test_overlap12u   rt   r   c                 ,    U R                  [        SS9  g rd   )rT   r%   r\   s    r   test_overlap13TestArrayOverlap.test_overlap13x   rt   r   c                 ,    U R                  [        SS9  g rd   )rT   r'   r\   s    r   test_overlap14TestArrayOverlap.test_overlap14{   rt   r   c                 ,    U R                  [        SS9  g rZ   )rT   r*   r\   s    r   test_overlap15TestArrayOverlap.test_overlap15~   rt   r   c                 ,    U R                  [        SS9  g rZ   )rT   r,   r\   s    r   test_overlap16TestArrayOverlap.test_overlap16   rt   r    N)F)__name__
__module____qualname____firstlineno__rT   rW   r]   ra   rf   ri   rl   ro   rr   rv   ry   r|   r   r   __static_attributes__r   r   r   r.   r.   >   sI    <:L>>>>>>888888r   r.   __main__)rI   )numpyrK   numbar   
numba.corer   numba.tests.supportr   r   unittestr   r   r   r   r   r   r!   r#   r%   r'   r*   r,   r.   r   mainr   r   r   <module>r      sw       - 
  
D8x D8N zMMO r   