
    `N`8                         d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm	Z	  ej
        ej        d           G d d	e	j                              Zed
k    r ej                     dS dS )z$Python 3 tests for yapf.reformatter.    N)	py3compat)reformatter)style)yapf_test_helperzRequires Python 3c                       e Zd ZdZed             Zd Zd Zd Z e	j
        ej        d          d             Zd Zd	 Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z!d Z"dS )TestsForPython3Codez3Test a few constructs that are new Python 3 syntax.c                 P    t          j        t          j                               d S )N)r   SetGlobalStyleCreatePEP8Style)clss    Blib/python3.11/site-packages/yapftests/reformatter_python3_test.py
setUpClasszTestsForPython3Code.setUpClass   s"    	.0011111    c                     t          j        d          }t          j        d          }t          j        |          }|                     |t          j        |                     d S )Nz        def x(aaaaaaaaaaaaaaa:int,bbbbbbbbbbbbbbbb:str,ccccccccccccccc:dict,eeeeeeeeeeeeee:set={1, 2, 3})->bool:
          pass
        z        def x(aaaaaaaaaaaaaaa: int,
              bbbbbbbbbbbbbbbb: str,
              ccccccccccccccc: dict,
              eeeeeeeeeeeeee: set = {1, 2, 3}) -> bool:
            pass
        textwrapdedentr   ParseAndUnwrapassertCodeEqualr   Reformatselfunformatted_codeexpected_formatted_codeuwliness       r   testTypedNamesz"TestsForPython3Code.testTypedNames#   sm     (   'o /   -.>??G0+2Fw2O2OPPPPPr   c                     t          j        d          }t          j        d          }t          j        |          }|                     |t          j        |                     d S )Nz}        def func(arg=long_function_call_that_pushes_the_line_over_eighty_characters()) -> ReturnType:
          pass
        z        def func(arg=long_function_call_that_pushes_the_line_over_eighty_characters()
                 ) -> ReturnType:
            pass
        r   r   s       r   testTypedNameWithLongNamedArgz1TestsForPython3Code.testTypedNameWithLongNamedArg2   sm     (   'o /  
 -.>??G0+2Fw2O2OPPPPPr   c                     t          j        d          }t          j        d          }t          j        |          }|                     |t          j        |                     d S )Nz9        def foo(a, *, kw):
          return a+kw
        z=        def foo(a, *, kw):
            return a + kw
        r   r   s       r   testKeywordOnlyArgSpecifierz/TestsForPython3Code.testKeywordOnlyArgSpecifier?   m     (   'o /   -.>??G0+2Fw2O2OPPPPPr   zRequires Python 3.6c                     t          j        d          }t          j        d          }t          j        |          }|                     |t          j        |                     d S )Nzd    { ** x }
    {   **{}   }
    { **{   **x },  **x }
    {'a': 1,   **kw , 'b':3,  **kw2   }
    zN    {**x}
    {**{}}
    {**{**x}, **x}
    {'a': 1, **kw, 'b': 3, **kw2}
    r   r   s       r   testPEP448ParameterExpansionz0TestsForPython3Code.testPEP448ParameterExpansionK   sm     ( 	 	 'o / 	 	 -.>??G0+2Fw2O2OPPPPPr   c                     t          j        d          }t          j        d          }t          j        |          }|                     |t          j        |                     d S )NzI        def foo(a: list, b: "bar") -> dict:
          return a+b
        zM        def foo(a: list, b: "bar") -> dict:
            return a + b
        r   r   s       r   testAnnotationsz#TestsForPython3Code.testAnnotations\   r!   r   c                     d}d}t          j        |          }|                     |t          j        |                     d S )Nz!methods.exec( sys.modules[name])
z methods.exec(sys.modules[name])
r   r   r   r   r   r   s       r   testExecAsNonKeywordz(TestsForPython3Code.testExecAsNonKeywordh   sG    ;A-.>??G0+2Fw2O2OPPPPPr   c                     t           j        d         dk     rd S t          j        d          }t	          j        |          }|                     |t          j        |d                     d S )N      a@          import asyncio
        import time


        @print_args
        async def slow_operation():
            await asyncio.sleep(1)
            # print("Slow operation {} complete".format(n))


        async def main():
            start = time.time()
            if (await get_html()):
                pass
        F)verify	sysversion_infor   r   r   r   r   r   r   r   coder   s      r   testAsyncFunctionsz&TestsForPython3Code.testAsyncFunctionsn   sn    
