3
Y'ícZ4 ã @ sB d dl mZ d dlZddlmZmZmZmZmZ G dd dZ dS )é )ÚcontextmanagerNé )Ú
ParserElementÚParseExceptionÚKeywordÚ__diag__Ú
__compat__c @ sf e Zd ZdZG dd dZG dd dZed 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 ú6/tmp/pip-build-gk9425m9/pyparsing/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 Z
_all_namesr r )r r r r
Úsave0 s
z+pyparsing_test.reset_pyparsing_context.savec C s t j| jd kr t j| jd | jd t _| jd t_t j| jd x.| jd j D ] \}}|rjt j
nt j | qXW dt _ | jd rt j
| 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 | }|jj| j |S )N)Útyper
Úupdate)r Úretr r r
Úcopyq s
z+pyparsing_test.reset_pyparsing_context.copyc C s | j S )N)r$ )r r r r
Ú __enter__v s z0pyparsing_test.reset_pyparsing_context.__enter__c G s | j d S )N)r* )r Úargsr r r
Ú__exit__y s z/pyparsing_test.reset_pyparsing_context.__exit__N)
Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r$ r* r. r/ r1 r r r r
Úreset_pyparsing_context s !r6 c @ sJ e Zd ZdZdddZdddZddd 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 ||j |d |dk r8| j ||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_listZas_dict)r ÚresultÚ
expected_listÚ
expected_dictr7 r r r
ÚassertParseResultsEquals s z?pyparsing_test.TestParseResultsAsserts.assertParseResultsEqualsTc C s@ |j |dd}|r t|j n t|j | 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: r7 N)Ú parse_stringÚprintÚdumpr8 r< )r ÚexprÚ test_stringr: r7 Úverboser9 r r r
ÚassertParseAndCheckList s
z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckListc C s@ |j |dd}|r t|j n t|j | 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; r7 N)r= r> r? r8 r< )r r@ rA r; r7 rB r9 r r r
ÚassertParseAndCheckDict s
z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckDictc C s |\}}|dk rðdd t ||D }xÊ|D ]Â\}}} tdd | D d}
tdd | D d} | dk r| j| |
pp|d t|tr|W dQ R X q*tdd | D d} td d | D d}
| |
fdkrÞ| j|| |
|
pÖ|d
q*td j| q*W | 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 | ]\}}||fqS r r )r ZrptÚexpectedr r r
ú
<listcomp>¹ s zOpyparsing_test.TestParseResultsAsserts.assertRunTestResults.<locals>.<listcomp>c s s | ]}t |tr|V qd S )N)Ú
isinstanceÚstr)r Úexpr r r
ú <genexpr>Á s zNpyparsing_test.TestParseResultsAsserts.assertRunTestResults.<locals>.<genexpr>c s s& | ] }t |trt|tr|V qd S )N)rH r+ Ú
issubclassÚ Exception)r rJ r r r
rK Å s )Úexpected_exceptionr7 c s s | ]}t |tr|V qd S )N)rH Úlist)r rJ r r r
rK Ó s c s s | ]}t |tr|V qd S )N)rH Údict)r rJ r r r
rK Ö s )r: r; r7 zno validation for {!r}zfailed runTests)r7 )NN) ÚzipÚnextÚ assertRaisesrH rM r< r> ÚformatZ
assertTrue)r Zrun_tests_reportZexpected_parse_resultsr7 Zrun_test_successZrun_test_resultsZmergedrA r9 rF Zfail_msgrN r: r; r r r
ÚassertRunTestResults© s:
z;pyparsing_test.TestParseResultsAsserts.assertRunTestResultsc c s$ | j ||d d V W d Q R X d S )N)r7 )rS )r Úexc_typer7 r r r
ÚassertRaisesParseExceptionè s zApyparsing_test.TestParseResultsAsserts.assertRaisesParseException)NNN)NT)NT)NN) r2 r3 r4 r5 r< rC rE rU r r rW r r r r
ÚTestParseResultsAsserts| s
>rX NTú|)ÚsÚ
start_lineÚend_lineÚ expand_tabsÚeol_markÚ mark_spacesÚ mark_controlÚreturnc s |r | j } dk rdkrPtjdd ttddtddD d d
iB }d n(tjfd dttdd
d g D }| j|} |dk rÂ|dkrÂ|dkr¶tjddd}| j|} n | jd|} |dkrÎd}|dkrÞt| }t |t| }t t
d||}dkr | j |d |
}n dd | j d|d |
D }|sFd S tt|t
dd |D } dd }
| dkr¦|
d j
dd tt
| d dD d } nd } | |
d j
d d t| d D d } |
d | d d }
| |
dj
fd dt||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 rc )r` r r
r s é ú i $ i#$ )é rh r c S s g | ] }|d qS )u âr )r Úliner r r
rG ' s z4pyparsing_test.with_line_numbers.<locals>.<listcomp>u âc s s | ]}t |V qd S )N)Úlen)r rk r r r
rK , s z3pyparsing_test.with_line_numbers.<locals>.<genexpr>éc c s s& | ] }d j dd |d d V qdS )z{}{}ri rm r éd N)rT )r Úir r r
rK 2 s rn Ú
c s s | ]}d j |d d V qdS )z {}r é
N)rT )r ro r r r
rK = s rq Z
1234567890c 3 s" | ]\}}d j || V qdS )z {:{}d}:{}{}N)rT )r ro rk )r^ Ú lineno_widthr r
rK G s )Ústart)Ú
expandtabsrI Ú maketransrQ ÚrangerO Ú translateÚreplacerl ÚminÚmaxÚ
splitlinesÚsplitÚjoinÚ enumerate)rZ r[ r\ r] r^ r_ r` ZtblZs_linesZ max_line_lenZleadZheader0Zheader1Zheader2r )r^ rr r` r
Úwith_line_numbersí sH $
,,z pyparsing_test.with_line_numbers)NNTrY NN)
r2 r3 r4 r5 r6 rX Ú staticmethodrI Ú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