a
    Ų
ždÆy  ć                   @   s   d Z ddlZddlZddlmZ ddlmZ ddlmZ G dd dejZ	G dd	 d	ejZ
G d
d dejZedkr~e ”  dS )z PEP8 tests for yapf.reformatter.é    N)Śreformatter)Śstyle)Śyapf_test_helperc                   @   s   e Zd Zedd Zdd Zdd Zdd Zd	d
 Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Z d;d< Z!d=S )>ŚTestsForPEP8Stylec                 C   s   t  t  ” ” d S )N)r   ŚSetGlobalStyleŚCreatePEP8Style)Ścls© r	   śh/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapftests/reformatter_pep8_test.pyŚ
setUpClass   s    zTestsForPEP8Style.setUpClassc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nz'        if a+b:
          pass
        z+        if a + b:
            pass
        ©ŚtextwrapŚdedentr   ŚParseAndUnwrapŚassertCodeEqualr   ŚReformat©ŚselfŚunformatted_codeŚexpected_formatted_codeŚllinesr	   r	   r
   ŚtestIndent4   s    


zTestsForPEP8Style.testIndent4c                 C   s*   t  d”}t |”}|  |t |”” d S )NzP        if True: a = 42
        elif False: b = 42
        else: c = 42
        r   ©r   Ścoder   r	   r	   r
   ŚtestSingleLineIfStatements+   s    

z,TestsForPEP8Style.testSingleLineIfStatementsc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )NzA        class Foo:
          def joe():
            pass
        zH        class Foo:

            def joe():
                pass
        r   r   r	   r	   r
   ŚtestBlankBetweenClassAndDef4   s    


z-TestsForPEP8Style.testBlankBetweenClassAndDefc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nzņ        class TestClass:
            def __init__(self):
                self.running = False
            def run(self):
                """Override in subclass"""
            def is_running(self):
                return self.running
        zõ        class TestClass:

            def __init__(self):
                self.running = False

            def run(self):
                """Override in subclass"""

            def is_running(self):
                return self.running
        r   r   r	   r	   r
   ŚtestBlankBetweenDefsInClassC   s    
	

z-TestsForPEP8Style.testBlankBetweenDefsInClassc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nz1        if a+b: # comment
          pass
        z6        if a + b:  # comment
            pass
        r   r   r	   r	   r
   Ś$testSingleWhiteBeforeTrailingComment\   s    


z6TestsForPEP8Style.testSingleWhiteBeforeTrailingCommentc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nz/        a = (
            1,
        )
        z        a = (1, )
        r   r   r	   r	   r
   Ś.testSpaceBetweenEndingCommandAndClosingBracketh   s    


z@TestsForPEP8Style.testSpaceBetweenEndingCommandAndClosingBracketc                 C   s*   t  d”}t |”}|  |t |”” d S )Nz²        class eld(d):
            if str(geom.geom_type).upper(
            ) != self.geom_type and not self.geom_type == 'GEOMETRY':
                ror(code='om_type')
        r   r   r	   r	   r
   ŚtestContinuedNonOutdentedLinet   s    

z/TestsForPEP8Style.testContinuedNonOutdentedLinec                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Na           def f():
            if True:
                zzzzz = '%s-%s' % (xxxxxxxxxxxxxxxxxxxxxxxxxx + 1, xxxxxxxxxxxxxxxxx.yyy + 1)
                zzzzz = '%s-%s'.ww(xxxxxxxxxxxxxxxxxxxxxxxxxx + 1, xxxxxxxxxxxxxxxxx.yyy + 1)
                zzzzz = '%s-%s' % (xxxxxxxxxxxxxxxxxxxxxxx + 1, xxxxxxxxxxxxxxxxxxxxx + 1)
                zzzzz = '%s-%s'.ww(xxxxxxxxxxxxxxxxxxxxxxx + 1, xxxxxxxxxxxxxxxxxxxxx + 1)
        a,          def f():
            if True:
                zzzzz = '%s-%s' % (xxxxxxxxxxxxxxxxxxxxxxxxxx + 1,
                                   xxxxxxxxxxxxxxxxx.yyy + 1)
                zzzzz = '%s-%s'.ww(xxxxxxxxxxxxxxxxxxxxxxxxxx + 1,
                                   xxxxxxxxxxxxxxxxx.yyy + 1)
                zzzzz = '%s-%s' % (xxxxxxxxxxxxxxxxxxxxxxx + 1,
                                   xxxxxxxxxxxxxxxxxxxxx + 1)
                zzzzz = '%s-%s'.ww(xxxxxxxxxxxxxxxxxxxxxxx + 1,
                                   xxxxxxxxxxxxxxxxxxxxx + 1)
        r   r   r	   r	   r
   ŚtestWrappingPercentExpressions~   s    


