File: //usr/lib/python3/dist-packages/pip/_vendor/pyparsing/__pycache__/testing.cpython-310.pyc
o
@%NeL4 � @ sF d dl mZ d dlmZ ddlmZmZmZmZm Z G dd� d�Z
dS )� )�contextmanager)�Optional� )�
ParserElement�ParseException�Keyword�__diag__�
__compat__c @ sx e Zd ZdZG dd� d�ZG dd� d�Ze dd ed
ee d ee d e
d
edee dee defdd��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 �?/usr/lib/python3/dist-packages/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 jr)t jj | jd< nd | 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 s2 �� ��
�� �
�z+pyparsing_test.reset_pyparsing_context.savec C s� t j| jd krt �| jd � | jd t _| jd t_t �| jd � | jd �� D ]\}}|r4t j
nt j |� q+dt _ | jd rKt �
| jd � n| jd t _| jd
t _| jd t_| S ) Nr r r r r Fr r r r r )r r r � set_default_whitespace_charsr r r �inlineLiteralsUsing�itemsr �enable�disabler" �enable_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 )r0 )r �argsr r r �__exit__y s z/pyparsing_test.reset_pyparsing_context.__exit__N)
�__name__�
__module__� __qualname__�__doc__r r( r0 r4 r5 r7 r r r r �reset_pyparsing_context s ! r< c @ sR e Zd ZdZ ddd�Z ddd�Z ddd �Z dd
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@ |dur| j ||�� |d� |dur | j ||�� |d� dS 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_list�as_dict)r �result�
expected_list�
expected_dictr> r r r �assertParseResultsEquals� s
�z?pyparsing_test.TestParseResultsAsserts.assertParseResultsEqualsTc C �@ |j |dd�}|rt|�� � nt|�� � | 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)� parse_all)rC r> N�� parse_string�print�dumpr@ rE )r �expr� test_stringrC r> �verboserB r r r �assertParseAndCheckList� �
z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckListc C rF )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)rD r> NrH )r rL rM rD r> rN rB r r r �assertParseAndCheckDict� rP z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckDictc C s |\}}|dur�dd� t ||�D �}|D ]k\}}} tdd� | D �d�}
tdd� | D �d�} | durR| j| |
p7|d�� t|t�rB|�W d � n1 sLw Y qtdd� | D �d�} td d� | D �d�}
| |
fd
krx| j|| |
|
pt|d � qtd �|�� q| j||dur�|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 | ] \}}g |�|�R �qS r r )r �rpt�expectedr r r �
<listcomp>� s ��zOpyparsing_test.TestParseResultsAsserts.assertRunTestResults.<locals>.<listcomp>c s � � | ]
}t |t�r|V qd S r )�
isinstance�str�r �expr r r � <genexpr>� � � zNpyparsing_test.TestParseResultsAsserts.assertRunTestResults.<locals>.<genexpr>c s s( � | ]}t |t�rt|t�r|V qd S r )rW r1 �
issubclass� ExceptionrY r r r r[ � s � ��
�)�expected_exceptionr> c s rV r )rW �listrY r r r r[ � r\ c s rV r )rW �dictrY r r r r[ � r\ �NN)rC rD r> zno validation for {!r}zfailed runTestsr= ) �zip�next� assertRaisesrW r^ rE rJ �format�
assertTrue)r �run_tests_report�expected_parse_resultsr> �run_test_success�run_test_results�mergedrM rB rT �fail_msgr_ rC rD r r r �assertRunTestResults� sR �����
� ���� �
�z;pyparsing_test.TestParseResultsAsserts.assertRunTestResultsc c s< � | j ||d�� d V W d � d S 1 sw Y d S )Nr= )re )r �exc_typer> r r r �assertRaisesParseException� s �"�zApyparsing_test.TestParseResultsAsserts.assertRaisesParseException)NNN)NTrb ) r8 r9 r: r; rE rO rR rn r r rp r r r r �TestParseResultsAsserts| s
�
�
�
�?rq NT�|�s�
start_line�end_line� expand_tabs�eol_mark� mark_spaces� mark_control�returnc s |r| � � } �durA�dkr(t�dd� ttdd�tdd��D �d d
iB �}d � nt��fd d�ttdd
��d g D ��}| �|�} |dura|dkra|dkr[t�ddd��}| �|�} n| �d|�} |du rgd}|du rot| �}t |t| ��}t t
d|�|�}�dkr�| � � |d |� }ndd� | � d�|d |� D �}|s�d S tt|���t
dd� |D ��} d�d }
| dkr�|
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 r| )ry r r r s � � i $ i#$ )� r� r c S s g | ]}|d �qS )� ␊r �r �liner r r rU ' s z4pyparsing_test.with_line_numbers.<locals>.<listcomp>r� c s s � | ]}t |�V qd S r )�lenr� r r r r[ , s � z3pyparsing_test.with_line_numbers.<locals>.<genexpr>�c c s s$ � | ]
}d � d|d d �V qdS )z{}{}zc r �d N�rf �r �ir r r r[ 1 s
� �
�r� �
c s s" � | ] }d � |d d �V qdS )z {}r �
Nr� r� r r r r[ < s
� �
�r� �
1234567890c 3 s$ � | ]
\}}d � |�|� �V qdS )z {:{}d}:{}{}Nr� )r r� r� )rw � lineno_widthr r r[ F s
� �
�)�start)�
expandtabsrX � maketransrc �ranger` � translate�replacer� �min�max�
splitlines�split�join� enumerate)rs rt ru rv rw rx ry �tbl�s_lines� max_line_len�lead�header0�header1�header2r )rw r� ry r �with_line_numbers� s� �� �
���� �
���� �
����z pyparsing_test.with_line_numbers)NNTrr NN) r8 r9 r: r; r< rq � staticmethodrX r �int�boolr� r r r r r
s8 hq���������r
N) �
contextlibr �typingr �corer r r r r r
r r r r �<module> s