| # |
| # Test script for the curses module |
| # |
| # This script doesn't actually display anything very coherent. but it |
| # does call (nearly) every method and function. |
| # |
| # Functions not tested: {def,reset}_{shell,prog}_mode, getch(), getstr(), |
| # init_color() |
| # Only called, not tested: getmouse(), ungetmouse() |
| # |
| |
| import os |
| import string |
| import sys |
| import tempfile |
| import unittest |
| |
| from test.support import (requires, import_module, verbose, run_unittest, |
| SaveSignals) |
| |
| |
| # 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. |
| requires('curses') |
| |
| # If either of these don't exist, skip the tests. |
| curses = import_module('curses') |
| import_module('curses.ascii') |
| import_module('curses.textpad') |
| try: |
| import curses.panel |
| except ImportError: |
| pass |
| |
| def requires_curses_func(name): |
| return unittest.skipUnless(hasattr(curses, name), |
| 'requires curses.%s' % name) |
| |
| term = os.environ.get('TERM') |
| |
| # If newterm was supported we could use it instead of initscr and not exit |
| @unittest.skipIf(not term or term == 'unknown', |
| "$TERM=%r, calling initscr() may cause exit" % term) |
| @unittest.skipIf(sys.platform == "cygwin", |
| "cygwin's curses mostly just hangs") |
| class TestCurses(unittest.TestCase): |
| |
| @classmethod |
| def setUpClass(cls): |
| if not sys.__stdout__.isatty(): |
| # Temporary skip tests on non-tty |
| raise unittest.SkipTest('sys.__stdout__ is not a tty') |
| cls.tmp = tempfile.TemporaryFile() |
| fd = cls.tmp.fileno() |
| else: |
| cls.tmp = None |
| fd = sys.__stdout__.fileno() |
| # testing setupterm() inside initscr/endwin |
| # causes terminal breakage |
| curses.setupterm(fd=fd) |
| |
| @classmethod |
| def tearDownClass(cls): |
| if cls.tmp: |
| cls.tmp.close() |
| del cls.tmp |
| |
| def setUp(self): |
| self.save_signals = SaveSignals() |
| self.save_signals.save() |
| if verbose: |
| # just to make the test output a little more readable |
| print('') |
| self.stdscr = curses.initscr() |
| curses.savetty() |
| |
| def tearDown(self): |
| curses.resetty() |
| curses.endwin() |
| self.save_signals.restore() |
| |
| def test_window_funcs(self): |
| "Test the methods of windows" |
| stdscr = self.stdscr |
| 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.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('|', '!', '-', '_', |
| '+', '\\', '#', '/') |
| with self.assertRaises(TypeError, |
| msg="Expected win.border() to raise TypeError"): |
| win.border(65, 66, 67, 68, |
| 69, [], 71, 72) |
| |
| win.box(65, 67) |
| win.box('!', '_') |
| win.box(b':', b'~') |
| self.assertRaises(TypeError, win.box, 65, 66, 67) |
| self.assertRaises(TypeError, win.box, 65) |
| win.box() |
| |
| 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) |
| if hasattr(stdscr, 'immedok'): |
| stdscr.immedok(1) |
| stdscr.immedok(0) |
| 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, 2, 1, 3, 3) |
| win2.overwrite(win, 1, 2, 2, 1, 3, 3) |
| 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) |
| if hasattr(stdscr, 'syncok') and not sys.platform.startswith("sunos"): |
| 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) |
| if hasattr(stdscr, 'chgat'): |
| 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(stdscr, 'resize'): |
| stdscr.resize(25, 80) |
| if hasattr(stdscr, 'enclose'): |
| stdscr.enclose(10, 10) |
| |
| self.assertRaises(ValueError, stdscr.getstr, -400) |
| self.assertRaises(ValueError, stdscr.getstr, 2, 3, -400) |
| self.assertRaises(ValueError, stdscr.instr, -2) |
| self.assertRaises(ValueError, stdscr.instr, 2, 3, -2) |
| |
| |
| def test_module_funcs(self): |
| "Test module-level functions" |
| for func in [curses.baudrate, curses.beep, curses.can_change_color, |
| curses.cbreak, curses.def_prog_mode, curses.doupdate, |
| 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]: |
| func() |
| if hasattr(curses, 'filter'): |
| curses.filter() |
| if hasattr(curses, 'getsyx'): |
| curses.getsyx() |
| |
| # Functions that actually need arguments |
| if curses.tigetstr("cnorm"): |
| curses.curs_set(1) |
| curses.delay_output(1) |
| curses.echo() ; curses.echo(1) |
| |
| with tempfile.TemporaryFile() as f: |
| self.stdscr.putwin(f) |
| f.seek(0) |
| curses.getwin(f) |
| |
| 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(b'abc') |
| curses.qiflush() |
| curses.raw() ; curses.raw(1) |
| if hasattr(curses, 'setsyx'): |
| curses.setsyx(5,5) |
| curses.tigetflag('hc') |
| curses.tigetnum('co') |
| curses.tigetstr('cr') |
| curses.tparm(b'cr') |
| if hasattr(curses, 'typeahead'): |
| curses.typeahead(sys.__stdin__.fileno()) |
| curses.unctrl('a') |
| curses.ungetch('a') |
| if hasattr(curses, 'use_env'): |
| curses.use_env(1) |
| |
| # Functions only available on a few platforms |
| def test_colors_funcs(self): |
| if not curses.has_colors(): |
| self.skipTest('requires colors support') |
| 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() |
| |
| @requires_curses_func('keyname') |
| def test_keyname(self): |
| curses.keyname(13) |
| |
| @requires_curses_func('has_key') |
| def test_has_key(self): |
| curses.has_key(13) |
| |
| @requires_curses_func('getmouse') |
| def test_getmouse(self): |
| (availmask, oldmask) = curses.mousemask(curses.BUTTON1_PRESSED) |
| if availmask == 0: |
| self.skipTest('mouse stuff not available') |
| curses.mouseinterval(10) |
| # just verify these don't cause errors |
| curses.ungetmouse(0, 0, 0, 0, curses.BUTTON1_PRESSED) |
| m = curses.getmouse() |
| |
| @requires_curses_func('panel') |
| def test_userptr_without_set(self): |
| w = curses.newwin(10, 10) |
| p = curses.panel.new_panel(w) |
| # try to access userptr() before calling set_userptr() -- segfaults |
| with self.assertRaises(curses.panel.error, |
| msg='userptr should fail since not set'): |
| p.userptr() |
| |
| @requires_curses_func('panel') |
| def test_userptr_memory_leak(self): |
| w = curses.newwin(10, 10) |
| p = curses.panel.new_panel(w) |
| obj = object() |
| nrefs = sys.getrefcount(obj) |
| for i in range(100): |
| p.set_userptr(obj) |
| |
| p.set_userptr(None) |
| self.assertEqual(sys.getrefcount(obj), nrefs, |
| "set_userptr leaked references") |
| |
| @requires_curses_func('panel') |
| def test_userptr_segfault(self): |
| w = curses.newwin(10, 10) |
| panel = curses.panel.new_panel(w) |
| class A: |
| def __del__(self): |
| panel.set_userptr(None) |
| panel.set_userptr(A()) |
| panel.set_userptr(None) |
| |
| @requires_curses_func('panel') |
| def test_new_curses_panel(self): |
| w = curses.newwin(10, 10) |
| panel = curses.panel.new_panel(w) |
| self.assertRaises(TypeError, type(panel)) |
| |
| @requires_curses_func('is_term_resized') |
| def test_is_term_resized(self): |
| curses.is_term_resized(*self.stdscr.getmaxyx()) |
| |
| @requires_curses_func('resize_term') |
| def test_resize_term(self): |
| curses.resize_term(*self.stdscr.getmaxyx()) |
| |
| @requires_curses_func('resizeterm') |
| def test_resizeterm(self): |
| stdscr = self.stdscr |
| lines, cols = curses.LINES, curses.COLS |
| new_lines = lines - 1 |
| new_cols = cols + 1 |
| curses.resizeterm(new_lines, new_cols) |
| |
| self.assertEqual(curses.LINES, new_lines) |
| self.assertEqual(curses.COLS, new_cols) |
| |
| def test_issue6243(self): |
| curses.ungetch(1025) |
| self.stdscr.getkey() |
| |
| def test_issue10570(self): |
| b = curses.tparm(curses.tigetstr("cup"), 5, 3) |
| self.assertIs(type(b), bytes) |
| |
| def test_issue13051(self): |
| stdscr = self.stdscr |
| if not hasattr(stdscr, 'resize'): |
| raise unittest.SkipTest('requires curses.window.resize') |
| box = curses.textpad.Textbox(stdscr, insert_mode=True) |
| lines, cols = stdscr.getmaxyx() |
| stdscr.resize(lines-2, cols-2) |
| # this may cause infinite recursion, leading to a RuntimeError |
| box._insert_printable_char('a') |
| |
| |
| class TestAscii(unittest.TestCase): |
| |
| def test_controlnames(self): |
| for name in curses.ascii.controlnames: |
| self.assertTrue(hasattr(curses.ascii, name), name) |
| |
| def test_ctypes(self): |
| def check(func, expected): |
| self.assertEqual(func(i), expected) |
| self.assertEqual(func(c), expected) |
| |
| for i in range(256): |
| c = b = chr(i) |
| check(curses.ascii.isalnum, b.isalnum()) |
| check(curses.ascii.isalpha, b.isalpha()) |
| check(curses.ascii.isdigit, b.isdigit()) |
| check(curses.ascii.islower, b.islower()) |
| check(curses.ascii.isspace, b.isspace()) |
| check(curses.ascii.isupper, b.isupper()) |
| |
| check(curses.ascii.isascii, i < 128) |
| check(curses.ascii.ismeta, i >= 128) |
| check(curses.ascii.isctrl, i < 32) |
| check(curses.ascii.iscntrl, i < 32 or i == 127) |
| check(curses.ascii.isblank, c in ' \t') |
| check(curses.ascii.isgraph, 32 < i <= 126) |
| check(curses.ascii.isprint, 32 <= i <= 126) |
| check(curses.ascii.ispunct, c in string.punctuation) |
| check(curses.ascii.isxdigit, c in string.hexdigits) |
| |
| for i in (-2, -1, 256, sys.maxunicode, sys.maxunicode+1): |
| self.assertFalse(curses.ascii.isalnum(i)) |
| self.assertFalse(curses.ascii.isalpha(i)) |
| self.assertFalse(curses.ascii.isdigit(i)) |
| self.assertFalse(curses.ascii.islower(i)) |
| self.assertFalse(curses.ascii.isspace(i)) |
| self.assertFalse(curses.ascii.isupper(i)) |
| |
| self.assertFalse(curses.ascii.isascii(i)) |
| self.assertFalse(curses.ascii.isctrl(i)) |
| self.assertFalse(curses.ascii.iscntrl(i)) |
| self.assertFalse(curses.ascii.isblank(i)) |
| self.assertFalse(curses.ascii.isgraph(i)) |
| self.assertFalse(curses.ascii.isprint(i)) |
| self.assertFalse(curses.ascii.ispunct(i)) |
| self.assertFalse(curses.ascii.isxdigit(i)) |
| |
| self.assertFalse(curses.ascii.ismeta(-1)) |
| |
| def test_ascii(self): |
| ascii = curses.ascii.ascii |
| self.assertEqual(ascii('\xc1'), 'A') |
| self.assertEqual(ascii('A'), 'A') |
| self.assertEqual(ascii(ord('\xc1')), ord('A')) |
| |
| def test_ctrl(self): |
| ctrl = curses.ascii.ctrl |
| self.assertEqual(ctrl('J'), '\n') |
| self.assertEqual(ctrl('\n'), '\n') |
| self.assertEqual(ctrl('@'), '\0') |
| self.assertEqual(ctrl(ord('J')), ord('\n')) |
| |
| def test_alt(self): |
| alt = curses.ascii.alt |
| self.assertEqual(alt('\n'), '\x8a') |
| self.assertEqual(alt('A'), '\xc1') |
| self.assertEqual(alt(ord('A')), 0xc1) |
| |
| def test_unctrl(self): |
| unctrl = curses.ascii.unctrl |
| self.assertEqual(unctrl('a'), 'a') |
| self.assertEqual(unctrl('A'), 'A') |
| self.assertEqual(unctrl(';'), ';') |
| self.assertEqual(unctrl(' '), ' ') |
| self.assertEqual(unctrl('\x7f'), '^?') |
| self.assertEqual(unctrl('\n'), '^J') |
| self.assertEqual(unctrl('\0'), '^@') |
| self.assertEqual(unctrl(ord('A')), 'A') |
| self.assertEqual(unctrl(ord('\n')), '^J') |
| # Meta-bit characters |
| self.assertEqual(unctrl('\x8a'), '!^J') |
| self.assertEqual(unctrl('\xc1'), '!A') |
| self.assertEqual(unctrl(ord('\x8a')), '!^J') |
| self.assertEqual(unctrl(ord('\xc1')), '!A') |
| |
| |
| def test_main(): |
| run_unittest(TestCurses, TestAscii) |
| |
| |
| if __name__ == "__main__": |
| unittest.main() |