z0TestsForPEP8Style.testWrappingPercentExpressionsc                 C   sd   t  d”}t  d”}t |”}|  |t |”” t  d”}t  d”}t |”}|  |t |”” d S )Nz        TEST_LIST = ('foo', 'bar',  # first comment
                     'baz'  # second comment
                    )
        z        TEST_LIST = (
            'foo',
            'bar',  # first comment
            'baz'  # second comment
        )
        z’        def f():

          def g():
            while (xxxxxxxxxxxxxxxxxxxx(yyyyyyyyyyyyy[zzzzz]) == 'aaaaaaaaaaa' and
                   xxxxxxxxxxxxxxxxxxxx(yyyyyyyyyyyyy[zzzzz].aaaaaaaa[0]) == 'bbbbbbb'
                  ):
              pass
        a          def f():

            def g():
                while (xxxxxxxxxxxxxxxxxxxx(yyyyyyyyyyyyy[zzzzz]) == 'aaaaaaaaaaa'
                       and xxxxxxxxxxxxxxxxxxxx(
                           yyyyyyyyyyyyy[zzzzz].aaaaaaaa[0]) == 'bbbbbbb'):
                    pass
        r   r   r	   r	   r
   Ś,testAlignClosingBracketWithVisualIndentation   s    



	
	
z>TestsForPEP8Style.testAlignClosingBracketWithVisualIndentationc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nzq        if True:
          runtime_mins = (program_end_time - program_start_time).total_seconds() / 60.0
        z        if True:
            runtime_mins = (program_end_time -
                            program_start_time).total_seconds() / 60.0
        r   r   r	   r	   r
   ŚtestIndentSizeChanging»   s    


z(TestsForPEP8Style.testIndentSizeChangingc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nae          if (aaaaaaaaaaaaaa + bbbbbbbbbbbbbbbb == ccccccccccccccccc and xxxxxxxxxxxxx or yyyyyyyyyyyyyyyyy):
            pass
        elif (xxxxxxxxxxxxxxx(aaaaaaaaaaa, bbbbbbbbbbbbbb, cccccccccccc, dddddddddd=None)):
            pass


        def h():
            if (xxxxxxxxxxxx.yyyyyyyy(zzzzzzzzzzzzz[0]) == 'aaaaaaaaaaa' and xxxxxxxxxxxx.yyyyyyyy(zzzzzzzzzzzzz[0].mmmmmmmm[0]) == 'bbbbbbb'):
                pass

            for connection in itertools.chain(branch.contact, branch.address, morestuff.andmore.andmore.andmore.andmore.andmore.andmore.andmore):
                dosomething(connection)
        a          if (aaaaaaaaaaaaaa + bbbbbbbbbbbbbbbb == ccccccccccccccccc and xxxxxxxxxxxxx
                or yyyyyyyyyyyyyyyyy):
            pass
        elif (xxxxxxxxxxxxxxx(aaaaaaaaaaa,
                              bbbbbbbbbbbbbb,
                              cccccccccccc,
                              dddddddddd=None)):
            pass


        def h():
            if (xxxxxxxxxxxx.yyyyyyyy(zzzzzzzzzzzzz[0]) == 'aaaaaaaaaaa' and
                    xxxxxxxxxxxx.yyyyyyyy(zzzzzzzzzzzzz[0].mmmmmmmm[0]) == 'bbbbbbb'):
                pass

            for connection in itertools.chain(
                    branch.contact, branch.address,
                    morestuff.andmore.andmore.andmore.andmore.andmore.andmore.andmore):
                dosomething(connection)
        r   r   r	   r	   r
   ŚtestHangingIndentCollisionČ   s    


z,TestsForPEP8Style.testHangingIndentCollisionc              
   C   sh   zRt  t  d”” t d”}t d”}t |”}|  |t 	|”” W t  t  
