Skip to content

Commit de3d820

Browse files
committed
Avoid needing self.loop in test
1 parent b83734c commit de3d820

File tree

1 file changed

+16
-19
lines changed

1 file changed

+16
-19
lines changed

Lib/test/test_asyncio/test_taskgroups.py

+16-19
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,6 @@ def get_error_types(eg):
2626

2727
class TestTaskGroup(unittest.IsolatedAsyncioTestCase):
2828

29-
async def asyncSetUp(self):
30-
self.loop = asyncio.get_event_loop()
31-
3229
async def test_taskgroup_01(self):
3330

3431
async def foo1():
@@ -115,7 +112,7 @@ async def runner():
115112
NUM += 10
116113

117114
with self.assertRaises(ExceptionGroup) as cm:
118-
await self.loop.create_task(runner())
115+
await asyncio.create_task(runner())
119116

120117
self.assertEqual(get_error_types(cm.exception), {ZeroDivisionError})
121118

@@ -161,7 +158,7 @@ async def runner():
161158
# The 3 foo1 sub tasks can be racy when the host is busy - if the
162159
# cancellation happens in the middle, we'll see partial sub errors here
163160
with self.assertRaises(ExceptionGroup) as cm:
164-
await self.loop.create_task(runner())
161+
await asyncio.create_task(runner())
165162

166163
self.assertEqual(get_error_types(cm.exception), {ZeroDivisionError})
167164
self.assertEqual(NUM, 0)
@@ -185,7 +182,7 @@ async def runner():
185182
for _ in range(5):
186183
g.create_task(foo())
187184

188-
r = self.loop.create_task(runner())
185+
r = asyncio.create_task(runner())
189186
await asyncio.sleep(0.1)
190187

191188
self.assertFalse(r.done())
@@ -219,7 +216,7 @@ async def runner():
219216
NUM += 10
220217
raise
221218

222-
r = self.loop.create_task(runner())
219+
r = asyncio.create_task(runner())
223220
await asyncio.sleep(0.1)
224221

225222
self.assertFalse(r.done())
@@ -245,7 +242,7 @@ async def runner():
245242
except asyncio.CancelledError:
246243
raise
247244

248-
r = self.loop.create_task(runner())
245+
r = asyncio.create_task(runner())
249246
await asyncio.sleep(0.1)
250247

251248
self.assertFalse(r.done())
@@ -329,7 +326,7 @@ async def runner():
329326
except asyncio.CancelledError:
330327
raise
331328

332-
r = self.loop.create_task(runner())
329+
r = asyncio.create_task(runner())
333330
await asyncio.sleep(0.1)
334331

335332
self.assertFalse(r.done())
@@ -356,7 +353,7 @@ async def runner():
356353
except asyncio.CancelledError:
357354
raise
358355

359-
r = self.loop.create_task(runner())
356+
r = asyncio.create_task(runner())
360357
await asyncio.sleep(0.1)
361358

362359
self.assertFalse(r.done())
@@ -377,7 +374,7 @@ async def runner():
377374
async with taskgroups.TaskGroup(name='g2') as g2:
378375
g2.create_task(crash_after(0.2))
379376

380-
r = self.loop.create_task(runner())
377+
r = asyncio.create_task(runner())
381378
with self.assertRaises(ExceptionGroup) as cm:
382379
await r
383380

@@ -396,7 +393,7 @@ async def runner():
396393
async with taskgroups.TaskGroup(name='g2') as g2:
397394
g2.create_task(crash_after(0.1))
398395

399-
r = self.loop.create_task(runner())
396+
r = asyncio.create_task(runner())
400397
with self.assertRaises(ExceptionGroup) as cm:
401398
await r
402399

@@ -419,7 +416,7 @@ async def runner():
419416
await asyncio.sleep(0.5)
420417
raise
421418

422-
r = self.loop.create_task(runner())
419+
r = asyncio.create_task(runner())
423420
await asyncio.sleep(0.1)
424421

425422
self.assertFalse(r.done())
@@ -443,10 +440,10 @@ async def nested_runner():
443440
raise
444441

445442
async def runner():
446-
t = self.loop.create_task(nested_runner())
443+
t = asyncio.create_task(nested_runner())
447444
await t
448445

449-
r = self.loop.create_task(runner())
446+
r = asyncio.create_task(runner())
450447
await asyncio.sleep(0.1)
451448

452449
self.assertFalse(r.done())
@@ -466,7 +463,7 @@ async def runner():
466463
NUM += 10
467464
raise
468465

469-
r = self.loop.create_task(runner())
466+
r = asyncio.create_task(runner())
470467
await asyncio.sleep(0.1)
471468

472469
self.assertFalse(r.done())
@@ -490,7 +487,7 @@ async def runner():
490487
# this weird case.
491488
raise MyExc
492489

493-
r = self.loop.create_task(runner())
490+
r = asyncio.create_task(runner())
494491
await asyncio.sleep(0.1)
495492

496493
self.assertFalse(r.done())
@@ -521,7 +518,7 @@ async def runner():
521518
g.create_task(crash_soon())
522519
await nested()
523520

524-
r = self.loop.create_task(runner())
521+
r = asyncio.create_task(runner())
525522
try:
526523
await r
527524
except ExceptionGroup as t:
@@ -634,7 +631,7 @@ async def runner():
634631
g.create_task(foo1())
635632
g.create_task(foo2())
636633

637-
r = self.loop.create_task(runner())
634+
r = asyncio.create_task(runner())
638635
await asyncio.sleep(0.05)
639636
r.cancel()
640637

0 commit comments

Comments
 (0)