tests/data/preview/remove_await_parens.py

   1 import asyncio
   2
   3 # Control example
   4 async def main():
   5     await asyncio.sleep(1)
   6
   7 # Remove brackets for short coroutine/task
   8 async def main():
   9     await (asyncio.sleep(1))
  10
  11 async def main():
  12     await (
  13         asyncio.sleep(1)
  14     )
  15
  16 async def main():
  17     await (asyncio.sleep(1)
  18     )
  19
  20 # Check comments
  21 async def main():
  22     await (  # Hello
  23         asyncio.sleep(1)
  24     )
  25
  26 async def main():
  27     await (
  28         asyncio.sleep(1)  # Hello
  29     )
  30
  31 async def main():
  32     await (
  33         asyncio.sleep(1)
  34     )  # Hello
  35
  36 # Long lines
  37 async def main():
  38     await asyncio.gather(asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1))
  39
  40 # Same as above but with magic trailing comma in function
  41 async def main():
  42     await asyncio.gather(asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1), asyncio.sleep(1),)
  43
  44 # Cr@zY Br@ck3Tz
  45 async def main():
  46     await (
  47         (((((((((((((
  48         (((        (((
  49         (((         (((
  50         (((         (((
  51         (((        (((
  52         ((black(1)))
  53         )))        )))
  54         )))         )))
  55         )))         )))
  56         )))        )))
  57         )))))))))))))
  58     )
  59
  60 # Keep brackets around non power operations and nested awaits
  61 async def main():
  62     await (set_of_tasks | other_set)
  63
  64 async def main():
  65     await (await asyncio.sleep(1))
  66
  67 # It's awaits all the way down...
  68 async def main():
  69     await (await x)
  70
  71 async def main():
  72     await (yield x)
  73
  74 async def main():
  75     await (await (asyncio.sleep(1)))
  76
  77 async def main():
  78     await (await (await (await (await (asyncio.sleep(1))))))
  79
  80 # output
  81 import asyncio
  82
  83
  84 # Control example
  85 async def main():
  86     await asyncio.sleep(1)
  87
  88
  89 # Remove brackets for short coroutine/task
  90 async def main():
  91     await asyncio.sleep(1)
  92
  93
  94 async def main():
  95     await asyncio.sleep(1)
  96
  97
  98 async def main():
  99     await asyncio.sleep(1)
 100
 101
 102 # Check comments
 103 async def main():
 104     await asyncio.sleep(1)  # Hello
 105
 106
 107 async def main():
 108     await asyncio.sleep(1)  # Hello
 109
 110
 111 async def main():
 112     await asyncio.sleep(1)  # Hello
 113
 114
 115 # Long lines
 116 async def main():
 117     await asyncio.gather(
 118         asyncio.sleep(1),
 119         asyncio.sleep(1),
 120         asyncio.sleep(1),
 121         asyncio.sleep(1),
 122         asyncio.sleep(1),
 123         asyncio.sleep(1),
 124         asyncio.sleep(1),
 125     )
 126
 127
 128 # Same as above but with magic trailing comma in function
 129 async def main():
 130     await asyncio.gather(
 131         asyncio.sleep(1),
 132         asyncio.sleep(1),
 133         asyncio.sleep(1),
 134         asyncio.sleep(1),
 135         asyncio.sleep(1),
 136         asyncio.sleep(1),
 137         asyncio.sleep(1),
 138     )
 139
 140
 141 # Cr@zY Br@ck3Tz
 142 async def main():
 143     await black(1)
 144
 145
 146 # Keep brackets around non power operations and nested awaits
 147 async def main():
 148     await (set_of_tasks | other_set)
 149
 150
 151 async def main():
 152     await (await asyncio.sleep(1))
 153
 154
 155 # It's awaits all the way down...
 156 async def main():
 157     await (await x)
 158
 159
 160 async def main():
 161     await (yield x)
 162
 163
 164 async def main():
 165     await (await asyncio.sleep(1))
 166
 167
 168 async def main():
 169     await (await (await (await (await asyncio.sleep(1)))))