def test_session_fail(): before_prompt() print("PROMPT$ ", end='') after_prompt() print("command") before_output() print("command output 1") print("command output 2") print("command output 3", file=sys.stderr) after_output(1)
def test_session_prompt_newline(): before_prompt() print("PROMPT\n$ ", end='') after_prompt() print("command") before_output() print("command output 1") print("command output 2") print("command output 3", file=sys.stderr) after_output(0)
def test_after_output_range(): after_output(0) after_output(255) raises(ValueError, after_output, 256) raises(ValueError, after_output, -1)
def test_Output(): with Output() as o: print("Output") with Output() as o: print("Output") o.set_command_status(0) with Output() as o: print("Output") o.set_command_status(1) def test_readline_invisible(): assert readline_invisible(BEFORE_PROMPT) == "\001" + BEFORE_PROMPT + "\002" if __name__ == '__main__': test_session() test_session_fail() test_session_prompt_newline() # In case this file is run in a shell that itself has shell integration # enabled, some dummy output. before_prompt() print("DUMMY PROMPT, IGNORE$ ", end='') after_prompt() print("dummy command") before_output() print("dummy output") after_output(0)