Ë
    èÿæiY  ã                   ó  — d dl Zd dlZd dlmZmZ d dlmZmZ d dlm	Z	 d dl
mZmZ d dl
mZ d dl
mZ dd	iZd
d	iZd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z  G d„ dee«      Z!e"dk(  r ejF                  «        yy)é    N)ÚjitÚnjit)ÚerrorsÚtypes)Útypeof)ÚTestCaseÚMemoryLeakMixin)Úno_pyobj_flags)Úforce_pyobj_flagsÚforceobjTÚnopythonc                 ó   — | \  }}}|||fS ©N© )ÚlÚaÚbÚcs       úu/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/tests/test_unpack_sequence.pyÚunpack_listr      s   € Ø�G€A€qˆ!Øˆq�!ˆ9Ðó    c                 ó2   — | j                   \  }}}||z   |z   S r   ©Úshape)r   ÚxÚyÚzs       r   Úunpack_shaper      s   € Ø�g‰g�G€A€qˆ!Øˆq‰5�1‰9Ðr   c                  ó0   — t        d«      \  } }}| |z   |z   S )Né   ©Úrange©r   r   r   s      r   Úunpack_ranger$      ó   € Ü�A‹h�G€A€qˆ!Øˆq‰5�1‰9Ðr   c                  ó0   — t        d«      \  } }}| |z   |z   S )Né   r!   r#   s      r   Úunpack_range_too_smallr(      r%   r   c                  ó0   — t        d«      \  } }}| |z   |z   S )Né   r!   r#   s      r   Úunpack_range_too_larger+   #   r%   r   c                  ó   — d\  } }}| |z   |z   S )N©é   r'   r    r   r#   s      r   Úunpack_tupler/   (   ó   € Ø�G€A€qˆ!Øˆq‰5�1‰9Ðr   c                  ó   — d\  } }}| |z   |z   S )N)r.   r'   r   r#   s      r   Úunpack_tuple_too_smallr2   -   s   € Ø�G€A€qˆ!Øˆq‰5�1‰9Ðr   c                  ó   — d\  } }}| |z   |z   S )N)r.   r'   r    r*   r   r#   s      r   Úunpack_tuple_too_larger4   2   ó   € Ø�G€A€qˆ!Øˆq‰5�1‰9Ðr   c                  ó   — d\  } }}| |z   |z   S )N)r.   y              @r   r#   s      r   Ú$unpack_heterogeneous_tuple_too_smallr7   7   r0   r   c                  ó   — d\  } }}| |z   |z   S )N)r.   ç      @ù              @r*   r   r#   s      r   Ú$unpack_heterogeneous_tuple_too_larger;   <   s   € Ø�G€A€qˆ!Øˆq‰5�1‰9Ðr   c                  ó   — d\  } }}| |z   |z   S )N)r.   r9   r:   r   r#   s      r   Úunpack_heterogeneous_tupler=   A   r5   r   c                  ó"   — d\  } \  }}| |z   |z   S )N)r.   )r9   r:   r   r#   s      r   Ú!unpack_nested_heterogeneous_tupler?   F   s   € Ø�I€A�vˆˆ1Øˆq‰5�1‰9Ðr   c                 ó   — | \  }}||fS r   r   )Úseqr   r   s      r   Úunpack_arbitraryrB   K   s   € Ø�D€A€qØˆaˆ4€Kr   c                  óp   — t        j                  d«      } t        j                  d«      }|| f}|\  }}||fS )Nr.   r'   ©ÚnpÚzeros)r   r   ÚtupÚalphaÚbetas        r   Ú
unpack_nrtrJ   P   s8   € Ü
�‰�‹€AÜ
�‰�‹€AØ
ˆQˆ$€CØ�K€Eˆ4Ø�$ˆ;Ðr   c                 ó4   — | |fx}\  }}|\  }}||z   |z   |z   S r   r   )r   r   r   r   r   ÚdÚes          r   Úchained_unpack_assign1rN   X   s/   € à�Q�Ð€A‰ˆˆAØ�F€QˆØˆq‰5�1‰9�q‰=Ðr   c                 ó   — | dkD  r|| }} | |fS )Nr   r   )r   r   s     r   Úconditional_swaprP   _   s   € àˆ1‚uØ�!ˆ1ˆØˆaˆ4€Kr   c                   óÎ   — e Zd Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z	d„ Z
d	„ Zd
„ Zd„ Zd„ Zd„ Zefd„Zd„ Zeefd„Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zefd„Zd„ Zd„ Zd„ Z d„ Z!d„ Z"y) Ú
TestUnpackc                 óR   — t        |«      }| j                   |«        |«       «       y r   )r   ÚassertPreciseEqual©ÚselfÚpyfuncÚcfuncs      r   Úcheck_nullary_npmzTestUnpack.check_nullary_npmh   s   € Ü�V“ˆØ×Ñ¡£©«Õ2r   c                 ó`   —  t        d¬«      |«      }| j                   |«        |«       «       y )NT©r   )r   rT   rU   s      r   Úcheck_nullary_objmodez TestUnpack.check_nullary_objmodel   s'   € Ø"”˜TÔ" 6Ó*ˆØ×Ñ¡£©«Õ2r   c                 óx   — t         } t        d¬«      |«      }g d¢}| j                   ||«       ||«      «       y )NTr[   r-   )r   r   ÚassertEqual)rV   rW   rX   r   s       r   Útest_unpack_listzTestUnpack.test_unpack_listp   s5   € ÜˆØ"”˜TÔ" 6Ó*ˆÚˆØ×Ñ™˜q›¡6¨!£9Õ-r   c                 ó"  — t         } t        t        j                  t        j                  dd¬«      fd¬«      |«      }t        j                  d¬«      j                  t
        j                  «      }| j                   ||«       ||«      «       y )Nr    ÚC©ÚdtypeÚndimÚlayoutTr[   r-   r   )	r   r   r   ÚArrayÚint32rE   rF   ÚastyperT   ©rV   rW   rX   r   s       r   Útest_unpack_shapezTestUnpack.test_unpack_shapev   sk   € Üˆð#””U—[‘[¤u§{¡{¸À3ÔGÐIØ!ô#Ø#)ó+ˆä�H‰H˜9Ô%×,Ñ,¬R¯X©XÓ6ˆØ×Ñ¡ a£©&°«)Õ4r   c                 ó  — t         } t        t        j                  t        j                  dd¬«      f«      |«      }t        j                  d¬«      j                  t
        j                  «      }| j                   ||«       ||«      «       y )Nr    ra   rb   r-   r   )	r   r   r   rf   rg   rE   rF   rh   rT   ri   s       r   Útest_unpack_shape_npmz TestUnpack.test_unpack_shape_npm}   sh   € Üˆð””e—k‘k¬¯©¸!ÀCÔHÐJó Øóˆä�H‰H˜9Ô%×,Ñ,¬R¯X©XÓ6ˆØ×Ñ¡ a£©&°«)Õ4r   c                 ó.   — | j                  t        «       y r   )r\   r$   ©rV   s    r   Útest_unpack_rangezTestUnpack.test_unpack_range„   ó   € Ø×"Ñ"¤<Õ0r   c                 ó.   — | j                  t        «       y r   )rY   r$   rn   s    r   Útest_unpack_range_npmz TestUnpack.test_unpack_range_npm‡   ó   € Ø×Ñœ|Õ,r   c                 ó.   — | j                  t        «       y r   )r\   r/   rn   s    r   Útest_unpack_tuplezTestUnpack.test_unpack_tupleŠ   rp   r   c                 ó.   — | j                  t        «       y r   )rY   r/   rn   s    r   Útest_unpack_tuple_npmz TestUnpack.test_unpack_tuple_npm�   rs   r   c                 ó.   — | j                  t        «       y r   )r\   r=   rn   s    r   Útest_unpack_heterogeneous_tuplez*TestUnpack.test_unpack_heterogeneous_tuple�   s   € Ø×"Ñ"Ô#=Õ>r   c                 ó.   — | j                  t        «       y r   )rY   r=   rn   s    r   Ú#test_unpack_heterogeneous_tuple_npmz.TestUnpack.test_unpack_heterogeneous_tuple_npm“   s   € Ø×ÑÔ9Õ:r   c                 ó.   — | j                  t        «       y r   )r\   r?   rn   s    r   Ú&test_unpack_nested_heterogeneous_tuplez1TestUnpack.test_unpack_nested_heterogeneous_tuple–   s   € Ø×"Ñ"Ô#DÕEr   c                 ó.   — | j                  t        «       y r   )rY   r?   rn   s    r   Ú*test_unpack_nested_heterogeneous_tuple_npmz5TestUnpack.test_unpack_nested_heterogeneous_tuple_npm™   s   € Ø×ÑÔ@ÕAr   c                 ó¤   — t         } t        t        j                  t        j                  ffi |¤Ž|«      }d}| j	                   ||Ž  ||Ž «       y )N)r*   é   )rN   r   r   rg   rT   )rV   ÚflagsrW   rX   Úargss        r   Útest_chained_unpack_assignz%TestUnpack.test_chained_unpack_assignœ   sG   € Ü'ˆØ8””U—[‘[¤%§+¡+Ð.Ñ8°%Ñ8¸Ó@ˆØˆØ×Ñ¡ t ©f°d¨mÕ<r   c                 ó0   — | j                  t        ¬«       y )N)r‚   )r„   r
   rn   s    r   Útest_chained_unpack_assign_npmz)TestUnpack.test_chained_unpack_assign_npm¢   s   € Ø×'Ñ'¬nÐ'Õ=r   c                 ó€   — | j                  |«      5   t        di |¤Ž|«      } |«        d d d «       y # 1 sw Y   y xY w)N)r   )ÚassertRaisesr   )rV   rW   r‚   ÚexcrX   s        r   Úcheck_unpack_errorzTestUnpack.check_unpack_error¥   s5   € à×Ñ˜sÕ#Ø$”CÑ$˜eÑ$ VÓ,ˆEÙŒG÷ $×#Ñ#ús   ’4´=c                 óX   — | j                  t        «       | j                  t        «       y r   )rŠ   r2   r7   rn   s    r   Útest_unpack_tuple_too_smallz&TestUnpack.test_unpack_tuple_too_small«   ó    € Ø×ÑÔ 6Ô7Ø×ÑÔ DÕEr   c                 ó¨   — | j                  t        t        t        j                  «       | j                  t
        t        t        j                  «       y r   )rŠ   r2   r
   r   ÚTypingErrorr7   rn   s    r   Útest_unpack_tuple_too_small_npmz*TestUnpack.test_unpack_tuple_too_small_npm¯   ó;   € Ø×ÑÔ 6¼Ü &× 2Ñ 2ô	4à×ÑÔ DÜ .´×0BÑ0Bõ	Dr   c                 óX   — | j                  t        «       | j                  t        «       y r   )rŠ   r4   r;   rn   s    r   Útest_unpack_tuple_too_largez&TestUnpack.test_unpack_tuple_too_largeµ   r�   r   c                 ó¨   — | j                  t        t        t        j                  «       | j                  t
        t        t        j                  «       y r   )rŠ   r4   r
   r   r�   r;   rn   s    r   Útest_unpack_tuple_too_large_npmz*TestUnpack.test_unpack_tuple_too_large_npm¹   r‘   r   c                 ó.   — | j                  t        «       y r   )rŠ   r(   rn   s    r   Útest_unpack_range_too_smallz&TestUnpack.test_unpack_range_too_small¿   ó   € Ø×ÑÔ 6Õ7r   c                 ó8   — | j                  t        t        «       y r   )rŠ   r(   r
   rn   s    r   Útest_unpack_range_too_small_npmz*TestUnpack.test_unpack_range_too_small_npmÂ   ó   € Ø×ÑÔ 6¼ÕGr   c                 ó.   — | j                  t        «       y r   )rŠ   r+   rn   s    r   Útest_unpack_range_too_largez&TestUnpack.test_unpack_range_too_largeÅ   r˜   r   c                 ó8   — | j                  t        t        «       y r   )rŠ   r+   r
   rn   s    r   Útest_unpack_range_too_large_npmz*TestUnpack.test_unpack_range_too_large_npmÈ   r›   r   c                 óÐ   —  t        t        j                  t        j                  ffi |¤Žt        «      }| j	                   |dd«      d«       | j	                   |dd«      d«       y )Nr*   r�   )r�   r*   r   )r   r�   )r   r   rg   rP   rT   )rV   r‚   rX   s      r   Úcheck_conditional_swapz!TestUnpack.check_conditional_swapË   sR   € Ø8””U—[‘[¤%§+¡+Ð.Ñ8°%Ñ8Ô9IÓJˆØ×Ñ¡ a¨£¨VÔ4Ø×Ñ¡ a¨£¨VÕ4r   c                 ó$   — | j                  «        y r   )r¡   rn   s    r   Útest_conditional_swapz TestUnpack.test_conditional_swapÐ   s   € Ø×#Ñ#Õ%r   c                 ó.   — | j                  t        «       y r   )r¡   r
   rn   s    r   Útest_conditional_swap_npmz$TestUnpack.test_conditional_swap_npmÓ   s   € Ø×#Ñ#¤NÕ3r   c                 ó¼   — t        d„ t        d«      D «       «      }t        |«      }t        } t	        |f«      |«      }| j                   ||«       ||«      «       y )Nc              3   óL   K  — | ]  }t        j                  |d z   «      –— Œ y­w)r.   NrD   )Ú.0Úis     r   Ú	<genexpr>z9TestUnpack.test_unpack_tuple_of_arrays.<locals>.<genexpr>×   s   è ø€ Ð6©X¨”B—H‘H˜Q ™U—O©Xùs   ‚"$r'   )Útupler"   r   rB   r   rT   )rV   rG   ÚtuptyrW   rX   s        r   Útest_unpack_tuple_of_arraysz&TestUnpack.test_unpack_tuple_of_arraysÖ   sN   € ÜÑ6¬U°1¬XÓ6Ó6ˆÜ�s“ˆÜ!ˆØ”�e�X“˜vÓ&ˆØ×Ñ¡ c£
©F°3«KÕ8r   c                 ój   — t         } t        d«      |«      }| j                   |«        |«       «       y )Nr   )rJ   r   rT   rU   s      r   Útest_unpack_nrtzTestUnpack.test_unpack_nrtÝ   s+   € ÜˆØ”�R“	˜&Ó!ˆØ×Ñ¡£©«Õ2r   c                 ó  — t         }| j                  t        j                  «      5 } t	        t
        j                  f«      |«       d d d «       | j                  dt        j                  «      «       y # 1 sw Y   Œ/xY w)Nzfailed to unpack int32)
rB   rˆ   r   r�   r   r   rg   ÚassertInÚstrÚ	exception)rV   rW   Úraisess      r   Útest_invalid_unpackzTestUnpack.test_invalid_unpackâ   sZ   € Ü!ˆØ×Ñœv×1Ñ1Ô2°fØ ŒD”%—+‘+�Ó  Ô(÷ 3à�‰Ð.´°F×4DÑ4DÓ0EÕF÷ 3Ð2ús   ¦!A5Á5A>N)#Ú__name__Ú
__module__Ú__qualname__rY   r\   r_   rj   rl   ro   rr   ru   rw   ry   r{   r}   r   r   r„   r†   Ú
ValueErrorrŠ   rŒ   r�   r“   r•   r—   rš   r�   rŸ   r¡   r£   r¥   r­   r¯   rµ   r   r   r   rR   rR   f   sµ   „ ò3ò3ò.ò5ò5ò1ò-ò1ò-ò?ò;òFòBð 0Aó =ò>ð 0AØ)óòFòDòFòDò8òHò8òHð ,=ó 5ò
&ò4ò9ò3ó
Gr   rR   Ú__main__)$ÚnumpyrE   ÚunittestÚnumbar   r   Ú
numba.corer   r   r   Únumba.tests.supportr   r	   r
   Únullary_no_pyobj_flagsr   Únullary_force_pyobj_flagsr   r   r$   r(   r+   r/   r2   r4   r7   r;   r=   r?   rB   rJ   rN   rP   rR   r¶   Úmainr   r   r   Ú<module>rÃ      s¯   ðÛ ã ß ß $Ý ß 9Ý HÝ Nà Ð&Ð Ø˜dÐ#€òò
ò
ò
ò
ò
ò
ò
ò
ò
ò
ò
ò
ò
òòô@G� (ô @GðF ˆzÒØ€H‡M�M…Oð r   