o
    1&i:%                     @   s   d Z ddlmZmZ ddlmZmZmZ ddlm	Z	m
Z
 ddlmZ ddlZddlmZ ddlZddlZeeeG dd	 d	eZG d
d deZedkrTe  dS dS )z8Tests for gdb interacting with the DWARF numba generates    )TestCase
linux_only)	needs_gdbskip_unless_pexpectGdbMIDriver)patchMock)	datamodelN)typeofc                   @   sT   e Zd ZddiZdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd ZdS )TestGDBDwarfZ	NUMBA_OPT0c                 C   s    t tdd}| }|  |S )z6Returns True if gdb has NumPy support, False otherwiseF)debug)r   __file__Zsupports_numpyquit)selfZdriverZ	has_numpy r   EC:\wamp64\www\opt\env\Lib\site-packages\numba/tests/test_gdb_dwarf.py_gdb_has_numpy   s   zTestGDBDwarf._gdb_has_numpyc                 C   s"   d| }| j |dd| jd d S )Nznumba.tests.gdb.ZTesttest)Ztest_moduleZ
test_classZ	test_nameenvvars)Zsubprocess_test_runner_NUMBA_OPT_0_ENV)r   Ztest_modZthemodr   r   r   _subprocess_test_runner   s   

z$TestGDBDwarf._subprocess_test_runnerc                 C      |  d d S )N
test_basicr   r   r   r   r   r   &      zTestGDBDwarf.test_basicc                 C   r   )Ntest_array_argr   r   r   r   r   r   )   r   zTestGDBDwarf.test_array_argc                 C   r   )Ntest_conditional_breakpointr   r   r   r   r   r   ,   r   z(TestGDBDwarf.test_conditional_breakpointc                 C   r   )Ntest_break_on_symbolr   r   r   r   r   r   /   r   z!TestGDBDwarf.test_break_on_symbolc                 C   r   )Ntest_break_on_symbol_versionr   r   r   r   r   r    2   r   z)TestGDBDwarf.test_break_on_symbol_versionc                 C   s$   |   sd}| | | d d S )Nz"Cannot find gdb with NumPy supporttest_pretty_print)r   ZskipTestr   )r   Z_msgr   r   r   r!   5   s   
zTestGDBDwarf.test_pretty_printN)__name__
__module____qualname__r   r   r   r   r   r   r   r    r!   r   r   r   r   r      s    
r   c                   @   s\   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd ZdS )TestGDBPrettyPrinterLogicc                    sb   t  t  d}td|| _| j  dd l}G dd d}|  |jd	i d fddi d S )
N)gdbzgdb.printingzsys.modulesr   c                   @   s   e Zd Zdd ZdS )z9TestGDBPrettyPrinterLogic.setUp.<locals>.SelectedInferiorc                 S   s   t j| |}|jS N)ctc_charfrom_addressraw)r   dataZextentbufr   r   r   read_memoryV   s   zETestGDBPrettyPrinterLogic.setUp.<locals>.SelectedInferior.read_memoryN)r"   r#   r$   r.   r   r   r   r   SelectedInferiorT   s    r/   Zselected_inferiorc                      s    S r'   r   r   sir   r   <lambda>[   s    z1TestGDBPrettyPrinterLogic.setUp.<locals>.<lambda>r   )r   r   dictpatched_sysstartr&   Zconfigure_mock)r   Zmock_modulesr&   r/   r   r0   r   setUpC   s   
 zTestGDBPrettyPrinterLogic.setUpc                 C   s   | j   d S r'   )r4   stopr   r   r   r   tearDown]   s   z"TestGDBPrettyPrinterLogic.tearDownc           
         s   ddl m} G dd dG fdddG fddd G  fd	d
d
}tj}tj|t|}| }t| d|  d}|||}|	|}	|	
  S )Nr   )gdb_print_extensionc                   @   s(   e Zd Zdd Zedd Zdd ZdS )z:TestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DISubrangec                 S      || _ || _d S r'   Z_loZ_hir   lohir   r   r   __init__n      
zCTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DISubrange.__init__c                 S   s   | S r'   r   r   r   r   r   typer   s   z?TestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DISubrange.typec                 S   s   | j | jfS r'   r;   r   r   r   r   rangev   s   z@TestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DISubrange.rangeN)r"   r#   r$   r?   propertyrA   rB   r   r   r   r   
DISubrangem   s
    
rD   c                       s    e Zd Zdd Z fddZdS )zATestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DW_TAG_array_typec                 S   s   ||| _ | _d S r'   r;   r<   r   r   r   r?   z   s   zJTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DW_TAG_array_type.__init__c                    s    | j | jgS r'   r;   r   rD   r   r   fields}   s   zHTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DW_TAG_array_type.fieldsN)r"   r#   r$   r?   rF   r   rE   r   r   DW_TAG_array_typey   s    rG   c                       s,   e Zd Z fddZedd Zdd ZdS )zCTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DIDerivedType_tuplec                    s    dt |d | _|| _d S )Nr      )len_type_tuple)r   Z	the_tuplerG   r   r   r?      s   
zLTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DIDerivedType_tuple.__init__c                 S      | j S r'   )rJ   r   r   r   r   rA         zHTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DIDerivedType_tuple.typec                 S   s
   | j | S r'   )rK   r   itemr   r   r   __getitem__      
zOTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DIDerivedType_tuple.__getitem__N)r"   r#   r$   r?   rC   rA   rQ   r   rL   r   r   DIDerivedType_tuple   s
    
rS   c                       s`   e Zd Zdd Zdd Zedd Zedd Ze fd	d
Ze fddZ	edd Z
dS )zETestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Arrayc                 S   r:   r'   )_arr	_type_str)r   arrtype_strr   r   r   r?      r@   zNTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.__init__c                 S   s
   t | |S r'   )getattrrO   r   r   r   rQ      rR   zQTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.__getitem__c                 S   s
   | j jjS r'   )rT   ctypesr,   r   r   r   r   r,      s   
zJTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.datac                 S   s   | j jS r'   )rT   itemsizer   r   r   r   rZ      s   zNTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.itemsizec                        | j jS r'   )rT   shaper   rS   r   r   r\         zKTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.shapec                    r[   r'   )rT   stridesr   r]   r   r   r_      r^   zMTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.stridesc                 S   rM   r'   )rU   r   r   r   r   rA      rN   zJTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.typeN)r"   r#   r$   r?   rQ   rC   r,   rZ   r\   r_   rA   r   r]   r   r   DICompositeType_Array   s    

r`   z ())Z
numba.miscr9   r	   Zdefault_managermodelsZ
ArrayModelr
   Zget_data_typeZstructure_reprZNumbaArrayPrinterZ	to_stringstrip)
r   arrayr9   r`   ZdmmZarray_modelZ	data_typerW   Zfake_gdb_arrprinterr   )rS   rD   rG   r   get_gdb_repra   s   

z&TestGDBPrettyPrinterLogic.get_gdb_reprc                 C   s"   |  |}| t|t| d S r'   )rf   assertEqualstr)r   rd   Zgdb_printedr   r   r   check   s   
zTestGDBPrettyPrinterLogic.checkc                 C   sN   d}d}t jt jt jt jt jfD ]}t j|| |d||}| | qd S )N      dtype)	npZint8Zuint16Zint64Zfloat32Z
complex128arangereshaperi   )r   nmdtrV   r   r   r   *test_np_array_printer_simple_numeric_types   s   zDTestGDBPrettyPrinterLogic.test_np_array_printer_simple_numeric_typesc              	   C   s   d}t jd t|D ]V}t jjddt jddd}t t ||}g }|D ]+}t jd|}t j|d t|d |d }t jdd}	t	|||	}
|
|
 q+|t| }| | qd S )N   r   rH         )highsizerk   )rn   randomseedrB   randintro   prodrp   maxsliceappendtupleri   )r   Zn_tests_r\   tmpZslicesxr5   r7   stepZstrdrV   r   r   r   2test_np_array_printer_simple_numeric_types_strided   s    zLTestGDBPrettyPrinterLogic.test_np_array_printer_simple_numeric_types_stridedc                 C   s   d}d}t jdt jfdt jfgdd}t jdt jfdt jfgdd}||fD ]8}t j|| |d||}t j|| |d d|||d< d	t j|| |d d|| |d< | | q&d S )
Nrj   rk   r   yTalignFrl   d   )rn   rm   int16float64emptyrp   ro   ri   )r   rq   rr   ZalignedZ	unalignedrs   rV   r   r   r   -test_np_array_printer_simple_structured_dtype   s   "&zGTestGDBPrettyPrinterLogic.test_np_array_printer_simple_structured_dtypec                 C   s   t dg}| | d S )NZabcde)rn   rd   ri   )r   rV   r   r   r   test_np_array_printer_chr_array   s   z9TestGDBPrettyPrinterLogic.test_np_array_printer_chr_arrayc                 C   sl   d}d}t jddt jfgdd}t j|| |d||}| |}| d| | d	| | d
| d S )Nrj   rk   )r   z<U5r   Tr   rl   array[Exception:Unsupported sub-typez[unichr x 5])rn   rm   r   zerosrp   rf   assertInr   rq   rr   rs   rV   repr   r   r   -test_np_array_printer_unichr_structured_dtype   s   
zGTestGDBPrettyPrinterLogic.test_np_array_printer_unichr_structured_dtypec                 C   st   d}d}t jdt jdfdt jfgdd}t j|| |d||}| |}| d	| | d
| | d| d S )Nrj   rk   r   )   r   Tr   rl   r   r   znestedarray(int16)rn   rm   r   r   r   rp   rf   r   r   r   r   r   3test_np_array_printer_nested_array_structured_dtype   s    
zMTestGDBPrettyPrinterLogic.test_np_array_printer_nested_array_structured_dtypeN)r"   r#   r$   r6   r8   rf   ri   rt   r   r   r   r   r   r   r   r   r   r%   =   s    S	r%   __main__)__doc__Znumba.tests.supportr   r   Znumba.tests.gdb_supportr   r   r   Zunittest.mockr   r   Z
numba.corer	   numpyrn   Znumbar
   rY   r(   Zunittestr   r%   r"   mainr   r   r   r   <module>   s$    . H