+
    :i4E                     "   ^ RI t ^ RIHt ^ RIt^ RIt^ RIt^ RIt^ RIHtH	t	H
t
 ^ RIHtHt ^ RIHtHt ^ RIHt ^ RIHt ^ RIHtHtHt ]'       * tR	 t ! R
 R]4      t ! R R] P                  4      t]R8X  d   ] P:                  ! 4        R# R# )    N)TestCase)jitnjittyped)typesutils)TypingErrorLoweringError)_header_lead)numpy_version)tag_32bitcaptured_stdoutc                 h    \        V 4       Uu. uF  qNK  	  pp^ pV F  pW1,          pK  	  V# u upi r   )range)nilss   &   \/var/www/html/photoedit/myenv/lib/python3.14/site-packages/numba/tests/test_comprehension.py	comp_listr      s:    !HHqHA	A	 H	 	s   
/c                   2   a  ] tR t^t o R tR tR tRtV tR# )TestListComprehensionc                   \         p\        \        P                  34      ! V4      pV P	                  V! ^4      V! ^4      4       V P	                  V! ^ 4      V! ^ 4      4       V P	                  V! R4      V! R4      4       R# )   N)r   r   r   intpassertEqual)selfpyfunccfuncs   &  r   test_comp_list$TestListComprehension.test_comp_list    sd    ejj]#F+q6!9-q6!9-rF2J/    c                   a a! \        RR7      V!3R l4       o!R pR pR pR pR pR	 pR
 pR pR p	R p
R pR pR pR pR pR pR p^	o V 3R lp^	o V 3R lpR pV!3R lpR pR pR pR pWW4WgWWWWVVVVVVVVVV.p. ROpV F4  p \        RR7      ! V4      pV P                  V! V4      V! V4      4       K6  	  V P                  \        4      ;_uu_ 4       p\        RR7      ! V4      pV! V4       RRR4       \        P                  R8  d   ^@pR# ^ pR#   \         dB     \        P                  P                  X! T4      T! T4      4        K    \         d    h i ; ii ; i  + '       g   i     L|; i) z2Tests the large number of use cases defined below Tnopythonc                 \   < V ^8  d   V # S! V ^,
          4      S! V ^,
          4      ,           #     )r   fib3s   &r   r-   7TestListComprehension.test_bulk_use_cases.<locals>.fib3+   s*    1uA;a!e,,r%   c                b    \        ^\        V 4      ^,
          4       Uu. uF  qNK  	  up# u upi )zTest basic list comprehension )r   lenxr   s   & r   list18TestListComprehension.test_bulk_use_cases.<locals>.list11   s+    $QA
343!A3444s   
,c                @    V  Uu. uF  q^8  g   K  VNK  	  up# u upi )z$Test conditional list comprehension r,   r2   ys   & r   list28TestListComprehension.test_bulk_use_cases.<locals>.list25   s      *q!EAAq***   c                @    V  Uu. uF  q^8  d   TMRNK  	  up# u upi )z Test ternary list comprehension r   r,   r6   s   & r   list38TestListComprehension.test_bulk_use_cases.<locals>.list39   s#    0121QAB&222s   c                2    \         P                  ! . RO4      # )z)Test list comprehension to np.array ctor    r+      nparrayr2   s   &r   list48TestListComprehension.test_bulk_use_cases.<locals>.list4=   s    88I&&r%   c                    \         P                  ! V  UUu. uF(  p\         P                  ! V  Uu. uF  q"NK  	  up4      NK*  	  upp4      # u upi u uppi )z0Test nested list comprehension to np.array ctor rB   r2   r7   zs   &  r   list58TestListComprehension.test_bulk_use_cases.<locals>.list5B   s;    88qAq!RXX!n!Qq!n5qABBnAs   A

