File: test_code.py

package info (click to toggle)
pypy 7.0.0%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 107,216 kB
  • sloc: python: 1,201,787; ansic: 62,419; asm: 5,169; cpp: 3,017; sh: 2,534; makefile: 545; xml: 243; lisp: 45; awk: 4
file content (19 lines) | stat: -rw-r--r-- 488 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
 
import sys
import cStringIO
import code


def test_flush_stdout_on_error():
    runner = code.InteractiveInterpreter()
    old_stdout = sys.stdout
    try:
        mystdout = cStringIO.StringIO()
        sys.stdout = mystdout
        runner.runcode(compile("print 5,;0/0", "<interactive>", "exec"))
    finally:
        sys.stdout = old_stdout

    if '__pypy__' in sys.builtin_module_names:
        assert mystdout.getvalue() == "5\n"
    else:
        assert mystdout.getvalue() == "5"