3 dZ4@sBddlmZddlZddlmZmZmZmZmZGdddZ dS))contextmanagerN) ParserElementParseExceptionKeyword__diag__ __compat__c @sfeZdZdZGdddZGdddZed eej e ej e e eej eej eed 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|_dS)N) _save_context)selfr 6/tmp/pip-build-7vycvbft/pyparsing/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).0namer r r Fsz?pyparsing_test.reset_pyparsing_context.save..rcollect_all_And_tokensr)rDEFAULT_WHITE_CHARSr rDEFAULT_KEYWORD_CHARS_literalStringClassr_packratEnabled packrat_cachesize_parse_left_recursion_enabledrZ _all_namesrr)r r r r save0s        z+pyparsing_test.reset_pyparsing_context.savecCstj|jdkr tj|jd|jdt_|jdt_tj|jdx.|jdjD]\}}|rjt j nt j |qXWdt_ |jdrtj |jdn |jd t_|jd t_|jd t_|S) NrrrrrFrrrrr)rrr Zset_default_whitespace_charsrrrinlineLiteralsUsingitemsrenabledisablerZenable_packratr"r#rr)r rvaluer r r restorePs$        z.pyparsing_test.reset_pyparsing_context.restorecCst|}|jj|j|S)N)typer update)r retr r r copyqs z+pyparsing_test.reset_pyparsing_context.copycCs|jS)N)r$)r r r r __enter__vsz0pyparsing_test.reset_pyparsing_context.__enter__cGs |jdS)N)r*)r argsr r r __exit__ysz/pyparsing_test.reset_pyparsing_context.__exit__N) __name__ __module__ __qualname____doc__rr$r*r.r/r1r r r r reset_pyparsing_contexts !r6c@sJeZdZdZdddZdddZddd 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. NcCs<|dk r|j||j|d|dk r8|j||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``. N)msg)Z assertEqualas_listZas_dict)r result expected_list expected_dictr7r r r assertParseResultsEqualssz?pyparsing_test.TestParseResultsAsserts.assertParseResultsEqualsTcCs@|j|dd}|r t|jn t|j|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:r7N) parse_stringprintdumpr8r<)r expr test_stringr:r7verboser9r r r assertParseAndCheckLists  z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckListcCs@|j|dd}|r t|jn t|j|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;r7N)r=r>r?r8r<)r r@rAr;r7rBr9r r r assertParseAndCheckDicts  z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckDictc Cs|\}}|dk rddt||D}x|D]\}}} tdd| Dd} tdd| Dd} | dk r|j| | pp|dt|tr|WdQRXq*tdd| Dd} td d| Dd} | | 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)] NcSsg|]\}}||fqSr r )rZrptexpectedr r r szOpyparsing_test.TestParseResultsAsserts.assertRunTestResults..css|]}t|tr|VqdS)N) isinstancestr)rexpr r r szNpyparsing_test.TestParseResultsAsserts.assertRunTestResults..css&|]}t|trt|tr|VqdS)N)rHr+ issubclass Exception)rrJr r r rKs)expected_exceptionr7css|]}t|tr|VqdS)N)rHlist)rrJr r r rKscss|]}t|tr|VqdS)N)rHdict)rrJr r r rKs)r:r;r7zno validation for {!r}zfailed runTests)r7)NN) zipnext assertRaisesrHrMr<r>formatZ assertTrue)r Zrun_tests_reportZexpected_parse_resultsr7Zrun_test_successZrun_test_resultsZmergedrAr9rFZfail_msgrNr:r;r r r assertRunTestResultss:   z;pyparsing_test.TestParseResultsAsserts.assertRunTestResultsc cs$|j||d dVWdQRXdS)N)r7)rS)r exc_typer7r r r assertRaisesParseExceptionszApyparsing_test.TestParseResultsAsserts.assertRaisesParseException)NNN)NT)NT)NN) r2r3r4r5r<rCrErUrrrWr r r r TestParseResultsAsserts|s   >rXNT|)s start_lineend_line expand_tabseol_mark mark_spaces mark_controlreturncs|r |j}dk rdkrPtjddttddtddDd d iB}d n(tjfd dttdd d gD}|j|}|dk r|dkr|dkrtjddd}|j|}n |jd|}|dkrd}|dkrt|}t |t|}t t d||}dkr|j |d|}n dd|j d|d|D}|sFd Stt|t dd|D} dd} | dkr| d j ddtt | ddDd} nd } | | d j ddt| d Dd} | d| d d} | | dj fddt||d DdS)!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 NunicodecSsi|]\}}||qSr r )rcur r r rsz4pyparsing_test.with_line_numbers..r!i$i3$i!$csi|] }|qSr r )rrc)r`r r rs  i $i#$) rhrcSsg|] }|dqS)u␊r )rliner r r rG'sz4pyparsing_test.with_line_numbers..u␊css|]}t|VqdS)N)len)rrkr r r rK,sz3pyparsing_test.with_line_numbers..ccss&|]}djdd|ddVqdS)z{}{}rirmrdN)rT)rir r r rK2srn css |]}dj|ddVqdS)z {}r N)rT)rror r r rK=srqZ 1234567890c3s"|]\}}dj||VqdS)z {:{}d}:{}{}N)rT)rrork)r^ lineno_widthr r rKGs)start) expandtabsrI maketransrQrangerO translatereplacerlminmax splitlinessplitjoin enumerate)rZr[r\r]r^r_r`ZtblZs_linesZ max_line_lenZleadZheader0Zheader1Zheader2r )r^rrr`r with_line_numberssH $        ,,z pyparsing_test.with_line_numbers)NNTrYNN) r2r3r4r5r6rX staticmethodrItypingOptionalintboolrr r r r r shq(r ) contextlibrrcorerrrrrr r r r r s