AA
A
c                @    R pV  Uu. uF
  q!! V4      NK  	  up# u upi )z1Test use of inner function in list comprehension c                     V ^,           # r@   r,   rE   s   &r   innerGTestListComprehension.test_bulk_use_cases.<locals>.list6.<locals>.innerH   s    1ur%   r,   )r2   rP   rJ   s   &  r   list68TestListComprehension.test_bulk_use_cases.<locals>.list6F   s#    &'(aE!Ha(((s   c                L   a ^oV3R lpV  Uu. uF
  q!! V4      NK  	  up# u upi )z*Test use of closure in list comprehension c                    < V S,           # Nr,   r6   s   &r   rP   GTestListComprehension.test_bulk_use_cases.<locals>.list7.<locals>.innerP   s    1ur%   r,   )r2   rP   rJ   r7   s   &  @r   list78TestListComprehension.test_bulk_use_cases.<locals>.list7L   s)    A&'(aE!Ha(((s   !c                T    V  Uu. uF  q^,           NK  	  ppR pV! V4      pV# u upi )z8Test use of list comprehension as arg to inner function c                 "    V ^ ,          ^,           # r   r,   rE   s   &r   rP   GTestListComprehension.test_bulk_use_cases.<locals>.list8.<locals>.innerX   s    taxr%   r,   )r2   rJ   r   rP   qs   &    r   list88TestListComprehension.test_bulk_use_cases.<locals>.list8T   s2     !"1QA" aAH #s   %c                Z   a V  Uu. uF  q^,           NK  	  upoV3R lpV! V 4      # u upi )z1Test use of list comprehension access in closure c                 2   < V ^ ,          S^,          ,           # r   r,   )r2   r   s   &r   rP   GTestListComprehension.test_bulk_use_cases.<locals>.list9.<locals>.innera   s    tad{"r%   r,   r2   rJ   rP   r   s   &  @r   list98TestListComprehension.test_bulk_use_cases.<locals>.list9]   s.     !"1Q"A#8O	 #   (c                Z   a V  Uu. uF  q^,           NK  	  upoV3R lpV! S4      # u upi )z<Test use of list comprehension access in closure and as arg c                 L   < V  Uu. uF  qS^ ,          ,           NK  	  up# u upi r   r,   )r2   r7   r   s   & r   rP   HTestListComprehension.test_bulk_use_cases.<locals>.list10.<locals>.inneri   s#    *+,!QAaD!,,,s   !r,   rc   s   &  @r   list109TestListComprehension.test_bulk_use_cases.<locals>.list10e   s.     !"1Q"A-8O	 #rf   c                Z    V  Uu. uF  p\         P                  ! V4      NK  	  ppV# u upi )z5Test scalar array construction in list comprehension rB   r2   rJ   r   s   &  r   list119TestListComprehension.test_bulk_use_cases.<locals>.list11m   s(    &'(a!aA(H )   (c                Z    V  Uu. uF  p\         P                  ! V4      NK  	  ppV# u upi )z?Test scalar type conversion construction in list comprehension )rC   float64rm   s   &  r   list129TestListComprehension.test_bulk_use_cases.<locals>.list12r   s(    ()*1AA*H +rp   c                Z    V  Uu. uF  p\         P                  ! V4      NK  	  ppV# u upi )zGTest use of explicit numpy scalar ctor reference in list comprehension )numpyrr   rm   s   &  r   list139TestListComprehension.test_bulk_use_cases.<locals>.list13w   s)    +,-1aq!1A-H .rp   c                D    V  Uu. uF  p\        V4      NK  	  ppV# u upi )z?Test use of python scalar ctor reference in list comprehension floatrm   s   &  r   list149TestListComprehension.test_bulk_use_cases.<locals>.list14|   s$    #$%1aq1A%H &s   c                l    V  Uu. uF  p\        V4      NK  	  pp\        P                  ! V4      # u upi )znTest use of python scalar ctor reference in list comprehension followed by np array construction from the listr{   rC   rD   rm   s   &  r   list159TestListComprehension.test_bulk_use_cases.<locals>.list15   s-    #$%1aq1A%88A; &s   1c                    V  Uu. uF  p\        V4      NK  	  ppV  Uu. uF  qNK  	  pp\        P                  ! V4      p\        P                  ! V4      pWE,           # u upi u upi )zGTest type unification from np array ctors consuming list comprehension r   )r2   rJ   l1l2zeoes   &     r   list169TestListComprehension.test_bulk_use_cases.<locals>.list16   sU    $%&Aq%(AB&Q!QB"B"B7N	 's
   A