” ” nt  t  
” ” 0 d S )Nz;{based_on_style: pep8, split_before_logical_operator: True}aG            def foo():
              return bool(update.message.new_chat_member or update.message.left_chat_member or
                          update.message.new_chat_title or update.message.new_chat_photo or
                          update.message.delete_chat_photo or update.message.group_chat_created or
                          update.message.supergroup_chat_created or update.message.channel_chat_created
                          or update.message.migrate_to_chat_id or update.message.migrate_from_chat_id or
                          update.message.pinned_message)
          ah            def foo():
              return bool(
                  update.message.new_chat_member or update.message.left_chat_member
                  or update.message.new_chat_title or update.message.new_chat_photo
                  or update.message.delete_chat_photo
                  or update.message.group_chat_created
                  or update.message.supergroup_chat_created
                  or update.message.channel_chat_created
                  or update.message.migrate_to_chat_id
                  or update.message.migrate_from_chat_id
                  or update.message.pinned_message)
          ©r   r   ŚCreateStyleFromConfigr   r   r   r   r   r   r   r   r   r	   r	   r
   Ś"testSplittingBeforeLogicalOperatorļ   s    ’’
	

’z4TestsForPEP8Style.testSplittingBeforeLogicalOperatorc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nz        if True:
            if True:
                keys.append(aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)  # may be unassigned.
        z        if True:
            if True:
                keys.append(
                    aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)  # may be unassigned.
        r   r   r	   r	   r
   Ś#testContiguousListEndingWithComment  s    


z5TestsForPEP8Style.testContiguousListEndingWithCommentc              
   C   sh   zRt  t  d”” t d”}t d”}t |”}|  |t 	|”” W t  t  
” ” nt  t  
” ” 0 d S )Nz9{based_on_style: pep8, split_before_first_argument: True}z²          a_very_long_function_name(long_argument_name_1=1, long_argument_name_2=2,
                                    long_argument_name_3=3, long_argument_name_4=4)
          zĒ          a_very_long_function_name(
              long_argument_name_1=1,
              long_argument_name_2=2,
              long_argument_name_3=3,
              long_argument_name_4=4)
          r$   r   r	   r	   r
   Ś testSplittingBeforeFirstArgument  s    ’’


’z2TestsForPEP8Style.testSplittingBeforeFirstArgumentc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nzt        def foo():
            df = df[(df['campaign_status'] == 'LIVE') & (df['action_status'] == 'LIVE')]
        z        def foo():
            df = df[(df['campaign_status'] == 'LIVE')
                    & (df['action_status'] == 'LIVE')]
        r   r   r	   r	   r
   Ś(testSplittingExpressionsInsideSubscripts5  s    


z:TestsForPEP8Style.testSplittingExpressionsInsideSubscriptsc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )NzĢ        DJANGO_TEMPLATES_OPTIONS = {"context_processors": []}
        DJANGO_TEMPLATES_OPTIONS = {"context_processors": [],}
        x = ["context_processors"]
        x = ["context_processors",]
        zų        DJANGO_TEMPLATES_OPTIONS = {"context_processors": []}
        DJANGO_TEMPLATES_OPTIONS = {
            "context_processors": [],
        }
        x = ["context_processors"]
        x = [
            "context_processors",
        ]
        r   r   r	   r	   r
   Ś/testSplitListsAndDictSetMakersIfCommaTerminatedB  s    



zATestsForPEP8Style.testSplitListsAndDictSetMakersIfCommaTerminatedc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nz        class a():

            def a(): return a(
             aaaaaaaaaa=aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)
        z¶        class a():

            def a():
                return a(
                    aaaaaaaaaa=aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
                )
        r   r   r	   r	   r
   ŚtestSplitAroundNamedAssignsV  s    


z-TestsForPEP8Style.testSplitAroundNamedAssignsc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )NzX        if not -3 < x < 3:
          pass
        if -3 < x < 3:
          pass
        z\        if not -3 < x < 3:
            pass
        if -3 < x < 3:
            pass
        r   r   r	   r	   r
   ŚtestUnaryOperatorh  s    


