o
    ۾iV                     @   s>   d dl Z d dlmZ G dd de jZedkre   dS dS )    N)captured_stdoutc                   @   s   e Zd Zdd ZdS )DocsLiterallyUsageTestc                    s   t  U}dd l dd  j fdd} jfdd}t|dd	 t|dd t|dd
 |dd	dks?J |dddksHJ |dd
dksQJ W d    n1 s[w   Y  | d|  d S )Nr   c                 S   s   t N)NotImplementedErrorxn r	   a/home/ubuntu/.local/lib/python3.10/site-packages/numba/tests/doc_examples/test_literally_usage.pypower   s   z:DocsLiterallyUsageTest.test_literally_usage.<locals>.powerc                    s`   t | jjr"|jdkrtd dd S |jdkr!td dd S n fddS td	 d
d S )N   squarec                 S   s   | |  S r   r	   r   r	   r	   r
   <lambda>       zODocsLiterallyUsageTest.test_literally_usage.<locals>.ov_power.<locals>.<lambda>   cubicc                 S   s   | |  |  S r   r	   r   r	   r	   r
   r      s    c                    s
     |S r   )	literallyr   numbar	   r
   r      s   
 genericc                 S   s   | | S r   r	   r   r	   r	   r
   r   "   r   )
isinstancetypesLiteralliteral_valueprintr   r   r	   r
   ov_power   s   

z=DocsLiterallyUsageTest.test_literally_usage.<locals>.ov_powerc                    s
    | |S r   r	   r   )r   r	   r
   
test_power$   s   
z?DocsLiterallyUsageTest.test_literally_usage.<locals>.test_powerr   r      	      Q   zsquare
9
cubic
27
generic
81
)r   r   	extendingoverloadnjitr   assertEqualgetvalue)selfstdoutr   r   r	   )r   r   r
   test_literally_usage	   s"   
,z+DocsLiterallyUsageTest.test_literally_usageN)__name__
__module____qualname__r(   r	   r	   r	   r
   r      s    r   __main__)unittestnumba.tests.supportr   TestCaser   r)   mainr	   r	   r	   r
   <module>   s   3