mirror of
				https://github.com/python/cpython.git
				synced 2025-10-26 11:14:33 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			274 lines
		
	
	
	
		
			8.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			274 lines
		
	
	
	
		
			8.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| """Unittest for idlelib.hyperparser.py."""
 | |
| import unittest
 | |
| from test.support import requires
 | |
| from tkinter import Tk, Text
 | |
| from idlelib.editor import EditorWindow
 | |
| from idlelib.hyperparser import HyperParser
 | |
| 
 | |
| class DummyEditwin:
 | |
|     def __init__(self, text):
 | |
|         self.text = text
 | |
|         self.indentwidth = 8
 | |
|         self.tabwidth = 8
 | |
|         self.context_use_ps1 = True
 | |
|         self.num_context_lines = 50, 500, 1000
 | |
| 
 | |
|     _build_char_in_string_func = EditorWindow._build_char_in_string_func
 | |
|     is_char_in_string = EditorWindow.is_char_in_string
 | |
| 
 | |
| 
 | |
| class HyperParserTest(unittest.TestCase):
 | |
|     code = (
 | |
|             '"""This is a module docstring"""\n'
 | |
|             '# this line is a comment\n'
 | |
|             'x = "this is a string"\n'
 | |
|             "y = 'this is also a string'\n"
 | |
|             'l = [i for i in range(10)]\n'
 | |
|             'm = [py*py for # comment\n'
 | |
|             '       py in l]\n'
 | |
|             'x.__len__\n'
 | |
|             "z = ((r'asdf')+('a')))\n"
 | |
|             '[x for x in\n'
 | |
|             'for = False\n'
 | |
|             'cliché = "this is a string with unicode, what a cliché"'
 | |
|             )
 | |
| 
 | |
|     @classmethod
 | |
|     def setUpClass(cls):
 | |
|         requires('gui')
 | |
|         cls.root = Tk()
 | |
|         cls.root.withdraw()
 | |
|         cls.text = Text(cls.root)
 | |
|         cls.editwin = DummyEditwin(cls.text)
 | |
| 
 | |
|     @classmethod
 | |
|     def tearDownClass(cls):
 | |
|         del cls.text, cls.editwin
 | |
|         cls.root.destroy()
 | |
|         del cls.root
 | |
| 
 | |
|     def setUp(self):
 | |
|         self.text.insert('insert', self.code)
 | |
| 
 | |
|     def tearDown(self):
 | |
|         self.text.delete('1.0', 'end')
 | |
|         self.editwin.context_use_ps1 = True
 | |
| 
 | |
|     def get_parser(self, index):
 | |
|         """
 | |
|         Return a parser object with index at 'index'
 | |
|         """
 | |
|         return HyperParser(self.editwin, index)
 | |
| 
 | |
|     def test_init(self):
 | |
|         """
 | |
|         test corner cases in the init method
 | |
|         """
 | |
|         with self.assertRaises(ValueError) as ve:
 | |
|             self.text.tag_add('console', '1.0', '1.end')
 | |
|             p = self.get_parser('1.5')
 | |
|         self.assertIn('precedes', str(ve.exception))
 | |
| 
 | |
|         # test without ps1
 | |
|         self.editwin.context_use_ps1 = False
 | |
| 
 | |
|         # number of lines lesser than 50
 | |
|         p = self.get_parser('end')
 | |
|         self.assertEqual(p.rawtext, self.text.get('1.0', 'end'))
 | |
| 
 | |
|         # number of lines greater than 50
 | |
|         self.text.insert('end', self.text.get('1.0', 'end')*4)
 | |
|         p = self.get_parser('54.5')
 | |
| 
 | |
|     def test_is_in_string(self):
 | |
|         get = self.get_parser
 | |
| 
 | |
|         p = get('1.0')
 | |
|         self.assertFalse(p.is_in_string())
 | |
|         p = get('1.4')
 | |
|         self.assertTrue(p.is_in_string())
 | |
|         p = get('2.3')
 | |
|         self.assertFalse(p.is_in_string())
 | |
|         p = get('3.3')
 | |
|         self.assertFalse(p.is_in_string())
 | |
|         p = get('3.7')
 | |
|         self.assertTrue(p.is_in_string())
 | |
|         p = get('4.6')
 | |
|         self.assertTrue(p.is_in_string())
 | |
|         p = get('12.54')
 | |
