# | |
# Test script for the curses module | |
# | |
# This script doesn't actually display anything very coherent. but it | |
# does call every method and function. | |
# | |
# Functions not tested: {def,reset}_{shell,prog}_mode, getch(), getstr(), | |
# init_color() | |
# Only called, not tested: getmouse(), ungetmouse() | |
# | |
import sys, tempfile, os | |
# Optionally test curses module. This currently requires that the | |
# 'curses' resource be given on the regrtest command line using the -u | |
# option. If not available, nothing after this line will be executed. | |
import unittest | |
from test.test_support import requires, import_module | |
requires('curses') | |
curses = import_module('curses') | |
curses.panel = import_module('curses.panel') | |
# XXX: if newterm was supported we could use it instead of initscr and not exit | |
term = os.environ.get('TERM') | |
if not term or term == 'unknown': | |
raise unittest.SkipTest, "$TERM=%r, calling initscr() may cause exit" % term | |
if sys.platform == "cygwin": | |
raise unittest.SkipTest("cygwin's curses mostly just hangs") | |
def window_funcs(stdscr): | |
"Test the methods of windows" | |
win = curses.newwin(10,10) | |
win = curses.newwin(5,5, 5,5) | |
win2 = curses.newwin(15,15, 5,5) | |
for meth in [stdscr.addch, stdscr.addstr]: | |
for args in [('a'), ('a', curses.A_BOLD), | |
(4,4, 'a'), (5,5, 'a', curses.A_BOLD)]: | |
meth(*args) | |
for meth in [stdscr.box, stdscr.clear, stdscr.clrtobot, | |
stdscr.clrtoeol, stdscr.cursyncup, stdscr.delch, | |
stdscr.deleteln, stdscr.erase, stdscr.getbegyx, | |
stdscr.getbkgd, stdscr.getkey, stdscr.getmaxyx, | |
stdscr.getparyx, stdscr.getyx, stdscr.inch, | |
stdscr.insertln, stdscr.instr, stdscr.is_wintouched, | |
win.noutrefresh, stdscr.redrawwin, stdscr.refresh, | |
stdscr.standout, stdscr.standend, stdscr.syncdown, | |
stdscr.syncup, stdscr.touchwin, stdscr.untouchwin]: | |
meth() | |
stdscr.addnstr('1234', 3) | |
stdscr.addnstr('1234', 3, curses.A_BOLD) | |
stdscr.addnstr(4,4, '1234', 3) | |
stdscr.addnstr(5,5, '1234', 3, curses.A_BOLD) | |
stdscr.attron(curses.A_BOLD) | |
stdscr.attroff(curses.A_BOLD) | |
stdscr.attrset(curses.A_BOLD) | |
stdscr.bkgd(' ') | |
stdscr.bkgd(' ', curses.A_REVERSE) | |
stdscr.bkgdset(' ') | |
stdscr.bkgdset(' ', curses.A_REVERSE) | |
win.border(65, 66, 67, 68, | |
69, 70, 71, 72) | |
win.border('|', '!', '-', '_', | |
'+', '\\', '#', '/') | |
try: | |
win.border(65, 66, 67, 68, | |
69, [], 71, 72) | |
except TypeError: | |
pass | |
else: | |
raise RuntimeError, "Expected win.border() to raise TypeError" | |
stdscr.clearok(1) | |
win4 = stdscr.derwin(2,2) | |
win4 = stdscr.derwin(1,1, 5,5) | |
win4.mvderwin(9,9) | |
stdscr.echochar('a') | |
stdscr.echochar('a', curses.A_BOLD) | |
stdscr.hline('-', 5) | |
stdscr.hline('-', 5, curses.A_BOLD) | |
stdscr.hline(1,1,'-', 5) | |
stdscr.hline(1,1,'-', 5, curses.A_BOLD) | |
stdscr.idcok(1) | |
stdscr.idlok(1) | |
stdscr.immedok(1) | |
stdscr.insch('c') | |
stdscr.insdelln(1) | |
stdscr.insnstr('abc', 3) | |
stdscr.insnstr('abc', 3, curses.A_BOLD) | |
stdscr.insnstr(5, 5, 'abc', 3) | |
stdscr.insnstr(5, 5, 'abc', 3, curses.A_BOLD) | |
stdscr.insstr('def') | |
stdscr.insstr('def', curses.A_BOLD) | |
stdscr.insstr(5, 5, 'def') | |
stdscr.insstr(5, 5, 'def', curses.A_BOLD) | |
stdscr.is_linetouched(0) | |
stdscr.keypad(1) | |
stdscr.leaveok(1) | |
stdscr.move(3,3) | |
win.mvwin(2,2) | |
stdscr.nodelay(1) | |
stdscr.notimeout(1) | |
win2.overlay(win) | |
win2.overwrite(win) | |
win2.overlay(win, 1, 2, 3, 3, 2, 1) | |
win2.overwrite(win, 1, 2, 3, 3, 2, 1) | |
stdscr.redrawln(1,2) | |
stdscr.scrollok(1) | |
stdscr.scroll() | |
stdscr.scroll(2) | |
stdscr.scroll(-3) | |
stdscr.move(12, 2) | |
stdscr.setscrreg(10,15) | |
win3 = stdscr.subwin(10,10) | |
win3 = stdscr.subwin(10,10, 5,5) | |
stdscr.syncok(1) | |
stdscr.timeout(5) | |
stdscr.touchline(5,5) | |
stdscr.touchline(5,5,0) | |
stdscr.vline('a', 3) | |
stdscr.vline('a', 3, curses.A_STANDOUT) | |
stdscr.chgat(5, 2, 3, curses.A_BLINK) | |
stdscr.chgat(3, curses.A_BOLD) | |
stdscr.chgat(5, 8, curses.A_UNDERLINE) | |
stdscr.chgat(curses.A_BLINK) | |
stdscr.refresh() | |
stdscr.vline(1,1, 'a', 3) | |
stdscr.vline(1,1, 'a', 3, curses.A_STANDOUT) | |
if hasattr(curses, 'resize'): | |
stdscr.resize() | |
if hasattr(curses, 'enclose'): | |
stdscr.enclose() | |
def module_funcs(stdscr): | |
"Test module-level functions" | |
for func in [curses.baudrate, curses.beep, curses.can_change_color, | |
curses.cbreak, curses.def_prog_mode, curses.doupdate, | |
curses.filter, curses.flash, curses.flushinp, | |
curses.has_colors, curses.has_ic, curses.has_il, | |
curses.isendwin, curses.killchar, curses.longname, | |
curses.nocbreak, curses.noecho, curses.nonl, | |
curses.noqiflush, curses.noraw, | |
curses.reset_prog_mode, curses.termattrs, | |
curses.termname, curses.erasechar, curses.getsyx]: | |
func() | |
# Functions that actually need arguments | |
if curses.tigetstr("cnorm"): | |
curses.curs_set(1) | |
curses.delay_output(1) | |
curses.echo() ; curses.echo(1) | |
f = tempfile.TemporaryFile() | |
stdscr.putwin(f) | |
f.seek(0) | |
curses.getwin(f) | |
f.close() | |
curses.halfdelay(1) | |
curses.intrflush(1) | |
curses.meta(1) | |
curses.napms(100) | |
curses.newpad(50,50) | |
win = curses.newwin(5,5) | |
win = curses.newwin(5,5, 1,1) | |
curses.nl() ; curses.nl(1) | |
curses.putp('abc') | |
curses.qiflush() | |
curses.raw() ; curses.raw(1) | |
curses.setsyx(5,5) | |
curses.tigetflag('hc') | |
curses.tigetnum('co') | |
curses.tigetstr('cr') | |
curses.tparm('cr') | |
curses.typeahead(sys.__stdin__.fileno()) | |
curses.unctrl('a') | |
curses.ungetch('a') | |
curses.use_env(1) | |
# Functions only available on a few platforms | |
if curses.has_colors(): | |
curses.start_color() | |
curses.init_pair(2, 1,1) | |
curses.color_content(1) | |
curses.color_pair(2) | |
curses.pair_content(curses.COLOR_PAIRS - 1) | |
curses.pair_number(0) | |
if hasattr(curses, 'use_default_colors'): | |
curses.use_default_colors() | |
if hasattr(curses, 'keyname'): | |
curses.keyname(13) | |
if hasattr(curses, 'has_key'): | |
curses.has_key(13) | |
if hasattr(curses, 'getmouse'): | |
(availmask, oldmask) = curses.mousemask(curses.BUTTON1_PRESSED) | |
# availmask indicates that mouse stuff not available. | |
if availmask != 0: | |
curses.mouseinterval(10) | |
# just verify these don't cause errors | |
curses.ungetmouse(0, 0, 0, 0, curses.BUTTON1_PRESSED) | |
m = curses.getmouse() | |
if hasattr(curses, 'is_term_resized'): | |
curses.is_term_resized(*stdscr.getmaxyx()) | |
if hasattr(curses, 'resizeterm'): | |
curses.resizeterm(*stdscr.getmaxyx()) | |
if hasattr(curses, 'resize_term'): | |
curses.resize_term(*stdscr.getmaxyx()) | |
def unit_tests(): | |
from curses import ascii | |
for ch, expected in [('a', 'a'), ('A', 'A'), | |
(';', ';'), (' ', ' '), | |
('\x7f', '^?'), ('\n', '^J'), ('\0', '^@'), | |
# Meta-bit characters | |
('\x8a', '!^J'), ('\xc1', '!A'), | |
]: | |
if ascii.unctrl(ch) != expected: | |
print 'curses.unctrl fails on character', repr(ch) | |
def test_userptr_without_set(stdscr): | |
w = curses.newwin(10, 10) | |
p = curses.panel.new_panel(w) | |
# try to access userptr() before calling set_userptr() -- segfaults | |
try: | |
p.userptr() | |
raise RuntimeError, 'userptr should fail since not set' | |
except curses.panel.error: | |
pass | |
def test_resize_term(stdscr): | |
if hasattr(curses, 'resizeterm'): | |
lines, cols = curses.LINES, curses.COLS | |
curses.resizeterm(lines - 1, cols + 1) | |
if curses.LINES != lines - 1 or curses.COLS != cols + 1: | |
raise RuntimeError, "Expected resizeterm to update LINES and COLS" | |
def test_issue6243(stdscr): | |
curses.ungetch(1025) | |
stdscr.getkey() | |
def main(stdscr): | |
curses.savetty() | |
try: | |
module_funcs(stdscr) | |
window_funcs(stdscr) | |
test_userptr_without_set(stdscr) | |
test_resize_term(stdscr) | |
test_issue6243(stdscr) | |
finally: | |
curses.resetty() | |
if __name__ == '__main__': | |
curses.wrapper(main) | |
unit_tests() | |
else: | |
if not sys.__stdout__.isatty(): | |
raise unittest.SkipTest("sys.__stdout__ is not a tty") | |
# testing setupterm() inside initscr/endwin | |
# causes terminal breakage | |
curses.setupterm(fd=sys.__stdout__.fileno()) | |
try: | |
stdscr = curses.initscr() | |
main(stdscr) | |
finally: | |
curses.endwin() | |
unit_tests() |