Qf?   D  -d33G{3GEJJJKKKKKr   c                    t          j        d          }t          j        d          }	 t          j        t          j        d                     t          j        |          }|                     |t          j	        |                     t          j        t          j
                               d S # t          j        t          j
                               w xY w)Nz        a**b
        z        a ** b
        z:{based_on_style: pep8, SPACES_AROUND_POWER_OPERATOR: True}r   r   r   r
   CreateStyleFromConfigr   r   r   r   r   r   r   s       r   testNoSpacesAroundPowerOperatorz3TestsForPython3Code.testNoSpacesAroundPowerOperator   s     (   'o /  	4

%JL LM M M !/0@AAg
2&/88: : : 502233333e50223333   A"B3 3'Cc                    t          j        d          }t          j        d          }	 t          j        t          j        d                     t          j        |          }|                     |t          j	        |                     t          j        t          j
                               d S # t          j        t          j
                               w xY w)Nz        f(a=5)
        z        f(a = 5)
        zC{based_on_style: pep8, SPACES_AROUND_DEFAULT_OR_NAMED_ASSIGN: True}r4   r   s       r   $testSpacesAroundDefaultOrNamedAssignz8TestsForPython3Code.testSpacesAroundDefaultOrNamedAssign   s     (   'o /  
4

%=> >? ? ?
 !/0@AAg
2&/88: : : 502233333e50223333r7   c                     t          j        d          }t          j        d          }t          j        |          }|                     |t          j        |                     d S )Nzh        def foo(x: int=42):
            pass


        def foo2(x: 'int' =42):
            pass
        zk        def foo(x: int = 42):
            pass


        def foo2(x: 'int' = 42):
            pass
        r   r   s       r   testTypeHintz TestsForPython3Code.testTypeHint   sm     (   'o /   -.>??G0+2Fw2O2OPPPPPr   c                     t          j        d          }t          j        d          }t          j        |          }|                     |t          j        |                     d S )Nz        a=b@c
        z        a = b @ c
        r   r   s       r   testMatrixMultiplicationz,TestsForPython3Code.testMatrixMultiplication   sm     (   'o /   -.>??G0+2Fw2O2OPPPPPr   c                     d}t          j        |          }|                     |t          j        |                     d S )NzNone.__ne__()
r'   r0   s      r   testNoneKeywordz#TestsForPython3Code.testNoneKeyword   sB    D -d33G{3G<<=====r   c                     t           j        d         dk     rd S t          j        d          }t          j        d          }t	          j        |          }|                     |t          j        |                     d S )Nr*   r+   z        import asyncio

        # Comment
        async def bar():
            pass

        async def foo():
            pass
        z        import asyncio


        # Comment
        async def bar():
            pass


        async def foo():
            pass
        r-   r   s       r   testAsyncWithPrecedingCommentz1TestsForPython3Code.testAsyncWithPrecedingComment   s    
Qf 	( 	 	 'o /   -.>??G0+2Fw2O2OPPPPPr   c                     t           j        d         dk     rd S t          j        d          }t	          j        |          }|                     |t          j        |                     d S )Nr*   r+   zW        async def outer():
            async def inner():
                pass
        r-   r0   s      r   testAsyncFunctionsNestedz,TestsForPython3Code.testAsyncFunctionsNested   si    
Qf?   D
 -d33G{3G<<=====r   c                     t           j        d         dk     rd S t          j        d          }t          j        d          }t	          j        |          }|                     |t          j        |                     d S )Nr*   r+   z        def _ReduceAbstractContainers(
            self, *args: Optional[automation_converter.PyiCollectionAbc]) -> List[
                automation_converter.PyiCollectionAbc]:
            pass
        z        def _ReduceAbstractContainers(
            self, *args: Optional[automation_converter.PyiCollectionAbc]
        ) -> List[automation_converter.PyiCollectionAbc]:
            pass
        r-   r   s       r   testKeepTypesIntactz'TestsForPython3Code.testKeepTypesIntact       
