?d dZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd l mZdd l mZdd l mZGddej"ZddZy)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) py3compat)stylec$eZdZfdZdZxZS)YAPFTestcjtt| |tjs|j |_yy)N)superr__init__r PY3assertRaisesRegexpassertRaisesRegex)selfargs __class__s %sz > %szActual:zDiff:actualexpected)fromfiletofilelineterm ) r Get splitlineslenappendextenddifflib unified_difffailjoin)r expected_codecodemsglinelenlines rassertCodeEqualzYAPFTest.assertCodeEqual(s } $k 2c .)g**,%$ t9w  **Wt^ $ **Wt^ $ %  jj//#%$ t9w  **Wt^ $ **Wt^ $ %  jj jj   oo&&(   ii #-r)__name__ __module__ __qualname__rr2 __classcell__)rs@rrr!s 7  rrctj|}tj|t j |t j|tj|tj|tj||r%tj|t j"t%j&|}|D]}|j)|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. ) target_stream)rParseCodeToTreerSpliceCommentsrSpliceContinuationsr AssignSubtypesr IdentifyContainersr ComputeSplitPenaltiesrCalculateBlankLinesr DumpPyTreesysstderrr UnwrapPyTreeCalculateFormattingInformation)r.dumptreetreellinesllines rParseAndUnwraprIBs  % %d +$  &**40!!$'''-%%d+++D1 d#**=  ( ( .&+e ((*+ -r)F)__doc__r)rAunittest yapf.pytreerrrrrrr r yapf.yapflibr r r TestCaserrIrrrPsL) -',($&%(+" x   Br