Consider stars in testlist_star_expr unpacking (because they are)

Fixes #297

diff --git a/black.py b/black.py
index 2783a1084dfd8aec941d93b78e531cd3dbf55702..46c0907894d8a9b713f1899eb4dbdb818439f13b 100644 (file)
--- a/black.py
+++ b/black.py
@@ -768,6 +768,7 @@ UNPACKING_PARENTS = {
     syms.dictsetmaker,
     syms.listmaker,
     syms.testlist_gexp,
+    syms.testlist_star_expr,
 }
 TEST_DESCENDANTS = {
     syms.test,

diff --git a/tests/expression.diff b/tests/expression.diff
index 263234927f5ace245728acb7104526ec99936e77..0d2ff68e4aa36bd6ce6606eca9f6c12f533f542a 100644 (file)
--- a/tests/expression.diff
+++ b/tests/expression.diff
@@ -147,7 +147,7 @@
  slice[0:1:2]
  slice[:]
  slice[:-1]
-@@ -133,107 +156,159 @@
+@@ -133,108 +156,160 @@
  numpy[-(c + 1) :, d]
  numpy[:, l[-2]]
  numpy[:, ::-1]
@@ -183,6 +183,7 @@
  c = 1
  d = (1,) + a + (2,)
  e = (1,).count(1)
+ f = 1, *range(10)
 -what_is_up_with_those_new_coord_names = (coord_names + set(vars_to_create)) + set(vars_to_remove)
 -what_is_up_with_those_new_coord_names = (coord_names | set(vars_to_create)) - set(vars_to_remove)
 -result = session.query(models.Customer.id).filter(models.Customer.account_id == account_id, models.Customer.email == email_address).order_by(models.Customer.id.asc(),).all()

diff --git a/tests/expression.py b/tests/expression.py
index f35a6fd2da73155f310838b938086d832c01bdad..ea927a59f402589949e3dfaa416e9b7db47fbf37 100644 (file)
--- a/tests/expression.py
+++ b/tests/expression.py
@@ -152,6 +152,7 @@ b = 1,
 c = 1
 d = (1,) + a + (2,)
 e = (1,).count(1)
+f = 1, *range(10)
 what_is_up_with_those_new_coord_names = (coord_names + set(vars_to_create)) + set(vars_to_remove)
 what_is_up_with_those_new_coord_names = (coord_names | set(vars_to_create)) - set(vars_to_remove)
 result = session.query(models.Customer.id).filter(models.Customer.account_id == account_id, models.Customer.email == email_address).order_by(models.Customer.id.asc(),).all()
@@ -426,6 +427,7 @@ b = (1,)
 c = 1
 d = (1,) + a + (2,)
 e = (1,).count(1)
+f = 1, *range(10)
 what_is_up_with_those_new_coord_names = (coord_names + set(vars_to_create)) + set(
     vars_to_remove
 )