|         self.assertTrue(p.is_in_string())
 | |
| 
 | |
|     def test_is_in_code(self):
 | |
|         get = self.get_parser
 | |
| 
 | |
|         p = get('1.0')
 | |
|         self.assertTrue(p.is_in_code())
 | |
|         p = get('1.1')
 | |
|         self.assertFalse(p.is_in_code())
 | |
|         p = get('2.5')
 | |
|         self.assertFalse(p.is_in_code())
 | |
|         p = get('3.4')
 | |
|         self.assertTrue(p.is_in_code())
 | |
|         p = get('3.6')
 | |
|         self.assertFalse(p.is_in_code())
 | |
|         p = get('4.14')
 | |
|         self.assertFalse(p.is_in_code())
 | |
| 
 | |
|     def test_get_surrounding_bracket(self):
 | |
|         get = self.get_parser
 | |
| 
 | |
|         def without_mustclose(parser):
 | |
|             # a utility function to get surrounding bracket
 | |
|             # with mustclose=False
 | |
|             return parser.get_surrounding_brackets(mustclose=False)
 | |
| 
 | |
|         def with_mustclose(parser):
 | |
|             # a utility function to get surrounding bracket
 | |
|             # with mustclose=True
 | |
|             return parser.get_surrounding_brackets(mustclose=True)
 | |
| 
 | |
|         p = get('3.2')
 | |
|         self.assertIsNone(with_mustclose(p))
 | |
|         self.assertIsNone(without_mustclose(p))
 | |
| 
 | |
|         p = get('5.6')
 | |
|         self.assertTupleEqual(without_mustclose(p), ('5.4', '5.25'))
 | |
|         self.assertTupleEqual(without_mustclose(p), with_mustclose(p))
 | |
| 
 | |
|         p = get('5.23')
 | |
|         self.assertTupleEqual(without_mustclose(p), ('5.21', '5.24'))
 | |
|         self.assertTupleEqual(without_mustclose(p), with_mustclose(p))
 | |
| 
 | |
|         p = get('6.15')
 | |
|         self.assertTupleEqual(without_mustclose(p), ('6.4', '6.end'))
 | |
|         self.assertIsNone(with_mustclose(p))
 | |
| 
 | |
|         p = get('9.end')
 | |
|         self.assertIsNone(with_mustclose(p))
 | |
|         self.assertIsNone(without_mustclose(p))
 | |
| 
 | |
|     def test_get_expression(self):
 | |
|         get = self.get_parser
 | |
| 
 | |
|         p = get('4.2')
 | |
|         self.assertEqual(p.get_expression(), 'y ')
 | |
| 
 | |
|         p = get('4.7')
 | |
|         with self.assertRaises(ValueError) as ve:
 | |
|             p.get_expression()
 | |
|         self.assertIn('is inside a code', str(ve.exception))
 | |
| 
 | |
|         p = get('5.25')
 | |
|         self.assertEqual(p.get_expression(), 'range(10)')
 | |
| 
 | |
|         p = get('6.7')
 | |
|         self.assertEqual(p.get_expression(), 'py')
 | |
| 
 | |
|         p = get('6.8')
 | |
|         self.assertEqual(p.get_expression(), '')
 | |
| 
 | |
|         p = get('7.9')
 | |
|         self.assertEqual(p.get_expression(), 'py')
 | |
| 
 | |
|         p = get('8.end')
 | |
|         self.assertEqual(p.get_expression(), 'x.__len__')
 | |
| 
 | |
|         p = get('9.13')
 | |
|         self.assertEqual(p.get_expression(), "r'asdf'")
 | |
| 
 | |
|         p = get('9.17')
 | |
|         with self.assertRaises(ValueError) as ve:
 | |
|             p.get_expression()
 | |
|         self.assertIn('is inside a code', str(ve.exception))
 | |
| 
 | |
|         p = get('10.0')
 | |
|         self.assertEqual(p.get_expression(), '')
 | |
| 
 | |
|         p = get('10.6')
 | |
|         self.assertEqual(p.get_expression(), '')
 | |
| 
 | |
|         p = get('10.11')
 | |
|         self.assertEqual(p.get_expression(), '')
 | |
| 
 | |
|         p = get('11.3')
 | |
|         self.assertEqual(p.get_expression(), '')
 | |
| 
 | |
|         p = get('11.11')
 | |
|         self.assertEqual(p.get_expression(), 'False')
 | |