A$c                    V  UUUu. uFI  q FA  q  F9  p\         P                  ! V^,          V^,          ,           4      V8X  g   K5  WV3NK;  	  KC  	  KK  	  uppp# u upppi )z5Test complex list comprehension including math calls )rC   sqrt)r2   abcs   &   r   list179TestListComprehension.test_bulk_use_cases.<locals>.list17   sh     S11a1q!t8LPQ8Q Q1I34 () S S Ss   :AAc                R   < . pV  F  pVS8  g   K  VP                  V4       K  	  V# )z2Test loop list with outer scope var as conditionalappend)r2   rJ   r   _OUTER_SCOPE_VARs   &  r   list189TestListComprehension.test_bulk_use_cases.<locals>.list18   s-    A''HHQK  Hr%   c                B   < V  Uu. uF  qS8  g   K  VNK  	  up# u upi )z7Test list comprehension with outer scope as conditionalr,   )r2   r   r   s   & r   list199TestListComprehension.test_bulk_use_cases.<locals>.list19   s#     9q!(8$8AAq999s   c                @    V  Uu. uF  qR8X  g   K  VNK  	  up# u upi )zTest return empty list ir,   r1   s   & r   list209TestListComprehension.test_bulk_use_cases.<locals>.list20   s      /q!JAAq///r:   c                >   < V  Uu. uF  pS! V4      NK  	  up# u upi )z4Test call a jitted function in a list comprehension r,   )r2   r   r-   s   & r   list219TestListComprehension.test_bulk_use_cases.<locals>.list21   s    %&'QDGQ'''s   c                    V  Uu. uF  q^,
          NK  	  ppV  Uu. uF  q^,           NK  	  ppV U Uu. uF  q F  qV8X  g   K  V NK  	  K  	  upp # u upi u upi u upp i )zGTest create two lists comprehensions and a third walking the first two r,   )r2   r7   r   r   s   &   r   list229TestListComprehension.test_bulk_use_cases.<locals>.list22   s`     !"1QA" !"1QA" 6q!!QAvAA!Aq66 #"6s   AAA
Ac                R    V  Uu. uF  qNK  	  ppVP                  ^4       V# u upi )z/Test operation on comprehension generated list r   rI   s   &  r   list239TestListComprehension.test_bulk_use_cases.<locals>.list23   s)    AqAAHHQKH s   
$c                T    V  Uu. uF  q^8  d   \        V4      MTNK  	  ppV# u upi )zTest type promotion rz   rI   s   &  r   list249TestListComprehension.test_bulk_use_cases.<locals>.list24   s-    3451aUq)1A5H 6s   %c                     \         P                  ! . RO4      p\        ^
4       Uu. uF  q"\        V4      9  g   K  VNK  	  ppV# u upi )r@   )r@   r+         )rC   rD   r   list)r2   includedr   not_includeds   &   r   list259TestListComprehension.test_bulk_use_cases.<locals>.list25   sA    
 xx-H',RyLy!T(^4KAAyLL Ms
   AANr@   r+   rA      r   l        )r   r   
ValueErrorrC   testingassert_allclose	ExceptionassertRaisesr	   sysmaxsize)"r    r3   r8   r<   rF   rK   rR   rX   r^   rd   rj   rn   rs   rw   r|   r   r   r   r   r   r   r   r   r   r   r   fvarrefr"   raisesbitsr   r-   s"   &                               @@r   test_bulk_use_cases)TestListComprehension.test_bulk_use_cases'   s    
d		- 
	-
	5	+	3	'
	C	)	)				
	
	
	
	
		S
 	 	:	0	(	7		
	  55VVVVVVVVV	5 CT*3/  sSX6  {++v&u-E#J , ;; DD!  JJ..uSz3s8D   ,+s*   0DE-E**,EE&&E*-E=	c                z    R  p\        RR7      ! V4      p. ROpV! V4      pV! V4      pV P                  WE4       R# )c                 D    \        4       pV  Uu. uF  q"NK  	  ppV# u upi rV   )object)r7   rJ   r2   inlineds   &   r   objmode_funcATestListComprehension.test_objmode_inlining.<locals>.objmode_func   s%    A"#n!Qq!GnN %s   
T)forceobjNr?   )r   assertPreciseEqual)r    r   r"   texpectedgots   &     r   test_objmode_inlining+TestListComprehension.test_objmode_inlining   s=    	
 T"<0?Ah.r%   r,   N)	__name__
