o ?n0j^ @sdZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd l mZdd l mZGd ddejZdddZdS)z"Support module for tests for yapf.N)blank_line_calculator)comment_splicer)continuation_splicer)pytree_unwrapper) pytree_utils)pytree_visitor) split_penalty)subtype_assigner)identify_container)stylecs$eZdZfddZddZZS)YAPFTestcstt|j|dS)N)superr __init__)selfargs __class__a/home/wildlama/miniconda3/envs/lam_a2e/lib/python3.10/site-packages/yapftests/yapf_test_helper.pyr"szYAPFTest.__init__c Cs||krjddg}td}|D]}t||kr!|d|q|d|q|d|D]}t||krB|d|q2|d|q2|d|tj||dd d d |d |dSdS) NzCode format mismatch:z Expected:Z COLUMN_LIMITz!> %sz > %szActual:zDiff:actualexpected)fromfiletofileZlineterm ) r ZGet splitlineslenappendextenddifflibZ unified_diffZfailjoin)rZ expected_codecodemsgZlinelenlinerrrassertCodeEqual%s0       zYAPFTest.assertCodeEqual)__name__ __module__ __qualname__rr$ __classcell__rrrrr s r FcCszt|}t|t|t|t |t |t ||r-tj|tjdt|}|D]}|q4|S)aVProduces logical 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 logical lines. )Z target_stream)rZParseCodeToTreerZSpliceCommentsrZSpliceContinuationsr ZAssignSubtypesr ZIdentifyContainersrZComputeSplitPenaltiesrZCalculateBlankLinesrZ DumpPyTreesysstderrrZ UnwrapPyTreeZCalculateFormattingInformation)r!ZdumptreetreeZllinesZllinerrrParseAndUnwrap?s         r,)F)__doc__rr)ZunittestZ yapf.pytreerrrrrrrr Z yapf.yapflibr r ZTestCaser r,rrrrs