def test_tp_await_2(self):
# Test tp_await to __await__ mapping
from _testcapi import awaitType as at
future = at(iter([1]))
self.assertEqual(next(future.__await__()), 1)
async def foo():
future = at(1)
return (await future)
async def foo():
future = at(iter([1]))
return (await future)
