a
    3gZ4                     @   sB   d dl mZ d dlZddlmZmZmZmZmZ G dd dZ	dS )    )contextmanagerN   )ParserElementParseExceptionKeyword__diag__
__compat__c                   @   sf   e Zd ZdZG dd dZG dd dZedeej	e
 ej	e
 eeej	e ej	e ed	d
dZdS )pyparsing_testzB
    namespace class for classes useful in writing unit tests
    c                   @   s@   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dS )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   k/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/pip/_vendor/pyparsing/testing.py__init__-   s    z/pyparsing_test.reset_pyparsing_context.__init__c                 C   s   t j| jd< tj| jd< t j| jd< t j| jd< t j| jd< t jrRt jj	| jd< n
d | jd< t j
| jd< t j| jd< d	d
 tjD | jd< dtji| jd< | 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|qS 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| jd kr t | jd  | jd t _| jd t_t | jd  | jd  D ]\}}|rht	j
nt	j| qVdt _| jd rt | jd  n| jd	 t _| jd
 t _| jd 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 Zd ZdZdddZdddZddd	Zdd
dZee	dfddZ
dS )z&pyparsing_test.TestParseResultsAssertszk
        A mixin class to add parse results assertion methods to normal unittest.TestCase classes.
        Nc                 C   s<   |dur| j || |d |dur8| j || |d dS )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   s@   |j |dd}|r t|  nt|  | j|||d dS )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)rB   r=   Nparse_stringprintdumpr?   rD   )r   exprtest_stringrB   r=   verboserA   r   r   r   assertParseAndCheckList   s
    z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckListc                 C   s@   |j |dd}|r t|  nt|  | j|||d dS )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)rC   r=   NrF   )r   rJ   rK   rC   r=   rL   rA   r   r   r   assertParseAndCheckDict   s
    z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckDictc              	   C   s"  |\}}|durdd t ||D }|D ]\}}}	tdd |	D d}
tdd |	D d}|dur| j||
pp|d t|tr|W d   n1 s0    Y  q*tdd |	D d}td	d |	D d}||fd
kr| j||||
p|d q*td| q*| j||dur|ndd dS )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 qS r   r   )r   rptexpectedr   r   r   
<listcomp>   s   zOpyparsing_test.TestParseResultsAsserts.assertRunTestResults.<locals>.<listcomp>c                 s   s   | ]}t |tr|V  qd S r
   )
isinstancestrr   expr   r   r   	<genexpr>       zNpyparsing_test.TestParseResultsAsserts.assertRunTestResults.<locals>.<genexpr>c                 s   s&   | ]}t |trt|tr|V  qd S r
   )rS   r0   
issubclass	ExceptionrU   r   r   r   rW      s   )expected_exceptionr=   c                 s   s   | ]}t |tr|V  qd S r
   )rS   listrU   r   r   r   rW      rX   c                 s   s   | ]}t |tr|V  qd S r
   )rS   dictrU   r   r   r   rW      rX   )NN)rB   rC   r=   zno validation for {!r}zfailed runTestsr<   )	zipnextassertRaisesrS   rZ   rD   rH   format
assertTrue)r   run_tests_reportexpected_parse_resultsr=   run_test_successrun_test_resultsmergedrK   rA   rQ   fail_msgr[   rB   rC   r   r   r   assertRunTestResults   sL    

$z;pyparsing_test.TestParseResultsAsserts.assertRunTestResultsc                 c   s8   | j ||d d V  W d    n1 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:   rD   rM   rO   ri   r   r   rk   r   r   r   r   TestParseResultsAsserts|   s    
 
 
 
?rl   NT|)s
start_lineend_lineexpand_tabseol_markmark_spacesmark_controlreturnc                    s  |r|   } durdkrPtdd ttddtddD d	d
iB }d n(tfddttddd	g D }| |} |dur|dkr|dkrtddd}| |} n| d|} |du rd}|du rt| }t	|t| }t	t
d||}dkr|  |d | }n dd | d|d | D }|sFdS tt|t
dd |D }	dd  }
|	dkr|
ddd tt
|	d dD  d }nd}||
 ddd t|	 d  D  d }|
d|	 d    d }|| d fd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 | ]\}}||qS r   r   )r   cur   r   r   r     rX   z4pyparsing_test.with_line_numbers.<locals>.<dictcomp>r   !   i $  i3$     i!$   c                    s   i | ]
}| qS r   r   )r   rw   )rt   r   r   r     rX        i	$  i#$  )	   r|   r   c                 S   s   g | ]}|d  qS )   ␊r   r   liner   r   r   rR   '  rX   z4pyparsing_test.with_line_numbers.<locals>.<listcomp>r   c                 s   s   | ]}t |V  qd S r
   )lenr   r   r   r   rW   ,  rX   z3pyparsing_test.with_line_numbers.<locals>.<genexpr>c   c                 s   s"   | ]}d  d|d d V  qdS )z{}{}zc                                                                                                   r   d   Nra   r   ir   r   r   rW   1  s   r   
c                 s   s    | ]}d  |d d V  qdS )z         {}r   
   Nr   r   r   r   r   rW   <  s   r   
1234567890c                 3   s"   | ]\}}d  || V  qdS )z{:{}d}:{}{}Nr   )r   r   r   )rr   lineno_widthr   r   rW   F  s   )start)
expandtabsrT   	maketransr^   ranger\   	translatereplacer   minmax
splitlinessplitjoin	enumerate)rn   ro   rp   rq   rr   rs   rt   tbls_linesmax_line_lenleadheader0header1header2r   )rr   r   rt   r   with_line_numbers   s     

 

	
	
z pyparsing_test.with_line_numbers)NNTrm   NN)r7   r8   r9   r:   r;   rl   staticmethodrT   typingOptionalintboolr   r   r   r   r   r	      s(   hq      r	   )

contextlibr   r   corer   r   r   r   r   r	   r   r   r   r   <module>   s   	