Qf (   'o /   -.>??G0+2Fw2O2OPPPPPr   c                     t           j        d         dk     rd S t          j        d          }t          j        d          }t	          j        |          }|                     |t          j        |                     d S )Nr*   r+   z        async def start_websocket():
            async with session.ws_connect(
                r"ws://a_really_long_long_long_long_long_long_url") as ws:
                pass
        z        async def start_websocket():
            async with session.ws_connect(
                    r"ws://a_really_long_long_long_long_long_long_url") as ws:
                pass
        r-   r   s       r   testContinuationIndentWithAsyncz3TestsForPython3Code.testContinuationIndentWithAsync	  rF   r   c                    t           j        d         dk     rd S d}d}	 t          j        t          j        d                     t          j        |          }|                     |t          j	        |                     t          j        t          j
                               d S # t          j        t          j
                               w xY w)Nr*   r+   a  async def open_file(file, mode='r', buffering=-1, encoding=None, errors=None, newline=None, closefd=True, opener=None):
    pass

async def run_sync_in_worker_thread(sync_fn, *args, cancellable=False, limiter=None):
    pass

def open_file(file, mode='r', buffering=-1, encoding=None, errors=None, newline=None, closefd=True, opener=None):
    pass

def run_sync_in_worker_thread(sync_fn, *args, cancellable=False, limiter=None):
    pass
a  async def open_file(
    file,
    mode='r',
    buffering=-1,
    encoding=None,
    errors=None,
    newline=None,
    closefd=True,
    opener=None
):
    pass


async def run_sync_in_worker_thread(
    sync_fn, *args, cancellable=False, limiter=None
):
    pass


def open_file(
    file,
    mode='r',
    buffering=-1,
    encoding=None,
    errors=None,
    newline=None,
    closefd=True,
    opener=None
):
    pass


def run_sync_in_worker_thread(sync_fn, *args, cancellable=False, limiter=None):
    pass
z{based_on_style: pep8, dedent_closing_brackets: true, coalesce_brackets: false, space_between_ending_comma_and_closing_bracket: false, split_arguments_when_comma_terminated: true, split_before_first_argument: true})r.   r/   r   r
   r5   r   r   r   r   r   r   r   s       r   testSplittingArgumentsz*TestsForPython3Code.testSplittingArguments  s    
Qf#J4

%34 45 5 5 !/0@AAg
2&/88: : : 502233333e50223333s   A"B' ''Cc                     t           j        d         dk     rd S d}d}t          j        |          }|                     |t          j        |                     d S )Nr*   r+   zclass Foo:
    def foo(self):
        foofoofoofoofoofoofoofoo('foofoofoofoofoo', {

            'foo': 'foo',

            **foofoofoo
        })
zclass Foo:
    def foo(self):
        foofoofoofoofoofoofoofoo('foofoofoofoofoo', {
            'foo': 'foo',
            **foofoofoo
        })
r.   r/   r   r   r   r   r   r   s       r   testDictUnpackingz%TestsForPython3Code.testDictUnpackinga  se    
Qf	 -.>??G0+2Fw2O2OPPPPPr   c                     t           j        d         dk     rd S d}t          j        |          }|                     |t          j        |                     d S )Nr*      z,# yapf: disable
(f'''
  ''')
