def test_insert_characters(): screen = update(Screen(3, 3), ["sam", "is ", "foo"], colored=[0]) # a) normal case cursor = copy.copy(screen.cursor) screen.insert_characters(2) assert (screen.cursor.y, screen.cursor.x) == (cursor.y, cursor.x) assert screen[0] == [ screen.default_char, screen.default_char, Char("s", fg="red") ] # b) now inserting from the middle of the line screen.cursor.y, screen.cursor.x = 2, 1 screen.insert_characters(1) assert screen[2] == [Char("f"), screen.default_char, Char("o")] # c) inserting more than we have screen.insert_characters(10) assert screen[2] == [Char("f"), screen.default_char, screen.default_char] # d) 0 is 1 screen = update(Screen(3, 3), ["sam", "is ", "foo"], colored=[0]) screen.cursor_position() screen.insert_characters() assert screen[0] == [screen.default_char, Char("s", fg="red"), Char("a", fg="red")] screen = update(Screen(3, 3), ["sam", "is ", "foo"], colored=[0]) screen.cursor_position() screen.insert_characters(1) assert screen[0] == [screen.default_char, Char("s", fg="red"), Char("a", fg="red")]
def test_erase_character(): screen = update(Screen(3, 3), ["sam", "is ", "foo"], colored=[0]) screen.erase_characters(2) assert (screen.cursor.y, screen.cursor.x) == (0, 0) assert screen.display == [" m", "is ", "foo"] assert screen[0] == [ screen.default_char, screen.default_char, Char("m", fg="red") ] screen.cursor.y, screen.cursor.x = 2, 2 screen.erase_characters() assert (screen.cursor.y, screen.cursor.x) == (2, 2) assert screen.display == [" m", "is ", "fo "] screen.cursor.y, screen.cursor.x = 1, 1 screen.erase_characters(0) assert (screen.cursor.y, screen.cursor.x) == (1, 1) assert screen.display == [" m", "i ", "fo "] # ! extreme cases. screen = update(Screen(5, 1), ["12345"], colored=[0]) screen.cursor.x = 1 screen.erase_characters(3) assert (screen.cursor.y, screen.cursor.x) == (0, 1) assert screen.display == ["1 5"] assert screen[0] == [ Char("1", fg="red"), screen.default_char, screen.default_char, screen.default_char, Char("5", "red") ] screen = update(Screen(5, 1), ["12345"], colored=[0]) screen.cursor.x = 2 screen.erase_characters(10) assert (screen.cursor.y, screen.cursor.x) == (0, 2) assert screen.display == ["12 "] assert screen[0] == [ Char("1", fg="red"), Char("2", fg="red"), screen.default_char, screen.default_char, screen.default_char ] screen = update(Screen(5, 1), ["12345"], colored=[0]) screen.erase_characters(4) assert (screen.cursor.y, screen.cursor.x) == (0, 0) assert screen.display == [" 5"] assert screen[0] == [ screen.default_char, screen.default_char, screen.default_char, screen.default_char, Char("5", fg="red") ]
def test_resize(): screen = Screen(2, 2) screen.set_mode(mo.DECOM) screen.set_margins(0, 1) assert screen.size == (2, 2) assert len(screen) == 2 assert len(screen[0]) == 2 assert screen == [[screen.default_char, screen.default_char]] * 2 screen.resize(3, 3) assert screen.size == (3, 3) assert len(screen) == 3 assert len(screen[0]) == 3 assert screen == [[screen.default_char, screen.default_char, screen.default_char]] * 3 assert mo.DECOM not in screen.mode assert screen.margins == (0, 2) screen.resize(2, 2) assert screen.size == (2, 2) assert len(screen) == 2 assert len(screen[0]) == 2 assert screen == [[screen.default_char, screen.default_char]] * 2 # quirks: # a) if the current display is thinner than the requested size, # new columns should be added to the right. screen = update(Screen(2, 2), ["bo", "sh"], [None, None]) screen.resize(2, 3) assert screen.display == ["bo ", "sh "] # b) if the current display is wider than the requested size, # columns should be removed from the right... screen = update(Screen(2, 2), ["bo", "sh"], [None, None]) screen.resize(2, 1) assert screen.display == ["b", "s"] # c) if the current display is shorter than the requested # size, new rows should be added on the bottom. screen = update(Screen(2, 2), ["bo", "sh"], [None, None]) screen.resize(3, 2) assert screen.display == ["bo", "sh", " "] # d) if the current display is taller than the requested # size, rows should be removed from the top. screen = update(Screen(2, 2), ["bo", "sh"], [None, None]) screen.resize(1, 2) assert screen.display == ["sh"]
def test_erase_in_line(): screen = update(Screen(5, 5), ["sam i", "s foo", "but a", "re yo", "u? "], colored=[0]) screen.cursor_position(1, 3) # a) erase from cursor to the end of line screen.erase_in_line(0) assert (screen.cursor.y, screen.cursor.x) == (0, 2) assert screen.display == ["sa ", "s foo", "but a", "re yo", "u? "] assert screen.buffer[0] == [ Char("s", fg="red"), Char("a", fg="red"), screen.default_char, screen.default_char, screen.default_char ] # b) erase from the beginning of the line to the cursor screen = update(screen, ["sam i", "s foo", "but a", "re yo", "u? "], colored=[0]) screen.erase_in_line(1) assert (screen.cursor.y, screen.cursor.x) == (0, 2) assert screen.display == [" i", "s foo", "but a", "re yo", "u? "] assert screen.buffer[0] == [ screen.default_char, screen.default_char, screen.default_char, Char(" ", fg="red"), Char("i", fg="red") ] # c) erase the entire line screen = update(screen, ["sam i", "s foo", "but a", "re yo", "u? "], colored=[0]) screen.erase_in_line(2) assert (screen.cursor.y, screen.cursor.x) == (0, 2) assert screen.display == [" ", "s foo", "but a", "re yo", "u? "] assert screen.buffer[0] == [screen.default_char] * 5
def test_report_device_status(): screen = Screen(10, 10) acc = [] screen.write_process_input = acc.append # a) noop screen.report_device_status(42) assert not acc # b) terminal status screen.report_device_status(5) assert acc.pop() == ctrl.CSI + b"0n" # c) cursor position, DECOM off screen.cursor_to_column(5) screen.report_device_status(6) assert acc.pop() == ctrl.CSI + "{0};{1}R".format(1, 5).encode("utf-8") # d) cursor position, DECOM on screen.cursor_position() screen.set_margins(5, 9) screen.set_mode(mo.DECOM) screen.cursor_to_line(5) screen.report_device_status(6) assert acc.pop() == ctrl.CSI + "{0};{1}R".format(5, 1).encode("utf-8")
def test_cursor_position(): screen = Screen(10, 10) # a) testing that we expect 1-indexed values screen.cursor_position(5, 10) assert (screen.cursor.y, screen.cursor.x) == (4, 9) # b) but (0, 0) is also accepted and should be the same as (1, 1) screen.cursor_position(0, 10) assert (screen.cursor.y, screen.cursor.x) == (0, 9) # c) moving outside the margins constrains to within the screen # bounds screen.cursor_position(100, 5) assert (screen.cursor.y, screen.cursor.x) == (9, 4) screen.cursor_position(5, 100) assert (screen.cursor.y, screen.cursor.x) == (4, 9) # d) DECOM on screen.set_margins(5, 9) screen.set_mode(mo.DECOM) screen.cursor_position() assert (screen.cursor.y, screen.cursor.x) == (4, 0) screen.cursor_position(2, 0) assert (screen.cursor.y, screen.cursor.x) == (5, 0) # Note that cursor position doesn't change. screen.cursor_position(10, 0) assert (screen.cursor.y, screen.cursor.x) == (5, 0)
def test_clear_tabstops(): screen = Screen(10, 10) screen.clear_tab_stop(3) # a) clear a tabstop at current cusor location screen.cursor.x = 1 screen.set_tab_stop() screen.cursor.x = 5 screen.set_tab_stop() screen.clear_tab_stop() assert screen.tabstops == set([1]) screen.set_tab_stop() screen.clear_tab_stop(0) assert screen.tabstops == set([1]) # b) all tabstops screen.set_tab_stop() screen.cursor.x = 9 screen.set_tab_stop() screen.clear_tab_stop(3) assert not screen.tabstops
def test_remove_non_existant_attribute(): screen = Screen(2, 2) assert screen == [[screen.default_char, screen.default_char]] * 2 screen.select_graphic_rendition(24) # underline-off. assert screen == [[screen.default_char, screen.default_char]] * 2 assert not screen.cursor.attrs.underscore
def test_restore_cursor_with_none_saved(): screen = Screen(10, 10) screen.set_mode(mo.DECOM) screen.cursor.x, screen.cursor.y = 5, 5 screen.restore_cursor() assert (screen.cursor.y, screen.cursor.x) == (0, 0) assert mo.DECOM not in screen.mode
def test_draw_width0_irm(): screen = Screen(10, 1) screen.set_mode(mo.IRM) # The following should not insert any blanks. screen.draw("\N{ZERO WIDTH SPACE}".encode("utf-8")) screen.draw("\u0007".encode("utf-8")) # DELETE. assert screen.display == [" " * screen.columns]
def test_backspace(): screen = Screen(2, 2) assert screen.cursor.x == 0 screen.backspace() assert screen.cursor.x == 0 screen.cursor.x = 1 screen.backspace() assert screen.cursor.x == 0
def test_multi_attribs(): screen = Screen(2, 2) assert screen == [[screen.default_char, screen.default_char]] * 2 screen.select_graphic_rendition(1) screen.select_graphic_rendition(3) assert screen.cursor.attrs.bold assert screen.cursor.attrs.italics
def test_draw_cp437(): screen = Screen(5, 1) assert screen.charset == 0 screen.define_charset(b"U", b"(") screen.select_other_charset(b"@") screen.draw("α ± ε".encode("cp437")) assert screen.display == ["α ± ε"]
def test_unicode(): screen = Screen(4, 2) stream = Stream(screen) try: stream.feed("тест".encode("utf-8")) except UnicodeDecodeError: pytest.fail("Check your code -- we do accept unicode.") assert screen.display == ["тест", " "]
def test_private_report_device_attributes(): # Some console apps (e.g. ADOM) might add ``?`` to the DA request, # even though the VT102/VT220 spec does not allow this. screen = Screen(10, 10) stream = Stream(screen) acc = [] screen.write_process_input = acc.append stream.feed(ctrl.CSI + b"?0c") assert acc.pop() == ctrl.CSI + b"?6c"
def test_screen_set_icon_name_title(): screen = Screen(10, 1) screen.select_other_charset(b"@") text = "±" screen.set_icon_name(text.encode("latin-1")) assert screen.icon_name == text screen.set_title(text.encode("latin-1")) assert screen.title == text
def test_draw_width0_decawm_off(): screen = Screen(10, 1) screen.reset_mode(mo.DECAWM) screen.draw(" コンニチハ".encode("utf-8")) assert screen.cursor.x == screen.columns # The following should not advance the cursor. screen.draw("\N{ZERO WIDTH SPACE}".encode("utf-8")) screen.draw("\u0007".encode("utf-8")) # DELETE. assert screen.cursor.x == screen.columns
def test_colors256(): screen = Screen(2, 2) # a) OK-case. screen.select_graphic_rendition(38, 5, 0) screen.select_graphic_rendition(48, 5, 15) assert screen.cursor.attrs.fg == "000000" assert screen.cursor.attrs.bg == "ffffff" # b) invalid color. screen.select_graphic_rendition(48, 5, 100500)
def test_colors(): screen = Screen(2, 2) screen.select_graphic_rendition(30) screen.select_graphic_rendition(40) assert screen.cursor.attrs.fg == "black" assert screen.cursor.attrs.bg == "black" screen.select_graphic_rendition(31) assert screen.cursor.attrs.fg == "red" assert screen.cursor.attrs.bg == "black"
def test_colors24bit(): screen = Screen(2, 2) # a) OK-case screen.select_graphic_rendition(38, 2, 0, 0, 0) screen.select_graphic_rendition(48, 2, 255, 255, 255) assert screen.cursor.attrs.fg == "000000" assert screen.cursor.attrs.bg == "ffffff" # b) invalid color. screen.select_graphic_rendition(48, 2, 255)
def test_reset_resets_colors(): screen = Screen(2, 2) assert screen == [[screen.default_char, screen.default_char]] * 2 screen.select_graphic_rendition(30) screen.select_graphic_rendition(40) assert screen.cursor.attrs.fg == "black" assert screen.cursor.attrs.bg == "black" screen.select_graphic_rendition(0) assert screen.cursor.attrs == screen.default_char
def test_draw(): # ``DECAWM`` on (default). screen = Screen(3, 3) screen.set_mode(mo.LNM) assert mo.DECAWM in screen.mode for ch in "abc": screen.draw(ch) assert screen.display == ["abc", " ", " "] assert (screen.cursor.y, screen.cursor.x) == (0, 3) # ... one` more character -- now we got a linefeed! screen.draw("a") assert (screen.cursor.y, screen.cursor.x) == (1, 1) # ``DECAWM`` is off. screen = Screen(3, 3) screen.reset_mode(mo.DECAWM) for ch in "abc": screen.draw(ch) assert screen.display == ["abc", " ", " "] assert (screen.cursor.y, screen.cursor.x) == (0, 3) # No linefeed is issued on the end of the line ... screen.draw("a") assert screen.display == ["aba", " ", " "] assert (screen.cursor.y, screen.cursor.x) == (0, 3) # ``IRM`` mode is on, expecting new characters to move the old ones # instead of replacing them. screen.set_mode(mo.IRM) screen.cursor_position() screen.draw("x") assert screen.display == ["xab", " ", " "] screen.cursor_position() screen.draw("y") assert screen.display == ["yxa", " ", " "]
def test_colors_ignore_invalid(): screen = Screen(2, 2) default_attrs = screen.cursor.attrs screen.select_graphic_rendition(100500) assert screen.cursor.attrs == default_attrs screen.select_graphic_rendition(38, 100500) assert screen.cursor.attrs == default_attrs screen.select_graphic_rendition(48, 100500) assert screen.cursor.attrs == default_attrs
def test_colors(): screen = Screen(2, 2) assert screen == [[screen.default_char, screen.default_char]] * 2 screen.select_graphic_rendition(30) # black foreground screen.select_graphic_rendition(40) # black background assert screen.cursor.attrs.fg == "black" assert screen.cursor.attrs.bg == "black" screen.select_graphic_rendition(31) # red foreground assert screen.cursor.attrs.fg == "red" assert screen.cursor.attrs.bg == "black"
def test_colors_aixterm(): # See issue #57 on GitHub. screen = Screen(2, 2) # a) foreground color. screen.select_graphic_rendition(94) assert screen.cursor.attrs.fg == "blue" assert screen.cursor.attrs.bold # b) background color. screen.select_graphic_rendition(104) assert screen.cursor.attrs.bg == "blue" assert screen.cursor.attrs.bold
def test_alignment_display(): screen = Screen(5, 5) screen.set_mode(mo.LNM) screen.draw(b"a") screen.linefeed() screen.linefeed() screen.draw(b"b") assert screen.display == ["a ", " ", "b ", " ", " "] screen.alignment_display() assert screen.display == ["EEEEE", "EEEEE", "EEEEE", "EEEEE", "EEEEE"]
def test_report_device_attributes(): screen = Screen(10, 10) acc = [] screen.write_process_input = acc.append # a) noop screen.report_device_attributes(42) assert not acc # b) OK case screen.report_device_attributes() assert acc.pop() == ctrl.CSI + b"?6c"
def test_save_cursor(): # a) cursor position screen = Screen(10, 10) screen.save_cursor() screen.cursor.x, screen.cursor.y = 3, 5 screen.save_cursor() screen.cursor.x, screen.cursor.y = 4, 4 screen.restore_cursor() assert screen.cursor.x == 3 assert screen.cursor.y == 5 screen.restore_cursor() assert screen.cursor.x == 0 assert screen.cursor.y == 0 # b) modes screen = Screen(10, 10) screen.set_mode(mo.DECAWM, mo.DECOM) screen.save_cursor() screen.reset_mode(mo.DECAWM) screen.restore_cursor() assert mo.DECAWM in screen.mode assert mo.DECOM in screen.mode # c) attributes screen = Screen(10, 10) screen.select_graphic_rendition(4) screen.save_cursor() screen.select_graphic_rendition(24) assert screen.cursor.attrs == screen.default_char screen.restore_cursor() assert screen.cursor.attrs != screen.default_char assert screen.cursor.attrs == Char(" ", underscore=True)
def test_attributes(): screen = Screen(2, 2) assert screen.buffer == [[screen.default_char, screen.default_char]] * 2 screen.select_graphic_rendition(1) # bold. # Still default, since we haven't written anything. assert screen.buffer == [[screen.default_char, screen.default_char]] * 2 assert screen.cursor.attrs.bold screen.draw(b"f") assert screen.buffer == [[ Char("f", "default", "default", bold=True), screen.default_char ], [screen.default_char, screen.default_char]]
def test_hide_cursor(): screen = Screen(10, 10) # DECTCEM is set by default. assert mo.DECTCEM in screen.mode assert not screen.cursor.hidden # a) resetting DECTCEM hides the cursor. screen.reset_mode(mo.DECTCEM) assert screen.cursor.hidden # b) ... and it's back! screen.set_mode(mo.DECTCEM) assert not screen.cursor.hidden