codekingpro/portable-devtools

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