Skip to content

Commit fa6a1fa

Browse files
committed
tests: use Python 3.6 APIs
1 parent de11f5e commit fa6a1fa

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

test.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1780,27 +1780,30 @@ async def func():
17801780
return 42
17811781

17821782
memoized = MemoizedMutex(func)
1783-
task1 = asyncio.create_task(memoized())
1784-
task2 = asyncio.create_task(memoized())
1785-
task3 = asyncio.create_task(memoized())
1783+
loop = asyncio.get_event_loop()
1784+
1785+
task1 = loop.create_task(memoized())
1786+
task2 = loop.create_task(memoized())
1787+
task3 = loop.create_task(memoized())
17861788

17871789
self.assertEqual(await task1, 42)
17881790
self.assertEqual(await task2, 42)
17891791
self.assertEqual(await task3, 42)
17901792

1791-
17921793
@async_test
17931794
async def test_usage_after_resolve(self):
17941795
async def func():
17951796
await asyncio.sleep(0)
17961797
return 42
17971798

17981799
memoized = MemoizedMutex(func)
1799-
task1 = asyncio.create_task(memoized())
1800+
loop = asyncio.get_event_loop()
1801+
1802+
task1 = loop.create_task(memoized())
18001803
self.assertEqual(await task1, 42)
18011804

1802-
task2 = asyncio.create_task(memoized())
1803-
task3 = asyncio.create_task(memoized())
1805+
task2 = loop.create_task(memoized())
1806+
task3 = loop.create_task(memoized())
18041807
self.assertEqual(await task2, 42)
18051808
self.assertEqual(await task3, 42)
18061809

0 commit comments

Comments
 (0)