Capture python output in a list

capture-output.py

import sys
from io import StringIO

def print_function() :
    print( "Hello")
    print ( "World")
    print ( "New\nLine")

class Capturing(list):
    def __enter__(self):
        self._stdout = sys.stdout
        sys.stdout = StringIO()
        self._current_string = sys.stdout
        return self

    def __exit__(self, *args):
        self.extend(self._current_string.getvalue().splitlines())
        del self._current_string
        sys.stdout = self._stdout

with Capturing() as output:
    print_function()

print(output)