z#TestsForPEP8Style.testUnaryOperatorc              
   C   sų   zāt  t  d”” t d”}t d”}t |”}|  |t 	|”” t d”}t d”}t |”}|  |t 	|”” t d”}t d”}t |”}|  |t 	|”” t d”}t d	”}t |”}|  |t 	|”” W t  t  
” ” nt  t  
” ” 0 d S )
NzĮ{based_on_style: pep8, allow_split_before_dict_value: false, coalesce_brackets: true, dedent_closing_brackets: true, each_dict_entry_on_separate_line: true, split_before_logical_operator: true}zž          some_dict = {
              'title': _("I am example data"),
              'description': _("Lorem ipsum dolor met sit amet elit, si vis pacem para bellum "
                               "elites nihi very long string."),
          }
          a            some_dict = {
              'title': _("I am example data"),
              'description': _(
                  "Lorem ipsum dolor met sit amet elit, si vis pacem para bellum "
                  "elites nihi very long string."
              ),
          }
          zz          X = {'a': 1, 'b': 2, 'key': this_is_a_function_call_that_goes_over_the_column_limit_im_pretty_sure()}
          z°          X = {
              'a': 1,
              'b': 2,
              'key': this_is_a_function_call_that_goes_over_the_column_limit_im_pretty_sure()
          }
          zź          attrs = {
              'category': category,
              'role': forms.ModelChoiceField(label=_("Role"), required=False, queryset=category_roles, initial=selected_role, empty_label=_("No access"),),
          }
          aT            attrs = {
              'category': category,
              'role': forms.ModelChoiceField(
                  label=_("Role"),
                  required=False,
                  queryset=category_roles,
                  initial=selected_role,
                  empty_label=_("No access"),
              ),
          }
          zė          css_class = forms.CharField(
              label=_("CSS class"),
              required=False,
              help_text=_("Optional CSS class used to customize this category appearance from templates."),
          )
          a            css_class = forms.CharField(
              label=_("CSS class"),
              required=False,
              help_text=_(
                  "Optional CSS class used to customize this category appearance from templates."
              ),
          )
          r$   r   r	   r	   r
   ŚtestNoSplitBeforeDictValuex  s:    ’

	
’


’


’

	
’z,TestsForPEP8Style.testNoSplitBeforeDictValuec                 C   s(   d}d}t  |”}|  |t |”” d S )Na   def _():
    include_values = np.where(
                (cdffile['Quality_Flag'][:] >= 5) & (
                cdffile['Day_Night_Flag'][:] == 1) & (
                cdffile['Longitude'][:] >= select_lon - radius) & (
                cdffile['Longitude'][:] <= select_lon + radius) & (
                cdffile['Latitude'][:] >= select_lat - radius) & (
                cdffile['Latitude'][:] <= select_lat + radius))
ac  def _():
    include_values = np.where(
        (cdffile['Quality_Flag'][:] >= 5) & (cdffile['Day_Night_Flag'][:] == 1)
        & (cdffile['Longitude'][:] >= select_lon - radius)
        & (cdffile['Longitude'][:] <= select_lon + radius)
        & (cdffile['Latitude'][:] >= select_lat - radius)
        & (cdffile['Latitude'][:] <= select_lat + radius))
©r   r   ŚassertEqualr   r   ©r   r   Zexpected_coder   r	   r	   r
   ŚtestBitwiseOperandSplittingŠ  s    
	
z-TestsForPEP8Style.testBitwiseOperandSplittingc                 C   s(   d}d}t  |”}|  |t |”” d S )Nz¢class Demo:
    """
    Demo docs
    """
    def foo(self):
        """
        foo docs
        """
    def bar(self):
        """
        bar docs
        """
z¤class Demo:
    """
    Demo docs
    """

    def foo(self):
        """
        foo docs
        """

    def bar(self):
        """
        bar docs
        """
r.   r0   r	   r	   r
   Ś(testNoBlankLinesOnlyForFirstNestedObjectē  s    
z:TestsForPEP8Style.testNoBlankLinesOnlyForFirstNestedObjectc              
   C   s\   zFt  t  d”” d}d}t |”}|  |t |”” W t  t  ” ” nt  t  ” ” 0 d S )Nz>{based_on_style: pep8, split_before_arithmetic_operator: true}z_def _():
    raise ValueError('This is a long message that ends with an argument: ' + str(42))
