cython

1
# mode: run
2
import sys
3

4
def test_print_refcount():
5
    """
6
    >>> test_print_refcount()
7
    """
8
    old_stdout = sys.stdout
9
    class StdoutGuard:
10
        def __getattr__(self, attr):
11
            sys.stdout = old_stdout
12
            raise RuntimeError
13
    sys.stdout = StdoutGuard()
14
    try:
15
        print "Hello", "world!"
16
    except RuntimeError:
17
        pass
18
    finally:
19
        sys.stdout = old_stdout
20
    class TriggerSIGSEGV(object):
21
        pass
22

23
def test_printone_refcount():
24
    """
25
    >>> test_printone_refcount()
26
    """
27
    old_stdout = sys.stdout
28
    class StdoutGuard:
29
        def __getattr__(self, attr):
30
            sys.stdout = old_stdout
31
            raise RuntimeError
32
    sys.stdout = StdoutGuard()
33
    try:
34
        print "Oops!"
35
    except RuntimeError:
36
        pass
37
    finally:
38
        sys.stdout = old_stdout
39
    class TriggerSIGSEGV(object):
40
        pass
41