def test_output(): '''Test outputting to specified stream.''' target = StringIO() formatter = Mock() formatter.format = Mock(return_value='A message') stream = Stream( stream=target, formatter=formatter ) log = Log(level='info', message='A message') stream.handle(log) assert target.getvalue() == 'A message'