test_stack_saved.py 446 B

12345678910111213141516171819
  1. import greenlet
  2. from . import TestCase
  3. class Test(TestCase):
  4. def test_stack_saved(self):
  5. main = greenlet.getcurrent()
  6. self.assertEqual(main._stack_saved, 0)
  7. def func():
  8. main.switch(main._stack_saved)
  9. g = greenlet.greenlet(func)
  10. x = g.switch()
  11. self.assertGreater(x, 0)
  12. self.assertGreater(g._stack_saved, 0)
  13. g.switch()
  14. self.assertEqual(g._stack_saved, 0)