`N` dZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZGd dejZddZdS)z"Support module for tests for yapf.N)blank_line_calculator)comment_splicer)continuation_splicer)identify_container)pytree_unwrapper) pytree_utils)pytree_visitor) split_penalty)style)subtype_assignerceZdZdZdS)YAPFTestc ||krrddg}tjd}|D]F}t||kr|d|z.|d|zG|d|D]F}t||kr|d|z.|d|zG|d|t j||dd d |d |dSdS) NzCode format mismatch:z Expected: COLUMN_LIMITz!> %sz > %szActual:zDiff:actualexpected)fromfiletofilelineterm ) r Get splitlineslenappendextenddifflib unified_difffailjoin)self expected_codecodemsglinelenls :lib/python3.11/site-packages/yapftests/yapf_test_helper.pyassertCodeEqualzYAPFTest.assertCodeEqual"s } $k 2c .))g''))""! q66G   **Wq[ ! ! ! ! **Wq[ ! ! ! ! jj  ""! q66G   **Wq[ ! ! ! ! **Wq[ ! ! ! ! jj jj  oo&&((     ii #-N)__name__ __module__ __qualname__r(r)r'rr s#     r)rFctj|}tj|t j|t j|tj |tj |tj ||r tj|t jt%j|}|D]}||S)aZProduces unwrapped lines from the given code. Parses the code into a tree, performs comment splicing and runs the unwrapper. Arguments: code: code to parse as a string dumptree: if True, the parsed pytree (after comment splicing) is dumped to stderr. Useful for debugging. Returns: List of unwrapped lines. ) target_stream)rParseCodeToTreerSpliceCommentsrSpliceContinuationsr AssignSubtypesrIdentifyContainersr ComputeSplitPenaltiesrCalculateBlankLinesr DumpPyTreesysstderrr UnwrapPyTreeCalculateFormattingInformation)r#dumptreetreeuwlinesuwls r'ParseAndUnwrapr@<s  %d + +$ &&&*4000!$''''---%d++++D111 >d#*====  )$ / /' ))c&&(((( .r))F)__doc__rr8unittest yapf.yapflibrrrrrrr r r r TestCaserr@r-r)r'rEs')( ......((((((------++++++))))))%%%%%%''''''&&&&&&))))))     x    8r)