Google
Edit File: doctestcompare.cpython-39.pyc
a ��a�G � @ sb d Z ddlmZ ddlZddlZddlZzddlmZ W n e yV ddl mZ Y n0 g d�Z ze Z W n ey� eefZ Y n0 ejd dkZe�d�Ze�d�Ze�d �ZejZd d � Zd d � Zejddd�Zdd� Z e� d�Z e� d�Z G dd� de�Z G dd� de �Z!d$dd�Z"d%dd �Z#G d d � d e$�Z%d d � Z&d!d"iZ'e(d#k�r^ddlZe�)� dS )&a� lxml-based doctest output comparison. Note: normally, you should just import the `lxml.usedoctest` and `lxml.html.usedoctest` modules from within a doctest, instead of this one:: >>> import lxml.usedoctest # for XML output >>> import lxml.html.usedoctest # for HTML output To use this module directly, you must call ``lxmldoctest.install()``, which will cause doctest to use this in all subsequent calls. This changes the way output is checked and comparisons are made for XML or HTML-like content. XML or HTML content is noticed because the example starts with ``<`` (it's HTML if it starts with ``<html``). You can also use the ``PARSE_HTML`` and ``PARSE_XML`` flags to force parsing. Some rough wildcard-like things are allowed. Whitespace is generally ignored (except in attributes). In text (attributes and text in the body) you can use ``...`` as a wildcard. In an example it also matches any trailing tags in the element, though it does not match leading tags. You may create a tag ``<any>`` or include an ``any`` attribute in the tag. An ``any`` tag matches any tag, while the attribute matches any and all attributes. When a match fails, the reformatted example and gotten text is displayed (indented), and a rough diff-like output is given. Anything marked with ``+`` is in the output but wasn't supposed to be, and similarly ``-`` means its in the example but wasn't in the output. You can disable parsing on one line with ``# doctest:+NOPARSE_MARKUP`` � )�etreeN)�escape)� PARSE_HTML� PARSE_XML�NOPARSE_MARKUP�LXMLOutputChecker�LHTMLOutputChecker�install� temp_install� r r r c C s | d u r d S | � � S d S �N)�strip��v� r �9/usr/lib64/python3.9/site-packages/lxml/doctestcompare.pyr ? s r c C s t �d| �S )N� )�_norm_whitespace_re�subr r r r �norm_whitespaceE s r FT)ZrecoverZremove_blank_textc C s t �| t�S r )r Z fromstring� _html_parser��htmlr r r �html_fromstringJ s r z^<[^>]+ (at|object) z[ \t\n][ \t\n]+c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z d'dd�Z d(dd�Z d)dd�Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd*d$d%�Zd&S )+r ) ZparamZimgZarea�brZbasefont�input�base�meta�link�colc C s t jS r )r �XML��selfr r r �get_default_parserW s z$LXMLOutputChecker.get_default_parserc C s� t | dd �}|d ur | j}|} ntj}| �|||�}|sF|| |||�S z ||�}W n tjyh Y dS 0 z ||�}W n tjy� Y dS 0 | �||�S )N�_temp_override_selfF)�getattr� _temp_call_super_check_output� OutputChecker� check_output� get_parserr �XMLSyntaxError� compare_docs) r" �want�got� optionflagsZalt_selfZ super_method�parser�want_doc�got_docr r r r( Z s& � z LXMLOutputChecker.check_outputc C st d }t |@ rd S t|@ r t}nRt|@ r.tj}nB|�� �� �d�rT|�� �d�rTt}n | � |�rp| � |�rp| � � }|S )Nz<html) r r r r r r r �lower� startswith�_looks_like_markupr# )r" r, r- r. r/ r r r r) o s � �z LXMLOutputChecker.get_parserc C s |� � }|�d�o t�|� S )N�<)r r3 �_repr_re�search)r" �sr r r r4 s �z$LXMLOutputChecker._looks_like_markupc C s | � |j|j�sdS | �|j|jd�s*dS | �|j|jd�s@dS d|jvr�t|j�� �}t|j�� �}||krrdS |D ]$}| �|j| |j| d�sv dS qv|jdks�t|��rt |�}t |�}|s�|�r|r�|s�dS |� d�}|� d�} | � || �s�dS |s�|jdkr��qq�dS )NFT�anyz...r ) � tag_compare�tag� text_compare�text�tail�attrib�sorted�keys�len�list�popr+ ) r" r, r- Z want_keysZgot_keys�key� want_children� got_childrenZ want_firstZ got_firstr r r r+ � s6 z LXMLOutputChecker.compare_docsc C s^ |pd}|pd}|r,t |��� }t |��� }dt�|� }|�dd�}t�||�rVdS dS d S )N� z^%s$z\.\.\.z.*TF)r r �rer �replacer7 )r" r, r- r r r r r< � s z LXMLOutputChecker.text_comparec C sj |dkr dS t |t�r t |t�s(||kS |p.d}|p6d}|�d�r^|�d�d |�d�d kS ||kS d S )Nr9 TrH z{...}�}���)� isinstance� _basestringr3 �split)r" r, r- r r r r: � s � z LXMLOutputChecker.tag_comparec C s |j }| �|||�}g }|d ur�z ||�}W n. tjyZ t�� d }|�d| � Y n0 z ||�} W n. tjy� t�� d }|�d| � Y n0 |d u s�|r�t�| |||�} |r�|�| � d� |�S | S |t u } d| � || d�d| � | | d�d| � || | d�g} d� | �S ) N� zIn example: %szIn actual output: %s� z Expected:� zGot:zDiff:) r, r) r r* �sys�exc_info�appendr' �output_difference�joinr � format_doc� collect_diff) r" Zexampler- r. r, r/ �errorsr0 �er1 �valuer Z diff_partsr r r rV � s> � �z#LXMLOutputChecker.output_differenceTc C s. |sdS |j | jvrdS |js&t|�r*dS dS )NFT)r; � empty_tagsr= rB )r" �elr r r r �html_empty_tag� s z LXMLOutputChecker.html_empty_tagrH c C s� g }t |�s�|�d| � |�|� |�| �|�� | �||�slt|j�r\|�| �|j�� |�| �|�� t|j�r�|�| �|j�� |�d� d� |�S |�d| � |�|� |�| �|�� | �||��s^|�d� t|j��r|�d| � |�| �|j�� |�d� |D ] }|�| � |||d �� �q|�d| � |�| �|�� |�d� t|j��r�|�d| � |�| �|j�� |�d� d� |�S )Nr rQ rH rR ) rB rU � format_tagr_ r r= � format_text�format_end_tagr> rW rX )r" �docr �indent�prefix�partsr^ r r r rX � s@ z LXMLOutputChecker.format_docc C s" |d u r dS |r|� � }t|d�S )NrH rP )r � html_escape)r" r= r r r r ra s z LXMLOutputChecker.format_textc C sh g }t |tj�rdS t|j�� �D ]"\}}|�d|| �|d�f � q"|sTd|j S d|jd� |�f S )Nz<!--�%s="%s"F�<%s>�<%s %s>r ) rM r � CommentBaser@ r? �itemsrU ra r; rW )r" r^ �attrs�namer\ r r r r` s z LXMLOutputChecker.format_tagc C s t |tj�rdS d|j S )Nz-->�</%s>)rM r rk r; )r" r^ r r r rb s z LXMLOutputChecker.format_end_tagc C s g }t |�s�t |�s�|�d| � |�| �||�� | �||�sh|�| �|j|j�� |�| �||�� |�| �|j|j�� |�d� d�|�S |�d| � |�| �||�� |�d� t |j�s�t |j�r�|�d| � |�| �|j|j�� |�d� t |�}t |�}|�s|�r�|�sD|�| � |� d�||d d�� �q|�sn|�| � |� d�||d d�� �q|�| � |� d�|� d�||d �� �q|�d| � |�| �||�� |�d� t |j��s�t |j��r|�d| � |�| �|j|j�� |�d� d�|�S )Nr rQ rH r rR �+�-)rB rU �collect_diff_tagr_ �collect_diff_textr= �collect_diff_end_tagr> rW r rC rX rD rY )r" r, r- r rd rf rF rG r r r rY ! sJ � z LXMLOutputChecker.collect_diffc C s* | � |j|j�s"d|j|jf }n|j}g }|jdkp>d|jv }t|j�� �D ]p\}}||jvr�|s�|�d|| �|d�f � qN||jv r�| �|j| |d�}n | �|d�}|�d||f � qN|�st|j�� �D ].\}}||jv r�q�|�d|| �|d�f � q�|�r d|d�|�f }nd | }|S ) N� %s (got: %s)r9 z+%s="%s"Frh z-%s="%s"rj r ri ) r: r; r? r@ rl rU ra rs rW ) r" r, r- r; rm r9 rn r\ r= r r r rr G s* z"LXMLOutputChecker.collect_diff_tagc C s, |j |j kr d|j |j f }n|j }d| S )Nru ro )r; )r" r, r- r; r r r rt b s z&LXMLOutputChecker.collect_diff_end_tagc C s: | � |||�r"|sdS | �||�S d||f }| �||�S )NrH ru )r< ra )r" r, r- r r= r r r rs i s z#LXMLOutputChecker.collect_diff_textN)T)rH )T)T)�__name__� __module__� __qualname__r] r# r( r) r4 r+ r< r: rV r_ rX ra r` rb rY rr rt rs r r r r r Q s$ # &r c @ s e Zd Zdd� ZdS )r c C s t S r )r r! r r r r# r s z%LHTMLOutputChecker.get_default_parserN)rv rw rx r# r r r r r q s r c C s | r t t_ntt_dS )z� Install doctestcompare for all future doctests. If html is true, then by default the HTML parser will be used; otherwise the XML parser is used. N)r �doctestr' r r r r r r u s r c C sz | r t }nt}t� }|jd }|� }|j}||_trJ|jd j}|jj}n|jd j}|jj}t t _ t ||||||� dS )z� Use this *inside* a doctest to enable this checker for this doctest only. If html is true, then by default the HTML parser will be used; otherwise the XML parser is used. r" �checkN) r r �_find_doctest_frame�f_localsZ_checker� _IS_PYTHON_3�__func__r( Zim_funcr ry �_RestoreChecker) r � del_moduleZChecker�frame�dt_self�checker� old_checker� check_funcZchecker_check_funcr r r r � s$ �r c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )r c C sD || _ || _| j| j_|| j_|| _|| _|| _| �� | � � d S r ) r� r� � call_superr& r$ r� � clone_funcr� � install_clone�install_dt_self)r" r� r� Z new_checkerr� r� r� r r r �__init__� s z_RestoreChecker.__init__c C sJ t r&| jj| _| jj| _| jj| j_n | jj| _| jj| _| jj| j_d S r )r} r� �__code__� func_code� __globals__Z func_globalsr� r! r r r r� � s z _RestoreChecker.install_clonec C s t r| j| j_n | j| j_d S r )r} r� r� r� r! r r r �uninstall_clone� s z _RestoreChecker.uninstall_clonec C s | j j| _| | j _d S r )r� � _DocTestRunner__record_outcome� prev_funcr! r r r r� � s z _RestoreChecker.install_dt_selfc C s | j | j_d S r )r� r� r� r! r r r �uninstall_dt_self� s z!_RestoreChecker.uninstall_dt_selfc C sL | j rHdd l}|j| j = d| j v rH| j �dd�\}}|j| }t||� d S )Nr �.rP )r� rS �modules�rsplit�delattr)r" rS �package�moduleZ package_modr r r �uninstall_module� s z _RestoreChecker.uninstall_modulec O s8 | � � | �� | j`| j`| j|i |��}| �� |S r )r� r� r� r$ r& r� r� )r" �args�kw�resultr r r �__call__� s z_RestoreChecker.__call__c O s2 | � � z| j|i |��W | �� S | �� 0 d S r )r� r� r� )r" r� r� r r r r� � s �z_RestoreChecker.call_superN) rv rw rx r� r� r� r� r� r� r� r� r r r r r � s r c C s< dd l } | �d�}|r0|j}d|v r(|S |j}qtd��d S )Nr rP ZBOOMzBCould not find doctest (only use this function *inside* a doctest))rS � _getframer| �f_back� LookupError)rS r� �lr r r r{ � s �r{ Zbasicai >>> temp_install() >>> print """<xml a="1" b="2">stuff</xml>""" <xml b="2" a="1">...</xml> >>> print """<xml xmlns="http://example.com"><tag attr="bar" /></xml>""" <xml xmlns="..."> <tag attr="..." /> </xml> >>> print """<xml>blahblahblah<foo /></xml>""" # doctest: +NOPARSE_MARKUP, +ELLIPSIS <xml>...foo /></xml> �__main__)F)FN)*�__doc__Zlxmlr rS rI ry r r rg � ImportErrorZcgi�__all__Z basestringrN � NameError�str�bytes� version_infor} Zregister_optionflagr r r r' r r Z HTMLParserr r �compiler6 r r r r r �objectr r{ Z__test__rv Ztestmodr r r r �<module> sJ % " '6 