o
    'ѹg5$                     @   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)debugN)r   __file__Zsupports_numpyquit)selfZdriverZ	has_numpy r   ^/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/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   _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    r0   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'   r0   r   r1   r   setUpC   s   
 zTestGDBPrettyPrinterLogic.setUpc                 C   s   | j   d S r(   )r5   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@   propertyrB   rC   r   r   r   r   
DISubrangem   s
    
rE   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   rE   r   r   fields}   s   zHTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DW_TAG_array_type.fieldsN)r#   r$   r%   r@   rG   r   rF   r   r   DW_TAG_array_typey   s    rH   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rH   r   r   r@      s   
zLTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DIDerivedType_tuple.__init__c                 S      | j S r(   )rK   r   r   r   r   rB         zHTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DIDerivedType_tuple.typec                 S   s
   | j | S r(   )rL   r   itemr   r   r   __getitem__      
zOTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DIDerivedType_tuple.__getitem__N)r#   r$   r%   r@   rD   rB   rR   r   rM   r   r   DIDerivedType_tuple   s
    
rT   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@      rA   zNTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.__init__c                 S   s
   t | |S r(   )getattrrP   r   r   r   rR      rS   zQTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.__getitem__c                 S   s
   | j jjS r(   )rU   ctypesr-   r   r   r   r   r-      s   
zJTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.datac                 S   s   | j jS r(   )rU   itemsizer   r   r   r   r[      s   zNTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.itemsizec                        | j jS r(   )rU   shaper   rT   r   r   r]         zKTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.shapec                    r\   r(   )rU   stridesr   r^   r   r   r`      r_   zMTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.stridesc                 S   rN   r(   )rV   r   r   r   r   rB      rO   zJTestGDBPrettyPrinterLogic.get_gdb_repr.<locals>.DICompositeType_Array.typeN)r#   r$   r%   r@   rR   rD   r-   r[   r]   r`   rB   r   r^   r   r   DICompositeType_Array   s    

ra   z ())Z
numba.miscr:   r	   Zdefault_managermodelsZ
ArrayModelr
   Zget_data_typeZstructure_reprZNumbaArrayPrinterZ	to_stringstrip)
r   arrayr:   ra   ZdmmZarray_modelZ	data_typerX   Zfake_gdb_arrprinterr   )rT   rE   rH   r   get_gdb_repra   s   

z&TestGDBPrettyPrinterLogic.get_gdb_reprc                 C   s"   |  |}| t|t| d S r(   )rg   assertEqualstr)r   re   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reshaperj   )r   nmdtrW   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   rI         )highsizerl   )ro   randomseedrC   randintrp   prodrq   maxsliceappendtuplerj   )r   Zn_tests_r]   tmpZslicesxr6   r8   stepZstrdrW   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 )
Nrk   rl   r   yTZalignFrm   d   )ro   rn   int16float64emptyrq   rp   rj   )r   rr   rs   ZalignedZ	unalignedrt   rW   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)ro   re   rj   )r   rW   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 )Nrk   rl   )r   z<U5r   Tr   rm   array[Exception:Unsupported sub-typez[unichr x 5])ro   rn   r   zerosrq   rg   assertInr   rr   rs   rt   rW   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 )Nrk   rl   r   )   r   Tr   rm   r   r   znestedarray(int16)ro   rn   r   r   r   rq   rg   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%   r7   r9   rg   rj   ru   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	   Znumpyro   Znumbar
   rZ   r)   Zunittestr   r&   r#   mainr   r   r   r   <module>   s$    . H