def test_teardown_open_stream(): '''Teardown open stream.''' target = StringIO() stream = Stream(stream=target) stream.output(['test']) stream.teardown() assert not target.closed assert target.getvalue() == 'test'
def test_teardown_closed_stream(): '''Teardown closed stream without error.''' target = StringIO() stream = Stream(stream=target) target.close() stream.teardown()