a 7XhZ9@sJddlmZddlZddlZddlmZmZmZmZm Z GdddZ dS))contextmanagerN) ParserElementParseExceptionKeyword__diag__ __compat__c@s|eZdZdZGdddZGdddZedd dd eej e ej e e eej eej eej ee fe ed d d Z dS)pyparsing_testzB namespace class for classes useful in writing unit tests c@s@eZdZdZddZddZddZdd 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 cCs i|_dSN) _save_contextselfrV/var/www/viveiro_mudafortebrasil/venv/lib/python3.9/site-packages/pyparsing/testing.py__init__/sz/pyparsing_test.reset_pyparsing_context.__init__cCstj|jd<tj|jd<tj|jd<tj|jd<tj|jd<tjrRtjj |jd<n d|jd<tj |jd<tj |jd<d d t j D|jd <d tji|jd <|S)Ndefault_whitespacedefault_keyword_charsliteral_string_classverbose_stacktracepackrat_enabledpackrat_cache_size packrat_parserecursion_enabledcSsi|]}|tt|qSr)getattrr).0namerrr Hsz?pyparsing_test.reset_pyparsing_context.save..rcollect_all_And_tokensr)rDEFAULT_WHITE_CHARSr rDEFAULT_KEYWORD_CHARS_literalStringClassr_packratEnabled packrat_cachesize_parse_left_recursion_enabledr _all_namesrrr rrrsave2s&         z+pyparsing_test.reset_pyparsing_context.savecCstj|jdkr t|jd|jdt_|jdt_t|jd|jdD]\}}|rht j nt j |qVdt_ |jdrt |jdn |jd t_|jd t_|jd t_|S) NrrrrrFrrrrr)rrr Zset_default_whitespace_charsrrrinlineLiteralsUsingitemsrenabledisabler!Zenable_packratr$r%rr)r rvaluerrrrestoreRs.     z.pyparsing_test.reset_pyparsing_context.restorecCst|}|j|j|Sr )typer update)r retrrrcopyss z+pyparsing_test.reset_pyparsing_context.copycCs|Sr )r'r rrr __enter__xsz0pyparsing_test.reset_pyparsing_context.__enter__cGs |dSr )r-)r argsrrr__exit__{sz/pyparsing_test.reset_pyparsing_context.__exit__N) __name__ __module__ __qualname____doc__rr'r-r1r2r4rrrrreset_pyparsing_contexts !r9c@sLeZdZdZdddZdddZddd Zdd d Zee ddfd d Z dS)z&pyparsing_test.TestParseResultsAssertszk A mixin class to add parse results assertion methods to normal unittest.TestCase classes. NcCs<|dur|j|||d|dur8|j|||ddS)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``. Nmsg) assertEqualas_listZas_dict)r result expected_list expected_dictr;rrrassertParseResultsEqualssz?pyparsing_test.TestParseResultsAsserts.assertParseResultsEqualsTcCs@|j|dd}|r t|n t||j|||ddS)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_stringprintdumpr=rA)r expr test_stringr?r;verboser>rrrassertParseAndCheckLists  z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckListcCs@|j|dd}|r t|n t||j|||ddS)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 rFrGr@r;rHr>rrrassertParseAndCheckDicts  z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckDictc CsB|\}}|dur.|j||dur"|ndddSddt||D}|D]\}}} tdd| Dd} tdd| Dd} | dur|j| | p|d t|tr|Wdn1s0YqFtd d| Dd} td d| Dd} | | fd kr|j|| | | p |d qFtd|qF|j||dur6|ndddS)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)] Nzfailed runTestsr:cSsg|]\}}g||RqSrr)rrptexpectedrrr szOpyparsing_test.TestParseResultsAsserts.assertRunTestResults..css|]}t|tr|VqdSr ) isinstancestrrexprrr zNpyparsing_test.TestParseResultsAsserts.assertRunTestResults..css&|]}t|trt|tr|VqdSr )rOr. issubclass ExceptionrQrrrrSs)expected_exceptionr;css|]}t|tr|VqdSr )rOlistrQrrrrSrTcss|]}t|tr|VqdSr )rOdictrQrrrrSrT)NN)r?r@r;zno validation for ) assertTruezipnext assertRaisesrOrVrArD)r run_tests_reportexpected_parse_resultsr;run_test_successrun_test_resultsmergedrGr>rMfail_msgrWr?r@rrrassertRunTestResultssP  $z;pyparsing_test.TestParseResultsAsserts.assertRunTestResultsccs|durTt|trt|}|j|||d}|VWdq1sH0Yn4|j||d}|VWdn1s~0YdS)Nr:)rOrPreescapeassertRaisesRegexr])r exc_typeZ expected_msgr;ctxrrrassertRaisesParseExceptions  &zApyparsing_test.TestParseResultsAsserts.assertRaisesParseException)NNN)NT)NT)NN) r5r6r7r8rArIrKrdrrrjrrrrTestParseResultsAsserts~s   BrkNT|)indentbase_1) s start_lineend_line expand_tabseol_mark mark_spaces mark_controlrnroreturnc s|r |}ttrd|durtt|}|dkrzddttddtdd D} d | d <t| } d n0t |tfd dt tddd gD} | | }|dur|dkr|dkrtddd} | | }n | d|}|durd}|durt |}t|t |}ttd||}|dkrR||||} n dd|d|||D} | s|d St t|tdd| D} dd} | dkr| |rd ndd ddt|rdndt| ddDd}nd }|rd nd| d ddt| d Dd}d}| |rDd nd|| d d}||dfd dt| ||d!DdS)"u Helpful method for debugging a parser - prints a string with line and column numbers. (Line and column numbers are 1-based by default - if debugging a parse action, pass base_1=False, to correspond to the loc value passed to the parse action.) :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 :param indent: str | int - (optional) string to indent with line and column numbers; if an int is passed, converted to " " * indent :param base_1: bool - (optional) whether to label string using base 1; if False, string will be labeled based at 0 (default=True) :return: str - input string with leading line numbers and column number headers  NunicodecSsi|]\}}||qSrr)rcurrrr'sz4pyparsing_test.with_line_numbers..r!i$i3$i!$rmcsi|] }|qSrr)rrz)ord_mark_controlrrr0rT i $i#$) rcSsg|] }|dqS)␊rrlinerrrrNCsz4pyparsing_test.with_line_numbers..rcss|]}t|VqdSr )lenrrrrrSJrTz3pyparsing_test.with_line_numbers..rccss"|]}d|ddVqdS)zc rdNrrirrrrSPsr css |]}d|ddVqdS)z r NrrrrrrS[rTrZ 12345678900c3s0|](\}}|dd|VqdS)d:Nr)rrr)rtrn lineno_widthrrrSes)start) expandtabsrOinttypingcastrPr[range maketransordrX translatereplacerminmax splitlinessplitjoin enumerate)rprqrrrsrtrurvrnroZtranstable_mapZtblZs_linesZ max_line_lenZleadZheader0Zheader1digitsZheader2r)rtrnrr~rwith_line_numberss$                $z pyparsing_test.with_line_numbers)NNTrlNN)r5r6r7r8r9rk staticmethodrPrOptionalrboolUnionrrrrrr s2h}  r ) contextlibrrercorerrrrrr rrrrs