def test(): queue = MagicMock() display = Display() display.set_queue(queue) display.display('foo') queue.send_display.assert_called_once_with( 'foo', color=None, stderr=False, screen_only=False, log_only=False, newline=True )
def test(): display = Display() display.set_queue('foo') assert display._final_q == 'foo'