def test_set_log(): task = Task("key") assert not hasattr(task, 'log') task.set_log() assert hasattr(task, 'log') assert hasattr(task, 'logstream') task.log.debug("please write me to log") assert "please write me to log" in task.logstream.getvalue()