20 lines
446 B
Python
20 lines
446 B
Python
|
import greenlet
|
||
|
from . import TestCase
|
||
|
|
||
|
|
||
|
class Test(TestCase):
|
||
|
|
||
|
def test_stack_saved(self):
|
||
|
main = greenlet.getcurrent()
|
||
|
self.assertEqual(main._stack_saved, 0)
|
||
|
|
||
|
def func():
|
||
|
main.switch(main._stack_saved)
|
||
|
|
||
|
g = greenlet.greenlet(func)
|
||
|
x = g.switch()
|
||
|
self.assertGreater(x, 0)
|
||
|
self.assertGreater(g._stack_saved, 0)
|
||
|
g.switch()
|
||
|
self.assertEqual(g._stack_saved, 0)
|