File: //usr/local/lib/python3.10/test/__pycache__/test_asdl_parser.cpython-310.opt-1.pyc
o
�i� � @ s� d Z ddlZddlZddlZddlmZ ddlZddlZddl Z e�
� s)e � d��eeee ���Z
ej�e
d�ZG dd� de j�ZedkrKe �� dS dS ) z+Tests for the asdl parser in Parser/asdl.py� N)�dirnamez'test irrelevant for an installed PythonZParserc @ s\ e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dS )�TestAsdlParserc C s� t j�dt� tj�dtj�td��}tj �
d|�}tj � |�}|� |� || _
| j
�tj�td��| _| �| j
�| j�d� d S )Nr �asdlzasdl.pyz Python.asdlzModule validation failed)�sys�path�insert�
parser_dir� importlib� machinery�SourceFileLoader�os�join�util�spec_from_loader�module_from_spec� exec_moduler �parse�mod�
assertTrue�check)�cls�loader�spec�module� r �2/usr/local/lib/python3.10/test/test_asdl_parser.py�
setUpClass s �
zTestAsdlParser.setUpClassc C s t jd= d S )Nr )r r )r r r r �
tearDownClass% s z TestAsdlParser.tearDownClassc C s t j| _t j| _| jj| _d S �N)r r r �types��selfr r r �setUp) s zTestAsdlParser.setUpc C s> | � | jjd� | �d| j� | �d| j� | �d| j� d S )N�Python�stmt�exprr )� assertEqualr �nameZassertInr r r r r � test_module/ s zTestAsdlParser.test_modulec C sb | j j}| �|d | jj� | �|d j| jj� | �| jd | jj� | �| jd | jj� d S )Nr Zwithitem�alias) r �dfns�assertIsInstancer ZType�valueZSumr ZProduct)r! Zdefsr r r �test_definitions5 s
z TestAsdlParser.test_definitionsc C s | j d }| �t|�d� d S )Nr) z�Product([Field(identifier, name), Field(identifier, asname, opt=True)], [Field(int, lineno), Field(int, col_offset), Field(int, end_lineno, opt=True), Field(int, end_col_offset, opt=True)]))r r& �str)r! r) r r r � test_product= s
�zTestAsdlParser.test_productc C sx | j d }| �t|j�d� | �t|jd �d� | �t|jd �d� | �t|jd �d� | �t|jd �d
� d S ) Nr$ � r zField(int, lineno)� zField(int, col_offset)� z Field(int, end_lineno, opt=True)� z$Field(int, end_col_offset, opt=True))r r& �len�
attributes�repr)r! r$ r r r �test_attributesE s
z TestAsdlParser.test_attributesc C s | j d }| �t|j �d� | �t|j�d� |j d }| �|| jj� | �t|j�d� |jd }| �|jd� | �|j d� | �
|j � |jd }| �|jd� | �|j d � | �
|j � |jd
}| �|jd � | �|j d � | � |j � | �
|j
� d S )
NZ
excepthandlerr1 r0 r r3 r% �typeZ
identifierr' r2 r$ �body)r r& r4 r5 r+ r Z Constructor�fieldsr8 r' r �optZ assertFalse�seq)r! Zehandler�consZf0�f1�f2r r r �test_constructor_fieldsM s&
z&TestAsdlParser.test_constructor_fieldsc C s@ G dd� d| j j�}|� }|�| jd � | �|jg d�� d S )Nc s< e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Z� ZS ) z2TestAsdlParser.test_visitor.<locals>.CustomVisitorc s t � �� g | _d S r )�super�__init__�names_with_seqr �� __class__r r rB h s
z;TestAsdlParser.test_visitor.<locals>.CustomVisitor.__init__c S � |j D ]}| �|� qd S r )r* �visit)r! r Zdfnr r r � visitModulel �
�z>TestAsdlParser.test_visitor.<locals>.CustomVisitor.visitModulec S s | � |j� d S r )rG r, )r! r8 r r r � visitTypep s z<TestAsdlParser.test_visitor.<locals>.CustomVisitor.visitTypec S rF r )r rG )r! �sum�tr r r �visitSums rI z;TestAsdlParser.test_visitor.<locals>.CustomVisitor.visitSumc S s$ |j D ] }|jr| j�|j� qd S r )r: r< rC �appendr' )r! r= �fr r r �visitConstructorw s
��zCTestAsdlParser.test_visitor.<locals>.CustomVisitor.visitConstructor) �__name__�
__module__� __qualname__rB rH rJ rM rP �
__classcell__r r rD r �
CustomVisitorg s rU r )�ModulerV Z Interactive� FunctionType)r Z VisitorBaserG r r& rC )r! rU �vr r r � test_visitorf s �zTestAsdlParser.test_visitorN)
rQ rR rS � classmethodr r r" r( r- r/ r7 r@ rY r r r r r s
r �__main__)�__doc__�importlib.machineryr �importlib.utilr Zos.pathr r Z sysconfigZunittestZis_python_buildZSkipTest�__file__Zsrc_baser r
r ZTestCaser rQ �mainr r r r �<module> s
n �