Testing capturer to output ANSI escape sequences over subprocesses.

test_capturer.py

#!/usr/bin/env python

"""
Testing capturer to output ANSI escape sequences over subprocesses.
"""

import subprocess

from capturer import CaptureOutput


cmd0 = """echo '{"result":42}' > foo.json; pygmentize foo.json"""
cmd1 = """pygmentize foo.json"""
cmd2 = """echo foo.json | pygmentize"""

for cmd in (cmd0, cmd1, cmd2):
    print('cmd')
    print('  ' + cmd)

    print('subprocess.getoutput(cmd)')
    print('  ' + subprocess.getoutput(cmd).strip())

    print("subprocess.check_output(cmd.split()).decode('utf-8')")
    print('  ' + subprocess.check_output(cmd.split()).decode('utf-8').strip())

    print('capturer, subprocess.call(cmd.split())')
    with CaptureOutput() as capturer:
        subprocess.call(cmd.split()) # also prints output
        for line in capturer.get_lines():
            print('  ' + line)

    print('capturer, subprocess.check_output(cmd.split())')
    with CaptureOutput() as capturer:
        output = subprocess.check_output(cmd.split()) # also prints output
        print(output.decode('utf-8'))
        for line in capturer.get_lines():
            print('  ' + line)

    print()