Skip to content
Snippets Groups Projects
test_hyperparser.py 5.55 KiB
Newer Older
  • Learn to ignore specific revisions
  • """Unittest for idlelib.HyperParser"""
    import unittest
    from test.test_support import requires
    from Tkinter import Tk, Text
    from idlelib.EditorWindow 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.im_func
        is_char_in_string = EditorWindow.is_char_in_string.im_func
    
    
    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'
                )
    
        @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())
    
        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('11.3')
            self.assertEqual(p.get_expression(), '')
    
            p = get('11.11')
            self.assertEqual(p.get_expression(), 'False')
    
    
    if __name__ == '__main__':
        unittest.main(verbosity=2)