actions/black
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix numpy-style array indexing

Browse Source 
Fixes #33
This commit is contained in:
Łukasz Langa 2018-03-15 23:39:45 -07:00
parent 907dc6c35e
commit fb84d8b2ee
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
black.py
View File

@ -960,7 +960,14 @@ def whitespace(leaf: Leaf) -> str:
elif p.type == syms.subscript:
# indexing
if not prev or prev.type == token.COLON:
if not prev:
assert p.parent is not None, "subscripts are always parented"
if p.parent.type == syms.subscriptlist:
return SPACE
return NO
elif prev.type == token.COLON:
return NO
elif p.type == syms.atom:

2
tests/expression.py
View File

@ -88,6 +88,7 @@
slice[:-1]
slice[1:]
slice[::-1]
numpy[:, 0:1]
(str or None) if (sys.version_info[0] > (3,)) else (str or bytes or None)
{'2.7': dead, '3.7': long_live or die_hard}
{'2.7', '3.6', '3.7', '3.8', '3.9', '4.0' if gilectomy else '3.10'}
@ -219,6 +220,7 @@ async def f():
slice[:-1]
slice[1:]
slice[::-1]
numpy[:, 0:1]
(str or None) if (sys.version_info[0] > (3,)) else (str or bytes or None)
{'2.7': dead, '3.7': long_live or die_hard}
{'2.7', '3.6', '3.7', '3.8', '3.9', '4.0' if gilectomy else '3.10'}