ztdef _():
    raise ValueError('This is a long message that ends with an argument: '
                     + str(42))
©	r   r   r%   r   r   r   r   r   r   r   r	   r	   r
   Ś"testSplitBeforeArithmeticOperators	  s    ’’
’z4TestsForPEP8Style.testSplitBeforeArithmeticOperatorsc                 C   s(   d}d}t  |”}|  |t |”” d S )Nzfoo([(1,1), (1,1), (1,1), (1,1), (1,1), (1,1), (1,1),
     (1,1), (1,1), (1,1), (1,1), (1,1), (1,1), (1,1),
     (1,10), (1,11), (1, 10), (1,11), (10,11)])
z®foo([(1, 1), (1, 1), (1, 1), (1, 1), (1, 1), (1, 1), (1, 1), (1, 1), (1, 1),
     (1, 1), (1, 1), (1, 1), (1, 1), (1, 1), (1, 10), (1, 11), (1, 10),
     (1, 11), (10, 11)])
)r   r   r   r   r   r0   r	   r	   r
   ŚtestListSplitting  s    
z#TestsForPEP8Style.testListSplittingc              
   C   s\   zFt  t  d”” d}d}t |”}|  |t |”” W t  t  ” ” nt  t  ” ” 0 d S )NzD{based_on_style: pep8, blank_line_before_nested_class_or_def: false}zõdef normal_function():
    """Return the nested function."""

    def nested_function():
        """Do nothing just nest within."""

        @nested(klass)
        class nested_class():
            pass

        pass

    return nested_function
zódef normal_function():
    """Return the nested function."""
    def nested_function():
        """Do nothing just nest within."""
        @nested(klass)
        class nested_class():
            pass

        pass

    return nested_function
r3   r   r	   r	   r
   Ś&testNoBlankLineBeforeNestedFuncOrClass,  s    ’’
’z8TestsForPEP8Style.testNoBlankLineBeforeNestedFuncOrClassc                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )Nav  class _():

    def __init__(self, title: Optional[str], diffs: Collection[BinaryDiff] = (), charset: Union[Type[AsciiCharset], Type[LineCharset]] = AsciiCharset, preprocess: Callable[[str], str] = identity,
            # TODO(somebody): Make this a Literal type.
            justify: str = 'rjust'):
        self._cs = charset
        self._preprocess = preprocess
        aĪ  class _():

    def __init__(
            self,
            title: Optional[str],
            diffs: Collection[BinaryDiff] = (),
            charset: Union[Type[AsciiCharset],
                           Type[LineCharset]] = AsciiCharset,
            preprocess: Callable[[str], str] = identity,
            # TODO(somebody): Make this a Literal type.
            justify: str = 'rjust'):
        self._cs = charset
        self._preprocess = preprocess
        r   r   r	   r	   r
   Ś"testParamListIndentationCollision1U  s    
	

z4TestsForPEP8Style.testParamListIndentationCollision1c                 C   s*   t  d”}t |”}|  |t |”” d S )Nz        def simple_pass_function_with_an_extremely_long_name_and_some_arguments(
                argument0, argument1):
            pass
        r   r   r	   r	   r
   Ś"testParamListIndentationCollision2q  s    

z4TestsForPEP8Style.testParamListIndentationCollision2c                 C   s*   t  d”}t |”}|  |t |”” d S )Nzø        def func1(
            arg1,
            arg2,
        ) -> None:
            pass


        def func2(
            arg1,
            arg2,
        ):
            pass
        r   r   r	   r	   r
   Ś"testParamListIndentationCollision3z  s    

z4TestsForPEP8Style.testParamListIndentationCollision3c                 C   s4   t  d”}t  d”}t |”}|  |t |”” d S )NzŠ        _ = (klsdfjdklsfjksdlfjdklsfjdslkfjsdkl is not ksldfjsdklfjdklsfjdklsfjdklsfjdsklfjdklsfj)
        _ = (klsdfjdklsfjksdlfjdklsfjdslkfjsdkl not in {ksldfjsdklfjdklsfjdklsfjdklsfjdsklfjdklsfj})
        zź        _ = (klsdfjdklsfjksdlfjdklsfjdslkfjsdkl
             is not ksldfjsdklfjdklsfjdklsfjdklsfjdsklfjdklsfj)
        _ = (klsdfjdklsfjksdlfjdklsfjdslkfjsdkl
             not in {ksldfjsdklfjdklsfjdklsfjdklsfjdsklfjdklsfj})
        r   r   r	   r	   r
   ŚtestTwoWordComparisonOperators  s    


