o bi\<@s^ddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z GdddZ dS))contextmanagerN) ParserElementParseExceptionKeyword__diag__ __compat__)core_builtin_exprsc@seZdZdZGdddZGdddejZe      dd dd d e d e j e d e j e de de de j e de j e de je e fde de fddZdS)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: .. testcode:: ppt = pyparsing.pyparsing_test class MyTestClass(ppt.TestParseResultsAsserts): def test_literal(self): with ppt.reset_pyparsing_context(): # test that literals used to construct # a grammar are automatically suppressed ParserElement.inline_literals_using(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_contextselfrK/home/kim/smarthome/.venv/lib/python3.10/site-packages/pyparsing/testing.py__init__>s z/pyparsing_test.reset_pyparsing_context.__init__cCstj|jd<tj|jd<tj|jd<tj|jd<tj|jd<tjr)tjj |jd<nd|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 Ws z?pyparsing_test.reset_pyparsing_context.save..rcollect_all_And_tokensr)rDEFAULT_WHITE_CHARSr rDEFAULT_KEYWORD_CHARSZ_literalStringClassr_packratEnabledZ packrat_cachesize_parse_left_recursion_enabledrZ _all_namesrrr rrrsaveAs&        z+pyparsing_test.reset_pyparsing_context.savecCstj|jdkrt|jd|jdt_|jdt_t|jd|jdD]\}}|r4t j nt j |q+dt_ |jdrKt |jdn|jd t_|jd t_tD]}|dqY|jd t_|S) NrrrrrFrrrrr)rrr Zset_default_whitespace_charsrrr Zinline_literals_usingitemsrenabledisabler!Zenable_packratr#r$r Z set_debugrr)rrvalueexprrrrrestoreas0      z.pyparsing_test.reset_pyparsing_context.restorecCst|}|j|j|Sr )typer update)rretrrrcopys z+pyparsing_test.reset_pyparsing_context.copycCs|Sr )r%r rrr __enter__sz0pyparsing_test.reset_pyparsing_context.__enter__cGs |dSr )r+)rargsrrr__exit__s z/pyparsing_test.reset_pyparsing_context.__exit__N) __name__ __module__ __qualname____doc__rr%r+r/r0r2rrrrreset_pyparsing_contexts% % r7c@sTeZdZdZ dddZ dddZ ddd Z dd 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|dur|j|||ddSdS)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)Z assertEqualas_listas_dict)rresult expected_list expected_dictr9rrrassertParseResultsEqualss z?pyparsing_test.TestParseResultsAsserts.assertParseResultsEqualsTcC@|j|dd}|rt|nt||j|||ddS)z Convenience wrapper assert to test a parser element and input string, and assert that the resulting :meth:`ParseResults.as_list` is equal to the ``expected_list``. TZ parse_all)r=r9N parse_stringprintdumpr:r?)rr* test_stringr=r9verboser<rrrassertParseAndCheckList  z>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckListcCr@)z Convenience wrapper assert to test a parser element and input string, and assert that the resulting :meth:`ParseResults.as_dict` is equal to the ``expected_dict``. TrA)r>r9NrB)rr*rFr>r9rGr<rrrassertParseAndCheckDictrIz>pyparsing_test.TestParseResultsAsserts.assertParseAndCheckDictc Cs<|\}}|dur|j||dur|ndddSddt||D}|D]k\}}} tdd| Dd} tdd| Dd} | dura|j| | pF|d t|trQ|Wdn1s[wYq#td d| Dd} td d| Dd} | | fd kr|j|| | | p|d q#td|q#|j||dur|ndddS)a2 Unit test assertion to evaluate output of :meth:`~ParserElement.run_tests`. 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 ``run_tests()`` and evaluated using :meth:`assertParseResultsEquals`. Finally, asserts that the overall `:meth:~ParserElement.run_tests` success value is ``True``. :param run_tests_report: the return value from :meth:`ParserElement.run_tests` :type run_tests_report: tuple[bool, list[tuple[str, ParseResults | Exception]]] :param expected_parse_results: (optional) :type expected_parse_results: list[tuple[str | list | dict | Exception, ...]] Nzfailed runTestsr8cSsg|] \}}g||RqSrr)rZrptexpectedrrr s zOpyparsing_test.TestParseResultsAsserts.assertRunTestResults..cs|] }t|tr|VqdSr ) isinstancestrrexprrr zNpyparsing_test.TestParseResultsAsserts.assertRunTestResults..css(|]}t|trt|tr|VqdSr )rNr, issubclass ExceptionrPrrrrRs )expected_exceptionr9csrMr )rNlistrPrrrrRrScsrMr )rNdictrPrrrrRrSNN)r=r>r9zno validation for )Z assertTruezipnext assertRaisesrNrUr?rD)rZrun_tests_reportZexpected_parse_resultsr9Zrun_test_successZrun_test_resultsZmergedrFr<rKZfail_msgrVr=r>rrrassertRunTestResultssV   z;pyparsing_test.TestParseResultsAsserts.assertRunTestResultsccs|dur-t|trt|}|j|||d }|VWddS1s&wYdS|j||d }|VWddS1sCwYdS)Nr8)rNrOreescapeassertRaisesRegexr\)rexc_typeZ expected_msgr9ctxrrrassertRaisesParseExceptions  ""zApyparsing_test.TestParseResultsAsserts.assertRaisesParseException)NNN)NTrY) r3r4r5r6r?rHrJr]rrrcrrrrTestParseResultsAssertss    IrdNT|)indentbase_1s start_lineend_line expand_tabseol_mark mark_spaces mark_controlrgrhreturncs|r|}ttrd|durZtt|}|dkr=ddttddtdd D} d | d <t| } d nt |tfd dt tddd gD} | | }|durz|dkrz|dkrttddd} | | }n| d|}|durd}|durt |}t|t |}ttd||}|dkr|t||d|} ndd|dt||d|D} | sd St t|tdd| D} dd} | dkr| |rd ndd ddtt| ddDd}nd }|rd nd| d ddt| d Dd}d}| |r"d 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: string to be printed with line and column numbers :param start_line: starting line number in s to print (default=1) :param end_line: ending line number in s to print (default=len(s)) :param expand_tabs: expand tabs to spaces, to match the pyparsing default :param eol_mark: string to mark the end of lines, helps visualize trailing spaces :param mark_spaces: special character to display in place of spaces :param mark_control: 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: string to indent with line and column numbers; if an int is passed, converted to ``" " * indent`` :param base_1: whether to label string using base 1; if False, string will be labeled based at 0 :returns: input string with leading line numbers and column number headers .. versionchanged:: 3.2.0 New ``indent`` and ``base_1`` arguments.  NunicodecSsi|]\}}||qSrr)rcurrrrGs z4pyparsing_test.with_line_numbers..r!i$i3$i!$rfcsi|]}|qSrr)rrs)ord_mark_controlrrrPs i $i#$) rxcSsg|]}|dqS)␊rrlinerrrrLcsz4pyparsing_test.with_line_numbers..rzcss|]}t|VqdSr )lenr{rrrrRksz3pyparsing_test.with_line_numbers..rccss$|] }d|ddVqdS)zc rdNrrirrrrRrs  r css"|] }d|ddVqdS)z r NrrrrrrR~s rZ 12345678900c3s2|]\}}|dd|VqdS)d:Nr)rrr|)rmrg lineno_widthrrrRs  )start) expandtabsrNinttypingcastrOrZrange maketransordrW translatereplacer} splitlinesminmaxsplitjoin enumerate)rirjrkrlrmrnrorgrhZtranstable_mapZtblZs_linesZ max_line_lenZleadZheader0Zheader1digitsZheader2r)rmrgrrwrwith_line_numberss*            $z pyparsing_test.with_line_numbers)NNTreNN)r3r4r5r6r7unittestZTestCaserd staticmethodrOrOptionalrboolUnionrrrrrr sHy    r ) contextlibrr^rrcorerrrrrrfr r rrrrs