This commit is contained in:
Jack Robison 2019-01-25 16:20:19 -05:00
parent a7610e3d34
commit 3e3f6c0e22
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2

View file

@ -49,8 +49,8 @@ class TestAsyncGeneratorJunction(AsyncioTestCase):
async def test_one_stopped_first(self): async def test_one_stopped_first(self):
expected_order = [1, 2, 1, 1, 2, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2] expected_order = [1, 2, 1, 1, 2, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2]
fast_gen = MockAsyncGen(self.loop, 1, 0.1, 5) fast_gen = MockAsyncGen(self.loop, 1, 0.101, 5)
slow_gen = MockAsyncGen(self.loop, 2, 0.2) slow_gen = MockAsyncGen(self.loop, 2, 0.201)
await self._test_junction(expected_order, fast_gen, slow_gen) await self._test_junction(expected_order, fast_gen, slow_gen)
self.assertEqual(fast_gen.called_close, True) self.assertEqual(fast_gen.called_close, True)
self.assertEqual(slow_gen.called_close, True) self.assertEqual(slow_gen.called_close, True)
@ -62,10 +62,10 @@ class TestAsyncGeneratorJunction(AsyncioTestCase):
for i in range(10): for i in range(10):
if i == 5: if i == 5:
return return
await asyncio.sleep(0.1) await asyncio.sleep(0.101)
yield 1 yield 1
slow_gen = MockAsyncGen(self.loop, 2, 0.2) slow_gen = MockAsyncGen(self.loop, 2, 0.201)
await self._test_junction(expected_order, fast_gen(), slow_gen) await self._test_junction(expected_order, fast_gen(), slow_gen)
self.assertEqual(slow_gen.called_close, True) self.assertEqual(slow_gen.called_close, True)