z0TestsForPEP8Style.testTwoWordComparisonOperatorsc                 C   sX   t  d”}t  d”}t |”}t |”}|  ||” t |”}t |”}|  ||” d S )Nz°        def _():
            url = "http://{0}/axis-cgi/admin/param.cgi?{1}".format(
                value, urllib.urlencode({'action': 'update', 'parameter': value}))
        zź        def _():
            url = "http://{0}/axis-cgi/admin/param.cgi?{1}".format(
                value, urllib.urlencode({
                    'action': 'update',
                    'parameter': value
                }))
        )r   r   r   r   r   r   r   )r   r   r   r   Zreformatted_coder	   r	   r
   Ś%testStableInlinedDictionaryFormatting  s    

	



z7TestsForPEP8Style.testStableInlinedDictionaryFormattingN)"Ś__name__Ś
__module__Ś__qualname__Śclassmethodr   r   r   r   r   r   r   r   r    r!   r"   r#   r&   r'   r(   r)   r*   r+   r,   r-   r1   r2   r4   r5   r6   r7   r8   r9   r:   r;   r	   r	   r	   r
   r      s>   
	
%'!X")	r   c                   @   s*   e Zd ZdZe d”Zdd Zdd ZdS )ŚTestsForSpacesInsideBracketsz,Test the SPACE_INSIDE_BRACKETS style option.aq      foo()
    foo(1)
    foo(1,2)
    foo((1,))
    foo((1, 2))
    foo((1, 2,))
    foo(bar['baz'][0])
    set1 = {1, 2, 3}
    dict1 = {1: 1, foo: 2, 3: bar}
    dict2 = {
        1: 1,
        foo: 2,
        3: bar,
    }
    dict3[3][1][get_index(*args,**kwargs)]
    dict4[3][1][get_index(**kwargs)]
    x = dict5[4](foo(*args))
    a = list1[:]
    b = list2[slice_start:]
    c = list3[slice_start:slice_end]
    d = list4[slice_start:slice_end:]
    e = list5[slice_start:slice_end:slice_step]
    # Print gets special handling
    print(set2)
    compound = ((10+3)/(5-2**(6+x)))
    string_idx = "mystring"[3]
    c                 C   s<   t  t  d”” t d”}t | j”}|  |t	 
|”” d S )Nz{space_inside_brackets: True}a        foo()
      foo( 1 )
      foo( 1, 2 )
      foo( ( 1, ) )
      foo( ( 1, 2 ) )
      foo( (
          1,
          2,
      ) )
      foo( bar[ 'baz' ][ 0 ] )
      set1 = { 1, 2, 3 }
      dict1 = { 1: 1, foo: 2, 3: bar }
      dict2 = {
          1: 1,
          foo: 2,
          3: bar,
      }
      dict3[ 3 ][ 1 ][ get_index( *args, **kwargs ) ]
      dict4[ 3 ][ 1 ][ get_index( **kwargs ) ]
      x = dict5[ 4 ]( foo( *args ) )
      a = list1[ : ]
      b = list2[ slice_start: ]
      c = list3[ slice_start:slice_end ]
      d = list4[ slice_start:slice_end: ]
      e = list5[ slice_start:slice_end:slice_step ]
      # Print gets special handling
      print( set2 )
      compound = ( ( 10 + 3 ) / ( 5 - 2**( 6 + x ) ) )
      string_idx = "mystring"[ 3 ]
      ©r   r   r%   r   r   r   r   r   r   r   r   ©r   r   r   r	   r	   r
   ŚtestEnabledŃ  s    ’
 z(TestsForSpacesInsideBrackets.testEnabledc                 C   s:   t  t  ” ” t d”}t | j”}|  |t	 