__module____qualname____firstlineno__r#   r   r   __static_attributes____classdictcell____classdict__s   @r   r   r      s     0zx
/ 
/r%   r   c                      a  ] tR t^t o RtR tR tR tR tR t	R t
R tR	 tR
 tR tR tR t]P$                  ! ]R8  R4      R 4       tR tR tR tR tR tR tR tR tR tR tR tR t R t!Rt"V t#R# ) TestArrayComprehensionFc                   VP                  RR4      pVP                  RR4      pVP                  RR4      p\        RVR7      ! V4      pV! V!  pV! V!  p	\        P                  P	                  W4       V'       d$   V P                  V	^,          P                  V4       V'       d4   V P                  RVP                  VP                  ^ ,          4      4       M2V P                  RVP                  VP                  ^ ,          4      4       V'       d5   V P                  RVP                  VP                  ^ ,          4      4       R	# R	# )
zVA generic check function that run both pyfunc, and jitted pyfunc,
and compare results.run_parallelFassert_allocate_listassert_dtypeT)r(   parallelzallocate listz@do_schedulingN)getr   rC   r   assert_array_equalr   dtypeassertIninspect_llvm
signaturesassertNotIn)
r    r!   argskwargsr   r   r   r"   pyrescress
   &&*,      r   checkTestArrayComprehension.check   s    zz.%8%zz*@%Hzz.%8T<8@d|


%%e2T!W]]L9MM/5+=+=e>N>Nq>Q+RS_e.@.@AQAQRSAT.UVMM*E,>,>u?O?OPQ?R,ST r%   c                t    R  pV P                  V^4       \        '       d   V P                  V^RR7       R# R# )c                     V ^,          p\         P                  ! \        V 4       Uu. uF  q"V,           NK  	  up4      pV# u upi r*   rC   rD   r   r   mr   r   s   &   r   comp_with_array_1HTestArrayComprehension.test_comp_with_array_1.<locals>.comp_with_array_1  :    AAq2Aa%%23AH 3   ATr   Nr   PARALLEL_SUPPORTED)r    r   s   & r   test_comp_with_array_1-TestArrayComprehension.test_comp_with_array_1  s6    	
 	

$a(JJ(!$J? r%   c                2    R  pV P                  V^^ 4       R# )c                     \         P                  ! V ) V 4      p\         P                  ! V Uu. uF  q3V8  d	   W3,          MV^,          NK  	  up4      # u upi r*   )rC   arangerD   )r   	thresholdAr2   s   &&  r   comp_with_array_2HTestArrayComprehension.test_comp_with_array_2.<locals>.comp_with_array_2  sD    		1"a A88LA9}ae!a%?LMMLs    ANr   )r    r   s   & r   test_comp_with_array_2-TestArrayComprehension.test_comp_with_array_2  s    	N 	

$a+r%   c                |    R  p^ RI Hu Hp  RVn        V P	                  V^RR7       RVn        R#   RTn        i ; i)c                     V ^,          p\         P                  ! \        V 4       Uu. uF  q"V,           NK  	  up4      pV# u upi r*   r   r   s   &   r   comp_with_array_noinlineVTestArrayComprehension.test_comp_with_array_noinline.<locals>.comp_with_array_noinline  r   r   NFTr   numba.core.inline_closurecallcoreinline_closurecallenable_inline_arraycallr   )r    r  ics   &  r   test_comp_with_array_noinline4TestArrayComprehension.test_comp_with_array_noinline  s@    	
 	32	.).B&JJ/JN)-B&B&   2 	;c                4    R  pV P                  V^RR7       R# )c                 ~    ^ .p\        V 4       F  pVP                  V4       K  	  \        P                  ! V4      pV# r   )r   r   rC   rD   )r   lstr   r   s   &   r   r  aTestArrayComprehension.test_comp_with_array_noinline_issue_6053.<locals>.comp_with_array_noinline&  s4    #C1X

