OwlCyberSecurity - MANAGER
Edit File: testing.cpython-39.pyc
a )��eZ4 � @ 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 �^/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-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 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 s2 �� �� ��� � 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 � 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 )r/ )r �argsr r r �__exit__y s z/pyparsing_test.reset_pyparsing_context.__exit__N) �__name__� __module__� __qualname__�__doc__r r' r/ r3 r4 r6 r r r r �reset_pyparsing_context s !r; 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< |dur | j ||�� |d� |dur8| 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)Z assertEqual�as_list�as_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)� parse_all)rA r= N�� parse_string�print�dumpr> rC )r �expr� test_stringrA 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)rB r= NrE )r rI rJ rB r= rK r@ r r r �assertParseAndCheckDict� s z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckDictc C s" |\}}|du�rdd� t ||�D �}|D ]�\}}} tdd� | D �d�} tdd� | D �d�} | dur�| j| | pp|d�� t|t�r�|�W d � n1 s�0 Y q*tdd� | D �d�} td d� | D �d�} | | fd kr�| j|| | | p�|d � q*td �|�� q*| j||du�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 | ]\}}g |�|�R �qS r r )r Zrpt�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>� � zNpyparsing_test.TestParseResultsAsserts.assertRunTestResults.<locals>.<genexpr>c s s& | ] }t |t�rt|t�r|V qd S r )rQ r0 � issubclass� ExceptionrS r r r rU � s �)�expected_exceptionr= c s s | ]}t |t�r|V qd S r )rQ �listrS r r r rU � rV c s s | ]}t |t�r|V qd S r )rQ �dictrS r r r rU � rV )NN)rA rB r= zno validation for {!r}zfailed runTestsr< ) �zip�next� assertRaisesrQ rX rC rG �formatZ assertTrue)r Zrun_tests_reportZexpected_parse_resultsr= Zrun_test_successZrun_test_resultsZmergedrJ r@ rO Zfail_msgrY rA rB r r r �assertRunTestResults� sL ����� $�� ��z;pyparsing_test.TestParseResultsAsserts.assertRunTestResultsc c s8 | j ||d�� d V W d � n1 s*0 Y d S )Nr< )r^ )r �exc_typer= r r r �assertRaisesParseException� s zApyparsing_test.TestParseResultsAsserts.assertRaisesParseException)NNN)NT)NT)NN) r7 r8 r9 r: rC rL rN r` r r rb r r r r �TestParseResultsAsserts| s � � � � ?rc NT�|)�s� start_line�end_line� expand_tabs�eol_mark� mark_spaces� mark_control�returnc s |r | � � } �dur��dkrPt�dd� ttdd�tdd��D �d d iB �}d � n(t��fd d�ttdd ��d g D ��}| �|�} |dur�|dkr�|dkr�t�ddd��}| �|�} n | �d|�} |du r�d}|du r�t| �}t |t| ��}t t d|�|�}�dk�r | � � |d |� }n dd� | � d�|d |� D �}|�sFd S tt|���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 rV z4pyparsing_test.with_line_numbers.<locals>.<dictcomp>r �! i $ i3$ � i!$ � c s i | ] }|� �qS r r )r rn )rk r r r rV � � i $ i#$ )� rs r c S s g | ] }|d �qS )� ␊r �r �liner r r rP ' rV z4pyparsing_test.with_line_numbers.<locals>.<listcomp>rv c s s | ]}t |�V qd S r )�lenrw r r r rU , rV z3pyparsing_test.with_line_numbers.<locals>.<genexpr>�c c s s" | ]}d � d|d d �V qdS )z{}{}zc r �d N�r_ �r �ir r r rU 1 s �r{ � c s s | ]}d � |d d �V qdS )z {}r � Nr| r} r r r rU < s �r� Z 1234567890c 3 s" | ]\}}d � |�|� �V qdS )z {:{}d}:{}{}Nr| )r r~ rx )ri � lineno_widthr r rU F s �)�start)� expandtabsrR � maketransr\ �rangerZ � translate�replacery �min�max� splitlines�split�join� enumerate)re rf rg rh ri rj rk ZtblZs_linesZ max_line_lenZleadZheader0Zheader1Zheader2r )ri r� rk r �with_line_numbers� s� �� � ���� � ���� � ����z pyparsing_test.with_line_numbers)NNTrd NN) r7 r8 r9 r: r; rc � staticmethodrR �typing�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