File: //opt/python38/lib/python3.8/site-packages/pip/_vendor/pyparsing/__pycache__/testing.cpython-38.pyc
U
��e�4 � @ sB d dl mZ d dlZddlmZmZmZmZmZ G dd� d�Z dS )� )�contextmanagerN� )�
ParserElement�ParseException�Keyword�__diag__�
__compat__c @ sf e Zd ZdZG dd� d�ZG dd� d�Zed eej e
ej e
e eej e ej e ed �d
d ��Z dS )
�pyparsing_testzB
namespace class for classes useful in writing unit tests
c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )z&pyparsing_test.reset_pyparsing_contexta�
Context manager to be used when writing unit tests that modify pyparsing config values:
- packrat parsing
- bounded recursion parsing
- default whitespace characters.
- default keyword characters
- literal string auto-conversion class
- __diag__ settings
Example::
with reset_pyparsing_context():
# test that literals used to construct a grammar are automatically suppressed
ParserElement.inlineLiteralsUsing(Suppress)
term = Word(alphas) | Word(nums)
group = Group('(' + term[...] + ')')
# assert that the '()' characters are not included in the parsed tokens
self.assertParseAndCheckList(group, "(abc 123 def)", ['abc', '123', 'def'])
# after exiting context manager, literals are converted to Literal expressions again
c C s
i | _ d S �N)�
_save_context��self� r �A/tmp/pip-unpacked-wheel-uzdrexel/pip/_vendor/pyparsing/testing.py�__init__- s z/pyparsing_test.reset_pyparsing_context.__init__c C s� t j| jd< tj| jd< t j| jd< t j| jd< t j| jd< t jrRt jj | jd< n
d | jd< t j
| jd< t j | jd< d d
� t j
D �| jd < d tji| jd
< | S )N�default_whitespace�default_keyword_chars�literal_string_class�verbose_stacktrace�packrat_enabled�packrat_cache_size�
packrat_parse�recursion_enabledc S s i | ]}|t t|��qS r )�getattrr )�.0�namer r r �
<dictcomp>F s z?pyparsing_test.reset_pyparsing_context.save.<locals>.<dictcomp>r �collect_all_And_tokensr )r �DEFAULT_WHITE_CHARSr r �DEFAULT_KEYWORD_CHARS�_literalStringClassr �_packratEnabled�
packrat_cache�size�_parse�_left_recursion_enabledr �
_all_namesr r r r r r �save0 s4 �� ��
��� �
z+pyparsing_test.reset_pyparsing_context.savec C s� t j| jd kr t �| jd � | jd t _| jd t_t �| jd � | jd �� D ] \}}|rht j
nt j |� qVdt _ | jd r�t �
| jd � n | jd t _| jd
t _| jd t_| S ) Nr r r r r Fr r r r r )r r r Z set_default_whitespace_charsr r r �inlineLiteralsUsing�itemsr �enable�disabler! Zenable_packratr$ r% r r )r
r �valuer r r �restoreP s. ��� �
� z.pyparsing_test.reset_pyparsing_context.restorec C s t | �� }|j�| j� |S r
)�typer �update)r
�retr r r �copyq s
z+pyparsing_test.reset_pyparsing_context.copyc C s | � � S r
)r' r r r r � __enter__v s z0pyparsing_test.reset_pyparsing_context.__enter__c G s | � � d S r
)r- )r
�argsr r r �__exit__y s z/pyparsing_test.reset_pyparsing_context.__exit__N)
�__name__�
__module__� __qualname__�__doc__r r' r- r1 r2 r4 r r r r �reset_pyparsing_context s !r9 c @ sJ e Zd ZdZddd�Zddd�Zddd �Zdd
d �Zee dfd d
��Z
dS )z&pyparsing_test.TestParseResultsAssertszk
A mixin class to add parse results assertion methods to normal unittest.TestCase classes.
Nc C s< |dk r | j ||�� |d� |dk r8| j ||�� |d� dS )z�
Unit test assertion to compare a :class:`ParseResults` object with an optional ``expected_list``,
and compare any defined results names with an optional ``expected_dict``.
N��msg)� assertEqual�as_listZas_dict)r
�result�
expected_list�
expected_dictr; r r r �assertParseResultsEquals� s z?pyparsing_test.TestParseResultsAsserts.assertParseResultsEqualsTc C s@ |j |dd�}|r t|�� � n t|�� � | j|||d� dS )z�
Convenience wrapper assert to test a parser element and input string, and assert that
the resulting ``ParseResults.asList()`` is equal to the ``expected_list``.
T)Z parse_all)r? r; N�� parse_string�print�dumpr= rA )r
�expr� test_stringr? r; �verboser> r r r �assertParseAndCheckList� s
z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckListc C s@ |j |dd�}|r t|�� � n t|�� � | j|||d� dS )z�
Convenience wrapper assert to test a parser element and input string, and assert that
the resulting ``ParseResults.asDict()`` is equal to the ``expected_dict``.
T)�parseAll)r@ r; NrB )r
rF rG r@ r; rH r> r r r �assertParseAndCheckDict� s
z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckDictc
C s
|\}}|dk r�dd� t ||�D �}|D ]�\}}} tdd� | D �d�}
tdd� | D �d�} | dk r�| j| |
pn|d�� t|t�r�|�W 5 Q R X q(tdd� | D �d�} td d� | D �d�}
| |
fd
kr�| j|| |
|
p�|d � q(td |��� q(| j||dk r�|nd
d� dS )ah
Unit test assertion to evaluate output of ``ParserElement.runTests()``. If a list of
list-dict tuples is given as the ``expected_parse_results`` argument, then these are zipped
with the report tuples returned by ``runTests`` and evaluated using ``assertParseResultsEquals``.
Finally, asserts that the overall ``runTests()`` success value is ``True``.
:param run_tests_report: tuple(bool, [tuple(str, ParseResults or Exception)]) returned from runTests
:param expected_parse_results (optional): [tuple(str, list, dict, Exception)]
Nc S s g | ]\}}||f��qS r r )r �rpt�expectedr r r �
<listcomp>� s �zOpyparsing_test.TestParseResultsAsserts.assertRunTestResults.<locals>.<listcomp>c s s | ]}t |t�r|V qd S r
)�
isinstance�str�r �expr r r � <genexpr>� s
zNpyparsing_test.TestParseResultsAsserts.assertRunTestResults.<locals>.<genexpr>c s s& | ] }t |t�rt|t�r|V qd S r
)rO r. �
issubclass� ExceptionrQ r r r rS � s
�)�expected_exceptionr; c s s | ]}t |t�r|V qd S r
)rO �listrQ r r r rS � s
c s s | ]}t |t�r|V qd S r
)rO �dictrQ r r r rS � s
)NN)r? r@ r; zno validation for zfailed runTestsr: )�zip�next� assertRaisesrO rU rA rD �
assertTrue)r
�run_tests_report�expected_parse_resultsr; �run_test_success�run_test_results�mergedrG r> rM �fail_msgrV r? r@ r r r �assertRunTestResults� sV � ��� �
� � � �z;pyparsing_test.TestParseResultsAsserts.assertRunTestResultsc c s$ | j ||d�� d V W 5 Q R X d S )Nr: )r[ )r
�exc_typer; r r r �assertRaisesParseException� s zApyparsing_test.TestParseResultsAsserts.assertRaisesParseException)NNN)NT)NT)NN) r5 r6 r7 r8 rA rI rK rc r r re r r r r �TestParseResultsAsserts| s �
�
�
�
?rf NT�|)�s�
start_line�end_line� expand_tabs�eol_mark� mark_spaces� mark_control�returnc s4 |r | � � } |dk r�t�t|�}|dkr`dd� ttdd�tdd��D �}d |d
< t�|�}d � n0t|��t��fd d�ttdd
��d
g D ��}| � |�} |dk r�|dkr�|dkr�t�ddd��}| � |�} n | �
d|�} |dkr�d}|dkr�t | �}t |t | ��}t t
d|�|�}|dk�r4| �� |d |� } n dd� | �d�|d |� D �} | �s^d S t t|���t
dd� | D ��}
d�d } |
dk�r�| d �dd� tt
|
d d��D �� d } nd } | | d �d d� t|
d �D �� d }
| d |
d d }|
| d�� �fd d�t| |d �D �� d S )!u
Helpful method for debugging a parser - prints a string with line and column numbers.
(Line and column numbers are 1-based.)
:param s: tuple(bool, str - string to be printed with line and column numbers
:param start_line: int - (optional) starting line number in s to print (default=1)
:param end_line: int - (optional) ending line number in s to print (default=len(s))
:param expand_tabs: bool - (optional) expand tabs to spaces, to match the pyparsing default
:param eol_mark: str - (optional) string to mark the end of lines, helps visualize trailing spaces (default="|")
:param mark_spaces: str - (optional) special character to display in place of spaces
:param mark_control: str - (optional) convert non-printing control characters to a placeholding
character; valid values:
- "unicode" - replaces control chars with Unicode symbols, such as "␍" and "␊"
- any single character string - replace control characters with given string
- None (default) - string is displayed as-is
:return: str - input string with leading line numbers and column number headers
N�unicodec S s i | ]\}}||�qS r r )r �c�ur r r r s z4pyparsing_test.with_line_numbers.<locals>.<dictcomp>r �! i $ i3$ i!$ � � c s i | ]
}|� �qS r r )r rq )�ord_mark_controlr r r s � � i $ i#$ )� rw r c S s g | ] }|d �qS )� ␊r �r �liner r r rN * s z4pyparsing_test.with_line_numbers.<locals>.<listcomp>rz c s s | ]}t |�V qd S r
)�lenr{ r r r rS / s z3pyparsing_test.with_line_numbers.<locals>.<genexpr>�c c s s" | ]}d � |d d � �V qdS )zc r �d Nr �r �ir r r rS 4 s �r �
c s s | ]}d |d d � �V qdS )z r �
Nr r� r r r rS ? s r� Z
1234567890c 3 s, | ]$\}}|�� d ��d|� � � �V qdS )�d�:Nr )r r� r| )rl � lineno_widthr r rS F s �)�start)�
expandtabs�typing�castrP rY �range� maketrans�ordrW � translate�replacer} �min�max�
splitlines�split�join� enumerate)rh ri rj rk rl rm rn Ztranstable_mapZtblZs_linesZ max_line_lenZleadZheader0Zheader1Zheader2r )rl r� rv r �with_line_numbers� s� �
�
���� � ����
����z pyparsing_test.with_line_numbers)NNTrg NN)
r5 r6 r7 r8 r9 rf � staticmethodrP r� �Optional�int�boolr� r r r r r s( hq ��r )
�
contextlibr r� �corer r r r r r r r r r �<module> s