1 AHr%   Tr  Nr   )r    r  s   & r   (test_comp_with_array_noinline_issue_6053?TestArrayComprehension.test_comp_with_array_noinline_issue_6053%  s    	 	

+QT
Jr%   c                t    R  pV P                  V^4       \        '       d   V P                  V^RR7       R# R# )c                     \         P                  ! \        V 4       UUu. uF$  p\        V 4       Uu. uF  q!V,          NK  	  upNK&  	  upp4      pV# u upi u uppi rV   r   r   r   jr   s   &   r   comp_nest_with_arrayNTestArrayComprehension.test_comp_nest_with_array.<locals>.comp_nest_with_array0  G    U1XFX%(3(Qq55(3XFGAH 4F   A
AA
A
Tr   Nr   )r    r  s   & r   test_comp_nest_with_array0TestArrayComprehension.test_comp_nest_with_array/  s6    	 	

'+JJ+QTJB r%   c                t    R  pV P                  V^4       \        '       d   V P                  V^RR7       R# R# )c                    \         P                  ! \        V 4       UUUu. uFF  p\        V 4       UUu. uF+  p\        V 4       Uu. uF  q1V,          V,          NK  	  upNK-  	  uppNKH  	  uppp4      pV# u upi u uppi u upppi rV   r   )r   r   r  kr   s   &    r   comp_nest_with_array_3RTestArrayComprehension.test_comp_nest_with_array_3.<locals>.comp_nest_with_array_39  sg    UZ[\U]^U]PQ%(K(QuQx8x!EAIIx8(KU]^_AH 9K^s(   BA<A7A<$B7A<<BTr   Nr   )r    r"  s   & r   test_comp_nest_with_array_32TestArrayComprehension.test_comp_nest_with_array_38  s6    	 	

)1-JJ-qtJD r%   c                |    R  p^ RI Hu Hp  RVn        V P	                  V^RR7       RVn        R#   RTn        i ; i)c                     \         P                  ! \        V 4       UUu. uF$  p\        V 4       Uu. uF  q!V,          NK  	  upNK&  	  upp4      pV# u upi u uppi rV   r   r  s   &   r   comp_nest_with_array_noinline`TestArrayComprehension.test_comp_nest_with_array_noinline.<locals>.comp_nest_with_array_noinlineB  r  r  NFTr  r  )r    r(  r  s   &  r   "test_comp_nest_with_array_noinline9TestArrayComprehension.test_comp_nest_with_array_noinlineA  sH    	 	32	.).B&JJ4a,0  2 *.B&B&r  c                2    R  pV P                  V^^
4       R# )c                 l    \         P                  ! \        W4       Uu. uF  q"NK  	  up4      pV# u upi rV   r   r   r   r   r   s   &&  r   comp_with_array_rangePTestArrayComprehension.test_comp_with_array_range.<locals>.comp_with_array_rangeO  s-    U1[1[![12AH 2s   
1Nr   )r    r/  s   & r   test_comp_with_array_range1TestArrayComprehension.test_comp_with_array_rangeN  s    	 	

(!R0r%   c                2    R  pV P                  V^^
4       R# )c                 n    \         P                  ! \        W^4       Uu. uF  q"NK  	  up4      pV# u upi r*   r   r.  s   &&  r   comp_with_array_range_and_stepbTestArrayComprehension.test_comp_with_array_range_and_step.<locals>.comp_with_array_range_and_stepV  s/    U1^4^!^45AH 5s   
2Nr   )r    r5  s   & r   #test_comp_with_array_range_and_step:TestArrayComprehension.test_comp_with_array_range_and_stepU  s    	 	

11b9r%   c                4    R  pV P                  V^
RR7       R# )c                     \         P                  ! \        V 4       Uu. uF  q^,          ^8X  g   K  VNK  	  up4      pV# u upi r*   r   r   r   r   s   &  r   comp_with_array_conditional\TestArrayComprehension.test_comp_with_array_conditional.<locals>.comp_with_array_conditional]  s7    U1X<XQ!!!X<=AH =s
   AATr  Nr   )r    r<  s   & r    test_comp_with_array_conditional7TestArrayComprehension.test_comp_with_array_conditional\  s    	 	

