wip

davidszotten · Aug 12, 2020 · bf639cc · bf639cc
1 parent b59a524
commit bf639cc
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/src/black/__init__.py b/src/black/__init__.py
@@ -3287,8 +3287,14 @@ def do_match(self, line: Line) -> TMatchResult:
                 and LL[next_idx].type == token.RPAR
                 and not is_empty_rpar(LL[next_idx])
             ):
-                # That RPAR should NOT be followed by a '.' symbol.
-                if is_valid_index(next_idx + 1) and LL[next_idx + 1].type == token.DOT:
+                # That RPAR should NOT be followed by...
+                if is_valid_index(next_idx + 1) and (
+                    # a '.' symbol
+                    LL[next_idx + 1].type == token.DOT
+                    or
+                    # a '[' symbol
+                    LL[next_idx + 1].type == token.LSQB
+                ):
                     continue
 
                 return Ok(string_idx)

diff --git a/tests/test_black.py b/tests/test_black.py
@@ -487,6 +487,14 @@ def test_slices(self) -> None:
         black.assert_equivalent(source, actual)
         black.assert_stable(source, actual, black.FileMode())
 
+    @patch("black.dump_to_file", dump_to_stderr)
+    def test_bracketed_slicing(self) -> None:
+        source, expected = read_data("bracketed_slicing")
+        actual = fs(source)
+        self.assertFormatEqual(expected, actual)
+        black.assert_equivalent(source, actual)
+        black.assert_stable(source, actual, black.FileMode())
+
     @patch("black.dump_to_file", dump_to_stderr)
     def test_comments(self) -> None:
         source, expected = read_data("comments")