HEX

File: //usr/local/lib64/python3.6/site-packages/lxml/__pycache__/doctestcompare.cpython-36.pyc
3

��?e�G�@spdZddlmZddlZddlZddlZyddlmZWn e	k
rXddl
mZYnXddddd	d
dgZyeZ
Wnek
r�eefZ
YnXejddkZejd�Zejd�Zejd�ZejZd
d�Zdd�Zejddd�Zdd�Zejd�Zejd�ZGdd�de�Z Gdd	�d	e �Z!d#dd
�Z"d$dd�Z#Gdd�de$�Z%dd�Z&d d!iZ'e(d"k�rlddlZej)�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�cCs|dkrdS|j�SdS)N)�strip)�v�r�3/tmp/pip-build-_veujq9v/lxml/lxml/doctestcompare.pyr?srcCstjd|�S)N� )�_norm_whitespace_re�sub)r
rrr�norm_whitespaceEsrFT)ZrecoverZremove_blank_textcCstj|t�S)N)r�
fromstring�_html_parser)�htmlrrr�html_fromstringJsrz^<[^>]+ (at|object) z[ \t\n][ \t\n]+c
@s�eZdZd0Zdd�Zd
d�Zdd�Zdd�Zdd�Zdd�Z	dd�Z
dd�Zd1dd�Zd2dd �Z
d3d!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�Zd4d-d.�Zd/S)5r�param�img�area�br�basefont�input�base�meta�link�colcCstjS)N)r�XML)�selfrrr�get_default_parserWsz$LXMLOutputChecker.get_default_parserc	Cs�t|dd�}|dk	r |j}|}ntj}|j|||�}|sF|||||�Sy||�}Wntjk
rhdSXy||�}Wntjk
r�dSX|j||�S)N�_temp_override_selfF)�getattr�_temp_call_super_check_output�
OutputChecker�check_output�
get_parserr�XMLSyntaxError�compare_docs)	r#�want�got�optionflagsZalt_selfZsuper_method�parser�want_doc�got_docrrrr)Zs$zLXMLOutputChecker.check_outputcCstd}t|@rdSt|@rt}nRt|@r.tj}nB|j�j�jd�rT|j�jd�rTt}n|j	|�rp|j	|�rp|j
�}|S)Nz<html)rrrrrr"r�lower�
startswith�_looks_like_markupr$)r#r-r.r/r0rrrr*os

zLXMLOutputChecker.get_parsercCs|j�}|jd�otj|�S)N�<)rr4�_repr_re�search)r#�srrrr5s
z$LXMLOutputChecker._looks_like_markupc
Cs|j|j|j�sdS|j|j|jd�s*dS|j|j|jd�s@dSd|jkr�t|jj��}t|jj��}||krrdSx*|D]"}|j|j||j|d�sxdSqxW|jdks�t|��rt	|�}t	|�}xV|s�|�r|s�|r�dS|j
d�}|j
d�}	|j||	��sdS|r�|jdkr�Pq�WdS)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_firstrrrr,�s6



zLXMLOutputChecker.compare_docscCs^|pd}|pd}|r,t|�j�}t|�j�}dtj|�}|jdd�}tj||�rVdSdSdS)N�z^%s$z\.\.\.z.*TF)rr�rer�replacer8)r#r-r.rrrrr=�szLXMLOutputChecker.text_comparecCsn|dkrdSt|t�s$t|t�r,||kS|p2d}|p:d}|jd�rb|jd�d|jd�dkS||kSdS)	Nr:TrIz{...}�}����rN)�
isinstance�_basestringr4�split)r#r-r.rrrr;�s
zLXMLOutputChecker.tag_comparec
Cs|j}|j|||�}g}|dk	r�y||�}Wn0tjk
r\tj�d}|jd|�YnXy||�}	Wn0tjk
r�tj�d}|jd|�YnX|dks�|r�tj||||�}
|r�|j|
�dj	|�S|
S|t
k}d|j||d�d|j|	|d�d|j||	|d�g}dj	|�S)	NrMzIn example: %szIn actual output: %s�
z	Expected:�zGot:zDiff:)
r-r*rr+�sys�exc_info�appendr(�output_difference�joinr�
format_doc�collect_diff)
r#Zexampler.r/r-r0�errorsr1�er2�valuerZ
diff_partsrrrrW�s:

z#LXMLOutputChecker.output_differenceTcCs.|sdS|j|jkrdS|js&t|�r*dSdS)NFT)r<�
empty_tagsr>rC)r#�elrrrr�html_empty_tag�sz LXMLOutputChecker.html_empty_tagrIcCs�g}t|�s�|jd|�|j|�|j|j|��|j||�slt|j�r\|j|j|j��|j|j|��t|j�r�|j|j|j��|jd�dj	|�S|jd|�|j|�|j|j|��|j||��sb|jd�t|j��r|jd|�|j|j|j��|jd�x&|D]}|j|j
|||d���qW|jd|�|j|j|��|jd�t|j��r�|jd|�|j|j|j��|jd�dj	|�S)NrrRrIrS)rCrV�
format_tagr`rr>�format_text�format_end_tagr?rXrY)r#�docr�indent�prefix�partsr_rrrrY�s@










zLXMLOutputChecker.format_doccCs"|dkrdS|r|j�}t|d�S)NrIrM)r�html_escape)r#r>rrrrrb	s
zLXMLOutputChecker.format_textcCslg}t|tj�rdSx4t|jj��D]"\}}|jd||j|d�f�q$W|sXd|jSd|jdj	|�fS)Nz<!--z%s="%s"Fz<%s>z<%s %s>r)
rOr�CommentBaserAr@�itemsrVrbr<rX)r#r_�attrs�namer]rrrras
zLXMLOutputChecker.format_tagcCst|tj�rdSd|jS)Nz-->z</%s>)rOrrir<)r#r_rrrrcsz LXMLOutputChecker.format_end_tagcCsg}t|�r�t|�r�|jd|�|j|j||��|j||�sl|j|j|j|j��|j|j||��|j|j|j|j��|jd�dj|�S|jd|�|j|j||��|jd�t	|j�s�t	|j��r|jd|�|j|j|j|j��|jd�t
|�}t
|�}x�|�s"|�r�|�sL|j|j|jd�||dd���q|�sv|j|j|jd�||dd���q|j|j
|jd�|jd�||d���qW|jd|�|j|j||��|jd�t	|j��s�t	|j��r|jd|�|j|j|j|j��|jd�dj|�S)NrrRrIrrS�+�-)rCrV�collect_diff_tagr`�collect_diff_textr>�collect_diff_end_tagr?rXrrDrYrErZ)r#r-r.rrergrGrHrrrrZ!sH



  $

zLXMLOutputChecker.collect_diffc	Cs4|j|j|j�s"d|j|jf}n|j}g}|jdkp>d|jk}x�t|jj��D]r\}}||jkr�|r�|jd||j|d�f�qP||jkr�|j|j||d�}n|j|d�}|jd||f�qPW|�sx@t|jj��D].\}}||jkr�q�|jd||j|d�f�q�W|�r(d|dj|�f}nd	|}|S)
Nz%s (got: %s)r:z+%s="%s"Fz%s="%s"z-%s="%s"z<%s %s>rz<%s>)	r;r<r@rArjrVrbrprX)	r#r-r.r<rkr:rlr]r>rrrroGs*

z"LXMLOutputChecker.collect_diff_tagcCs,|j|jkrd|j|jf}n|j}d|S)Nz%s (got: %s)z</%s>)r<)r#r-r.r<rrrrqbsz&LXMLOutputChecker.collect_diff_end_tagcCs:|j|||�r"|sdS|j||�Sd||f}|j||�S)NrIz%s (got: %s))r=rb)r#r-r.rr>rrrrpisz#LXMLOutputChecker.collect_diff_textN)
rrrrrrrrr r!)T)rI)T)T)�__name__�
__module__�__qualname__r^r$r)r*r5r,r=r;rWr`rYrbrarcrZrorqrprrrrrQs&
 


#
&c@seZdZdd�ZdS)rcCstS)N)r)r#rrrr$rsz%LHTMLOutputChecker.get_default_parserN)rrrsrtr$rrrrrqscCs|rtt_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)rrrrr	usc	Csz|r
t}nt}t�}|jd}|�}|j}||_trJ|jdj}|jj}n|jdj}|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)rr�_find_doctest_frame�f_localsZ_checker�_IS_PYTHON_3�__func__r)Zim_funcrru�_RestoreChecker)	r�
del_moduleZChecker�frame�dt_self�checker�old_checker�
check_funcZchecker_check_funcrrrr
�s"


c@sLeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dS)r{cCsD||_||_|j|j_||j_||_||_||_|j�|j	�dS)N)
r~r�
call_superr'r%r��
clone_funcr|�
install_clone�install_dt_self)r#r~r�Znew_checkerr�r�r|rrr�__init__�s
z_RestoreChecker.__init__cCsJtr&|jj|_|jj|_|jj|j_n |jj|_|jj|_|jj|j_dS)N)ryr��__code__�	func_code�__globals__Zfunc_globalsr�)r#rrrr��s



z_RestoreChecker.install_clonecCstr|j|j_n
|j|j_dS)N)ryr�r�r�)r#rrr�uninstall_clone�sz_RestoreChecker.uninstall_clonecCs|jj|_||j_dS)N)r~�_DocTestRunner__record_outcome�	prev_func)r#rrrr��s
z_RestoreChecker.install_dt_selfcCs|j|j_dS)N)r�r~r�)r#rrr�uninstall_dt_self�sz!_RestoreChecker.uninstall_dt_selfcCsL|jrHddl}|j|j=d|jkrH|jjdd�\}}|j|}t||�dS)Nr�.rM)r|rT�modules�rsplit�delattr)r#rT�package�moduleZpackage_modrrr�uninstall_module�s


z _RestoreChecker.uninstall_modulecOs4|j�|j�|j`|j`|j||�}|j�|S)N)r�r�rr%r'r�r�)r#�args�kw�resultrrr�__call__�sz_RestoreChecker.__call__c
Os$|j�z|j||�S|j�XdS)N)r�r�r�)r#r�r�rrrr��sz_RestoreChecker.call_superN)rrrsrtr�r�r�r�r�r�r�r�rrrrr{�s	r{cCs@ddl}|jd�}x |r2|j}d|kr*|S|j}qWtd��dS)NrrMZBOOMzBCould not find doctest (only use this function *inside* a doctest))rT�	_getframerx�f_back�LookupError)rTr}�lrrrrw�s

rw�basicai
    >>> 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__�lxmlrrTrJrurrrh�ImportError�cgi�__all__�
basestringrP�	NameError�str�bytes�version_inforyZregister_optionflagrrrr(rr�
HTMLParserrr�compiler7rrrr	r
�objectr{rwZ__test__rr�testmodrrrr�<module>$sL





"

'6