test_pyshell.py

"Test pyshell, coverage 12%."
# Plus coverage of test_warning. Was 20% with test_openshell.

fromidlelibimportpyshell
importunittest
fromtest.supportimportrequires
fromtkinterimportTk


classFunctionTest(unittest.TestCase):
# Test stand-alone module level non-gui functions.

deftest_restart_line_wide(self):
eq=self.assertEqual
forfile, mul, extrain (('', 22, ''), ('finame', 21, '=')):
width=60
bar=mul*'='
withself.subTest(file=file, bar=bar):
file=fileor'Shell'
line=pyshell.restart_line(width, file)
eq(len(line), width)
eq(line, f"{bar+extra} RESTART: {file}{bar}")

deftest_restart_line_narrow(self):
expect, taglen="= RESTART: Shell", 16
forwidthin (taglen-1, taglen, taglen+1):
withself.subTest(width=width):
self.assertEqual(pyshell.restart_line(width, ''), expect)
self.assertEqual(pyshell.restart_line(taglen+2, ''), expect+' =')


classPyShellFileListTest(unittest.TestCase):

@classmethod
defsetUpClass(cls):
requires('gui')
cls.root=Tk()
cls.root.withdraw()

@classmethod
deftearDownClass(cls):
#cls.root.update_idletasks()
## for id in cls.root.tk.call('after', 'info'):
## cls.root.after_cancel(id) # Need for EditorWindow.
cls.root.destroy()
delcls.root

deftest_init(self):
psfl=pyshell.PyShellFileList(self.root)
self.assertEqual(psfl.EditorWindow, pyshell.PyShellEditorWindow)
self.assertIsNone(psfl.pyshell)

# The following sometimes causes 'invalid command name "109734456recolorize"'.
# Uncommenting after_cancel above prevents this, but results in
# TclError: bad window path name ".!listedtoplevel.!frame.text"
# which is normally prevented by after_cancel.
## def test_openshell(self):
## pyshell.use_subprocess = False
## ps = pyshell.PyShellFileList(self.root).open_shell()
## self.assertIsInstance(ps, pyshell.PyShell)


classPyShellRemoveLastNewlineAndSurroundingWhitespaceTest(unittest.TestCase):
regexp=pyshell.PyShell._last_newline_re

defall_removed(self, text):
self.assertEqual('', self.regexp.sub('', text))

defnone_removed(self, text):
self.assertEqual(text, self.regexp.sub('', text))

defcheck_result(self, text, expected):
self.assertEqual(expected, self.regexp.sub('', text))

deftest_empty(self):
self.all_removed('')

deftest_newline(self):
self.all_removed('\n')

deftest_whitespace_no_newline(self):
self.all_removed(' ')
self.all_removed(' ')
self.all_removed(' ')
self.all_removed(' '*20)
self.all_removed('\t')
self.all_removed('\t\t')
self.all_removed('\t\t\t')
self.all_removed('\t'*20)
self.all_removed('\t ')
self.all_removed(' \t')
self.all_removed(' \t\t ')
self.all_removed('\t\t\t')

deftest_newline_with_whitespace(self):
self.all_removed(' \n')
self.all_removed('\t\n')
self.all_removed(' \t\n')
self.all_removed('\t\n')
self.all_removed('\n ')
self.all_removed('\n\t')
self.all_removed('\n\t')
self.all_removed('\n\t ')
self.all_removed(' \n ')
self.all_removed('\t\n ')
self.all_removed(' \n\t')
self.all_removed('\t\n\t')
self.all_removed('\t\t\t\n')
self.all_removed(' \t\t\n')
self.all_removed('\n\t\t\t')
self.all_removed('\n\t\t ')

deftest_multiple_newlines(self):
self.check_result('\n\n', '\n')
self.check_result('\n'*5, '\n'*4)
self.check_result('\n'*5+'\t', '\n'*4)
self.check_result('\n'*20, '\n'*19)
self.check_result('\n'*20+' ', '\n'*19)
self.check_result(' \n\n ', ' \n')
self.check_result(' \n\n ', ' \n')
self.check_result(' \n\n', ' \n')
self.check_result('\t\n\n', '\t\n')
self.check_result('\n\n ', '\n')
self.check_result('\n\n\t', '\n')
self.check_result(' \n\n ', ' \n')
self.check_result('\t\n\t\n\t', '\t\n')

deftest_non_whitespace(self):
self.none_removed('a')
self.check_result('a\n', 'a')
self.check_result('a\n ', 'a')
self.check_result('a \n ', 'a')
self.check_result('a \n\t', 'a')
self.none_removed('-')
self.check_result('-\n', '-')
self.none_removed('.')
self.check_result('.\n', '.')

deftest_unsupported_whitespace(self):
self.none_removed('\v')
self.none_removed('\n\v')
self.check_result('\v\n', '\v')
self.none_removed(' \n\v')
self.check_result('\v\n ', '\v')


if__name__=='__main__':
unittest.main(verbosity=2)