|”” d S )NaĪ        foo()
      foo(1)
      foo(1, 2)
      foo((1, ))
      foo((1, 2))
      foo((
          1,
          2,
      ))
      foo(bar['baz'][0])
      set1 = {1, 2, 3}
      dict1 = {1: 1, foo: 2, 3: bar}
      dict2 = {
          1: 1,
          foo: 2,
          3: bar,
      }
      dict3[3][1][get_index(*args, **kwargs)]
      dict4[3][1][get_index(**kwargs)]
      x = dict5[4](foo(*args))
      a = list1[:]
      b = list2[slice_start:]
      c = list3[slice_start:slice_end]
      d = list4[slice_start:slice_end:]
      e = list5[slice_start:slice_end:slice_step]
      # Print gets special handling
      print(set2)
      compound = ((10 + 3) / (5 - 2**(6 + x)))
      string_idx = "mystring"[3]
      ©r   r   r   r   r   r   r   r   r   r   r   rB   r	   r	   r
   ŚtestDefaultų  s    
 z(TestsForSpacesInsideBrackets.testDefaultN)	r<   r=   r>   Ś__doc__r   r   r   rC   rE   r	   r	   r	   r
   r@   ²  s   
'r@   c                   @   s2   e Zd ZdZe d”Zdd Zdd Zdd Z	d	S )
Ś"TestsForSpacesAroundSubscriptColonz4Test the SPACES_AROUND_SUBSCRIPT_COLON style option.a%      a = list1[ : ]
    b = list2[ slice_start: ]
    c = list3[ slice_start:slice_end ]
    d = list4[ slice_start:slice_end: ]
    e = list5[ slice_start:slice_end:slice_step ]
    a1 = list1[ : ]
    b1 = list2[ 1: ]
    c1 = list3[ 1:20 ]
    d1 = list4[ 1:20: ]
    e1 = list5[ 1:20:3 ]
  c                 C   s<   t  t  d”” t d”}t | j”}|  |t	 
|”” d S )Nz%{spaces_around_subscript_colon: True}a;        a = list1[:]
      b = list2[slice_start :]
      c = list3[slice_start : slice_end]
      d = list4[slice_start : slice_end :]
      e = list5[slice_start : slice_end : slice_step]
      a1 = list1[:]
      b1 = list2[1 :]
      c1 = list3[1 : 20]
      d1 = list4[1 : 20 :]
      e1 = list5[1 : 20 : 3]
    rA   rB   r	   r	   r
   rC   .  s    ’
z.TestsForSpacesAroundSubscriptColon.testEnabledc                 C   s<   t  t  d”” t d”}t | j”}|  |t	 
|”” d S )NzC{spaces_around_subscript_colon: true, space_inside_brackets: true,}aO        a = list1[ : ]
      b = list2[ slice_start : ]
      c = list3[ slice_start : slice_end ]
      d = list4[ slice_start : slice_end : ]
      e = list5[ slice_start : slice_end : slice_step ]
      a1 = list1[ : ]
      b1 = list2[ 1 : ]
      c1 = list3[ 1 : 20 ]
      d1 = list4[ 1 : 20 : ]
      e1 = list5[ 1 : 20 : 3 ]
    rA   rB   r	   r	   r
   ŚtestWithSpaceInsideBrackets@  s    ’
z>TestsForSpacesAroundSubscriptColon.testWithSpaceInsideBracketsc                 C   s:   t  t  ” ” t d”}t | j”}|  |t	 
|”” d S )Na'        a = list1[:]
      b = list2[slice_start:]
      c = list3[slice_start:slice_end]
      d = list4[slice_start:slice_end:]
      e = list5[slice_start:slice_end:slice_step]
      a1 = list1[:]
      b1 = list2[1:]
      c1 = list3[1:20]
      d1 = list4[1:20:]
      e1 = list5[1:20:3]
    rD   rB   r	   r	   r
   rE   U  s    
z.TestsForSpacesAroundSubscriptColon.testDefaultN)
r<   r=   r>   rF   r   r   r   rC   rH   rE   r	   r	   r	   r
   rG     s
   
rG   Ś__main__)rF   r   ZunittestZyapf.yapflibr   r   Z	yapftestsr   ZYAPFTestr   r@   rG   r<   Śmainr	   r	   r	   r
   Ś<module>   s        mH