Capture and test sys.stdout with nosetest

nosetest-stdout.md

Capture and test sys.stdout with nosetest

import sys
import tempfile

import sys
import tempfile

class Base(object):
    def setup(self):
        sys.stdout = self._stdout = tempfile.NamedTemporaryFile('w+', delete=False)
        sys.stderr.write(self._stdout.name + ' ')

    def teardown(self):
        self._stdout.close()
        sys.stdout = sys.__stdout__

    @property
    def stdout(self):
        self._stdout.seek(0)
        return self._stdout.read()[:-1]    # Remove trailing \n:w


class Test(Base):

    def test1(self):
        print("123")
        assert self.stdout == "123"
        print("456")
        assert self.stdout == "123\n456"