a d^ @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__c/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapftests/yapf_test_helper.pyr"szYAPFTest.__init__c Cs||krddg}td}|D].}t||krB|d|q"|d|q"|d|D].}t||kr|d|qd|d|qd|d|tj||dd d d |d |dS) NzCode format mismatch:z Expected:Z COLUMN_LIMITz!> %sz > %szActual:zDiff:actualexpected)fromfiletofileZlineterm ) r ZGet splitlineslenappendextenddifflibZ unified_diffZfailjoin)rZ expected_codecodemsgZlinelenlinerrrassertCodeEqual%s.       zYAPFTest.assertCodeEqual)__name__ __module__ __qualname__rr$ __classcell__rrrrr s r FcCszt|}t|t|t|t |t |t ||rZtj|tjdt|}|D] }|qh|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