.
Nr%   c                4    R  pV P                  V^RR7       R# )c                     \         P                  ! \        V 4       UUu. uF3  q^,          ^8X  g   K  \        V 4       Uu. uF  q!V,          NK  	  upNK5  	  upp4      pV# u upi u uppi r*   r   r  s   &   r    comp_nest_with_array_conditionalfTestArrayComprehension.test_comp_nest_with_array_conditional.<locals>.comp_nest_with_array_conditionald  sS    U1XTXQRS3%(3(Qq55(3XTUAH 4Ts   A'
A'
 A"A'
"A'
Tr  Nr   )r    rB  s   & r   %test_comp_nest_with_array_conditional<TestArrayComprehension.test_comp_nest_with_array_conditionalc  s"    	 	

3Q(, 	 	.r%   zASetting an array element with a sequence is removed in NumPy 1.24c                >   R  pV P                  \        4      ;_uu_ 4       pV P                  V^4       RRR4       V P                  \        \        XP                  4      4       V P                  R\        VP                  4      4       R#   + '       g   i     L`; i)c                     \         P                  ! \        V 4       UUu. uF+  p\        V^,           4       Uu. uF  q!V,          NK  	  upNK-  	  upp4      pV# u upi u uppi rO   r   r  s   &   r   comp_nest_with_dependencyXTestArrayComprehension.test_comp_nest_with_dependency.<locals>.comp_nest_with_dependencyn  sK    uQxHx!%!*5*Qq55*5xHIAH 6Hs   A
A	A
A
Nzarray(undefined,r   r	   r   r   r   str	exception)r    rH  r   s   &  r   test_comp_nest_with_dependency5TestArrayComprehension.test_comp_nest_with_dependencyj  sl    	 {++vJJ0!4 ,lC(8(8$9:(#f.>.>*?@ ,+s   BB	c                <   R  pV P                  \        4      ;_uu_ 4       pV P                  V4       RRR4       V P                  \        \        XP                  4      4       V P                  R\        VP                  4      4       R#   + '       g   i     L`; i)c                      \        . RO. RO4      p \        P                  ! V  UUu. uF  w  rVNK	  	  upp4      # u uppi )r@   r?   )r   r   r   )ziprC   rD   )valr   r   s      r   comp_unsupported_iterPTestArrayComprehension.test_comp_unsupported_iter.<locals>.comp_unsupported_iterx  s1    i+C883/341Q3/00/s   ;
Nz1Unsupported iterator found in array comprehensionrJ  )r    rS  r   s   &  r   test_comp_unsupported_iter1TestArrayComprehension.test_comp_unsupported_iterw  sm    	1 {++vJJ,- ,lC(8(8$9:I&**+	- ,+s   BB	c                b    R  pV P                  V^
RR7       R pV P                  V^
RR7       R# )c                 :    . ROp\         P                  ! V4      pV# r@   )r@   r+   rA   r   rB   r   r   r   s   &  r   no_array_comp1ATestArrayComprehension.test_no_array_comp.<locals>.no_array_comp1  s    AAHr%   Fr  c                 \    . ROp\         P                  ! V4      pVP                  ^4       V# rY  )rC   rD   r   rZ  s   &  r   no_array_comp2ATestArrayComprehension.test_no_array_comp.<locals>.no_array_comp2  s$    AAHHQKHr%   TNr   )r    r[  r^  s   &  r   test_no_array_comp)TestArrayComprehension.test_no_array_comp  s4    	
 	

>2E
B	
 	

>2D
Ar%   c                0    R  pV P                  V^
4       R# )c                     \         P                  ! \        V 4       UUu. uF1  p\         P                  ! \        V 4       Uu. uF  q"NK  	  up4      NK3  	  upp4      pV# u upi u uppi rV   r   r   r7   r2   r   s   &   r   nested_array>TestArrayComprehension.test_nested_array.<locals>.nested_array  L    E!HMHq288a$91Q$9:HMNAH %:M   "A%

A A%
 A%
Nr   r    re  s   & r   test_nested_array(TestArrayComprehension.test_nested_array  s    	 	

<$r%   c                0    R  pV P                  V^ 4       R# )c                     \         P                  ! \        ^4       UUu. uF1  p\         P                  ! \        ^4       Uu. uF  q"NK  	  up4      NK3  	  upp4      pV# u upi u uppi )r   r   rd  s   &   r   re  ITestArrayComprehension.test_nested_array_with_const.<locals>.nested_array  rg  rh  Nr   ri  s   & r   test_nested_array_with_const3TestArrayComprehension.test_nested_array_with_const  s    	 	

<#r%   c                   R  p. ROpV P                  W4       V P                  V\        P                  ! V4      4       V P                  V\        V4      4       V P                  V\        P
                  ! V4      4       R# )c                 h    \         P                  ! V  Uu. uF  qV,          NK  	  up4      pV# u upi rV   rB   )r   r2   r   s   &  r   
array_compDTestArrayComprehension.test_array_comp_with_iter.<locals>.array_comp  s-    !-!Qq55!-.AH .s   /Nr   )r   rC   rD   tupler   List)r    rs  r   s   &  r   test_array_comp_with_iter0TestArrayComprehension.test_array_comp_with_iter  sX    	 

:!

:rxx{+

:uQx(

:uzz!}-r%   c                P    R  pV P                  V^
\        P                  R7       R# )c                     \         P                  ! \        V 4       Uu. uF  qNK  	  up\         P                  R 7      pV# u upi ))r   )rC   rD   r   	complex64r;  s   &  r   rs  ETestArrayComprehension.test_array_comp_with_dtype.<locals>.array_comp  s3    U1X.X!X.bllCAH /s   
Ar   N)r   rC   r{  r    rs  s   & r   test_array_comp_with_dtype1TestArrayComprehension.test_array_comp_with_dtype  s    	 	

:r
=r%   c                0    R  pV P                  V^
4       R# )c                 z    \         P                  ! \        V 4       Uu. uF  qR ,          NK  	  up4      pV# u upi )y              ?r   r;  s   &  r   rs  ITestArrayComprehension.test_array_comp_inferred_dtype.<locals>.array_comp  s1    %(3(Qb&&(34AH 4s   8Nr   r~  s   & r   test_array_comp_inferred_dtype5TestArrayComprehension.test_array_comp_inferred_dtype      	 	

:r"r%   c                0    R  pV P                  V^
4       R# )c                     \         P                  ! \        V 4       UUu. uF$  p\        V 4       Uu. uF  q!V,          NK  	  upNK&  	  upp4      pV# u upi u uppi rV   r   r  s   &   r   rs  PTestArrayComprehension.test_array_comp_inferred_dtype_nested.<locals>.array_comp  r  r  Nr   r~  s   & r   %test_array_comp_inferred_dtype_nested<TestArrayComprehension.test_array_comp_inferred_dtype_nested  r  r%   c                0    R  pV P                  V^
4       R# )c                     \         P                  ! \        V 4       UUu. uF$  p\        V 4       Uu. uF  q!V,          NK  	  upNK&  	  upp4      pV# u upi u uppi rV   r   r  s   &   r   rs  TTestArrayComprehension.test_array_comp_inferred_dtype_nested_sum.<locals>.array_comp  sG    U1XFX%(3(Qq55(3XFGAH 4Fr  Nr   r~  s   & r   )test_array_comp_inferred_dtype_nested_sum@TestArrayComprehension.test_array_comp_inferred_dtype_nested_sum  s    	
 	

:r"r%   c                   R  pRpV P                  V^
V\        P                  R7       V P                  \        4      ;_uu_ 4       p\        RR7      ! V4      pV! ^
R4       RRR4       V P                  \        RP                  \        P                  4      ,           \        XP                  4      4       V P                  RP                  \        P                  4      \        VP                  4      4       R#   + '       g   i     L; i)	c                 t    \         P                  ! \        V 4       Uu. uF  q"NK  	  up4      pW^ &   V# u upi r   r   )r   vr   arrs   &&  r   rs  YTestArrayComprehension.test_array_comp_inferred_dtype_outside_setitem.<locals>.array_comp  s4    ((uQx0x!Ax01CFJ 1s   
5g333333?r}  Tr'   y        ffffff@Nz Function({})z/(array({}, 1d, C), Literal[int](0), complex128))r   rC   r   r   r	   r   r   r   formatoperatorsetitemrK  rL  r   )r    rs  r  r   r"   s   &    r   .test_array_comp_inferred_dtype_outside_setitemETestArrayComprehension.test_array_comp_inferred_dtype_outside_setitem  s    	 

:r1277
;{++v&z2E"dO , 	?11(2B2BCC  !	
 	=DDUZZP  !	
 ,+s   C;;D	c           
     2  a ^do\        RR7      V3R l4       p\        4       ;_uu_ 4       pV! 4        RRR4       XP                  4       P                  4       p\        4       ;_uu_ 4       p\	        \
        P                  ! \        S4       Uu. uF  qUNK  	  up4      4       RRR4       XP                  4       P                  4       pV P                  W64       V P                  VR4       R#   + '       g   i     L; iu upi   + '       g   i     Lk; i)d   Tr'   c                     < \         P                  ! \        S4       U u. uF  q NK  	  up 4      p\         P                  P	                  V4       \        V4       R # u up i rV   )rC   rD   r   randomshuffleprint)r   numbersnelems     r   fooFTestArrayComprehension.test_array_comp_shuffle_sideeffect.<locals>.foo  sB    hh5<8<a<89GIIg&'N  9s   
ANz\[(\s*\d+)+\])
r   r   getvaluestripr  rC   rD   r   assertNotEqualassertRegex)r    r  gotbufr   	expectbufr   expectr  s   &      @r   "test_array_comp_shuffle_sideeffect9TestArrayComprehension.test_array_comp_shuffle_sideeffect  s    	d		 
	
 &E oo%%')"((uU|4|!A|456 ##%++- 	C(./ 
 5 s)   C.+#D
DD.C>	DD	c                4    R  pV P                  V^RR7       R# )c                     . p\         P                  ! ^.4      p\         P                  P                  V^4      pVP	                  W,           4       V# rO   )rC   rD   r  choicer   )r2   r   myListr   s   &   r   r   =TestArrayComprehension.test_empty_list_not_removed.<locals>.f  s>    AXXqc]F		  +AHHQUOHr%   Tr  Nr   )r    r   s   & r   test_empty_list_not_removed2TestArrayComprehension.test_empty_list_not_removed  s    	 	

1ad
3r%   c                4    R pV P                  V^
RR7       R# )zTest issue 3742 c                     \        ^4       Uu. uF  qNK  	   p\        P                  ! V 4      p\        V 4       F  p\        P                  ! V 4      pWV&   K  	  V# u upi rO   )r   rC   emptyzeros)r   r   rJ   s   &  r   r  ;TestArrayComprehension.test_reuse_of_array_var.<locals>.foo  sU     a!1Q!A1XHHQK!  H "s   
ATr  Nr   )r    r  s   & r   test_reuse_of_array_var.TestArrayComprehension.test_reuse_of_array_var  s    		 	

3
6r%   r,   N)r@      )$r   r   r   r   _numba_parallel_test_r   r   r   r  r  r  r$  r*  r1  r7  r>  rD  unittest
skipUnlessr   rM  rU  r`  rj  ro  rw  r  r  r  r  r  r  r  r  r   r   r   s   @r   r   r      s     !U&@,.KCE.1:O. 0)*A*A-B%$.>###
.0,47 7r%   r   __main__)r  numba.tests.supportr   r   r  rv   rC   numbar   r   r   
numba.corer   r   numba.core.errorsr	   r
   numba.core.types.functionsr   numba.np.numpy_supportr   r   r   r   r   r   r   r   r   mainr,   r%   r   <module>r     sz     ( 
    " " # 8 3 0 < <  Z O/H O/d]7X.. ]7@	 zMMO r%   