# yapf: enable
rL   r0   s      r   testMultilineFormatStringz-TestsForPython3Code.testMultilineFormatStringy  sZ    
QfD -d33G{3G<<=====r   c                     t           j        d         dk     rd S d}t          j        |          }|                     |t          j        |                     d S )Nr*   rO   z[def dirichlet(x12345678901234567890123456789012345678901234567890=...) -> None:
    return
rL   r0   s      r   testEllipsesz TestsForPython3Code.testEllipses  sZ    
QfD
 -d33G{3G<<=====r   c                     d}t          j        |          }|                     |t          j        |                     d S )Nzdef _GenerateStatsEntries(
    process_id: Text,
    timestamp: Optional[ffffffff.FFFFFFFFFFF] = None
) -> Sequence[ssssssssssss.SSSSSSSSSSSSSSS]:
    pass
r'   r0   s      r   testFunctionTypedReturnNextLinez3TestsForPython3Code.testFunctionTypedReturnNextLine  sB    D -d33G{3G<<=====r   c                     d}t          j        |          }|                     |t          j        |                     d S )Nzudef rrrrrrrrrrrrrrrrrrrrrr(
        ccccccccccccccccccccccc: Tuple[Text, Text]) -> List[Tuple[Text, Text]]:
    pass
r'   r0   s      r   testFunctionTypedReturnSameLinez3TestsForPython3Code.testFunctionTypedReturnSameLine  sB    D
 -d33G{3G<<=====r   c                     t           j        d         dk     rd S t          j        d          }t	          j        |          }|                     |t          j        |                     d S )Nr*   r+   z    async def fn():
        async for message in websocket:
            for i in range(10):
                pass
            else:
                pass
        else:
            pass
    r-   r0   s      r   %testAsyncForElseNotIndentedInsideBodyz9TestsForPython3Code.testAsyncForElseNotIndentedInsideBody  si    
Qf? 	 		 		D -d33G{3G<<=====r   c                     t           j        d         dk     rd S t          j        d          }t	          j        |          }|                     |t          j        |                     d S )Nr*   r+   zd    async def fn():
        for i in range(10):
            pass
        else:
            pass
    r-   r0   s      r   &testForElseInAsyncNotMixedWithAsyncForz:TestsForPython3Code.testForElseInAsyncNotMixedWithAsyncFor  si    
Qf?  	 	D -d33G{3G<<=====r   c                     t          j        d          }t          j        d          }t          j        |          }|                     |t          j        |                     d S )Nz        def raw_message(  # pylint: disable=too-many-arguments
                    self, text, user_id=1000, chat_type='private', forward_date=None, forward_from=None):
                pass
        a          def raw_message(  # pylint: disable=too-many-arguments
                self,
                text,
                user_id=1000,
                chat_type='private',
                forward_date=None,
                forward_from=None):
            pass
        r   r   s       r   %testParameterListIndentationConflictsz9TestsForPython3Code.testParameterListIndentationConflicts  sm     (  
 'o 	/ 	 	 -.>??G0+2Fw2O2OPPPPPr   N)#__name__
__module____qualname____doc__classmethodr   r   r   r    unittest
skipUnlessr   PY36r#   r%   r(   r2   r6   r9   r;   r=   r?   rA   rC   rE   rH   rJ   rM   rP   rR   rT   rV   rX   rZ   r\    r   r   r   r      s       ;;2 2 ;2Q Q QQ Q Q
Q 
Q 
Q 8y~'<==Q Q >=Q 
Q 
Q 
QQ Q QL L L,4 4 4&4 4 4(Q Q Q(Q Q Q> > >Q Q Q8	> 	> 	>Q Q Q$Q Q Q$D4 D4 D4LQ Q Q0> > >	> 	> 	>	> 	> 	>> > >> > > > > >Q Q Q Q Qr   r   __main__)r`   r.   r   rb   yapf.yapflibr   r   r   	yapftestsr   rc   PY3YAPFTestr   r]   mainre   r   r   <module>rl      s    + * 



   " " " " " " $ $ $ $ $ $       & & & & & & Y]$788wQ wQ wQ wQ wQ*3 wQ wQ 98wQt z(-///// r   