These files are a subset of the python-2.7.2.tgz distribution from python.org. Changed files from PyMod-2.7.2 have been copied into the corresponding directories of this tree, replacing the original files in the distribution. Signed-off-by: daryl.mcdaniel@intel.com git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@13197 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			291 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			291 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| #
 | |
| # 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()
 |