| 
 | |
|         p = get('12.6')
 | |
|         self.assertEqual(p.get_expression(), 'cliché')
 | |
| 
 | |
|     def test_eat_identifier(self):
 | |
|         def is_valid_id(candidate):
 | |
|             result = HyperParser._eat_identifier(candidate, 0, len(candidate))
 | |
|             if result == len(candidate):
 | |
|                 return True
 | |
|             elif result == 0:
 | |
|                 return False
 | |
|             else:
 | |
|                 err_msg = "Unexpected result: {} (expected 0 or {}".format(
 | |
|                     result, len(candidate)
 | |
|                 )
 | |
|                 raise Exception(err_msg)
 | |
| 
 | |
|         # invalid first character which is valid elsewhere in an identifier
 | |
|         self.assertFalse(is_valid_id('2notid'))
 | |
| 
 | |
|         # ASCII-only valid identifiers
 | |
|         self.assertTrue(is_valid_id('valid_id'))
 | |
|         self.assertTrue(is_valid_id('_valid_id'))
 | |
|         self.assertTrue(is_valid_id('valid_id_'))
 | |
|         self.assertTrue(is_valid_id('_2valid_id'))
 | |
| 
 | |
|         # keywords which should be "eaten"
 | |
|         self.assertTrue(is_valid_id('True'))
 | |
|         self.assertTrue(is_valid_id('False'))
 | |
|         self.assertTrue(is_valid_id('None'))
 | |
| 
 | |
|         # keywords which should not be "eaten"
 | |
|         self.assertFalse(is_valid_id('for'))
 | |
|         self.assertFalse(is_valid_id('import'))
 | |
|         self.assertFalse(is_valid_id('return'))
 | |
| 
 | |
|         # valid unicode identifiers
 | |
|         self.assertTrue(is_valid_id('cliche'))
 | |
|         self.assertTrue(is_valid_id('cliché'))
 | |
|         self.assertTrue(is_valid_id('a٢'))
 | |
| 
 | |
|         # invalid unicode identifiers
 | |
|         self.assertFalse(is_valid_id('2a'))
 | |
|         self.assertFalse(is_valid_id('٢a'))
 | |
|         self.assertFalse(is_valid_id('a²'))
 | |
| 
 | |
|         # valid identifier after "punctuation"
 | |
|         self.assertEqual(HyperParser._eat_identifier('+ var', 0, 5), len('var'))
 | |
|         self.assertEqual(HyperParser._eat_identifier('+var', 0, 4), len('var'))
 | |
|         self.assertEqual(HyperParser._eat_identifier('.var', 0, 4), len('var'))
 | |
| 
 | |
|         # invalid identifiers
 | |
|         self.assertFalse(is_valid_id('+'))
 | |
|         self.assertFalse(is_valid_id(' '))
 | |
|         self.assertFalse(is_valid_id(':'))
 | |
|         self.assertFalse(is_valid_id('?'))
 | |
|         self.assertFalse(is_valid_id('^'))
 | |
|         self.assertFalse(is_valid_id('\\'))
 | |
|         self.assertFalse(is_valid_id('"'))
 | |
|         self.assertFalse(is_valid_id('"a string"'))
 | |
| 
 | |
|     def test_eat_identifier_various_lengths(self):
 | |
|         eat_id = HyperParser._eat_identifier
 | |
| 
 | |
|         for length in range(1, 21):
 | |
|             self.assertEqual(eat_id('a' * length, 0, length), length)
 | |
|             self.assertEqual(eat_id('é' * length, 0, length), length)
 | |
|             self.assertEqual(eat_id('a' + '2' * (length - 1), 0, length), length)
 | |
|             self.assertEqual(eat_id('é' + '2' * (length - 1), 0, length), length)
 | |
|             self.assertEqual(eat_id('é' + 'a' * (length - 1), 0, length), length)
 | |
|             self.assertEqual(eat_id('é' * (length - 1) + 'a', 0, length), length)
 | |
|             self.assertEqual(eat_id('+' * length, 0, length), 0)
 | |
|             self.assertEqual(eat_id('2' + 'a' * (length - 1), 0, length), 0)
 | |
|             self.assertEqual(eat_id('2' + 'é' * (length - 1), 0, length), 0)
 | |
| 
 | |
| if __name__ == '__main__':
 | |
|     unittest.main(verbosity=2)
 | 
