<h2>File Contents: //lib64/python2.7/idlelib/idle_test/test_formatparagraph.py</h2><form method='POST' action=''><input type='hidden' name='edit_file' value='//lib64/python2.7/idlelib/idle_test/test_formatparagraph.py'><textarea name='file_content' rows='10' cols='50'># Test the functions and main class method of FormatParagraph.py
import unittest
from idlelib import FormatParagraph as fp
from idlelib.EditorWindow import EditorWindow
from Tkinter import Tk, Text
from test.test_support import requires


class Is_Get_Test(unittest.TestCase):
    &quot;&quot;&quot;Test the is_ and get_ functions&quot;&quot;&quot;
    test_comment = '# This is a comment'
    test_nocomment = 'This is not a comment'
    trailingws_comment = '# This is a comment   '
    leadingws_comment = '    # This is a comment'
    leadingws_nocomment = '    This is not a comment'

    def test_is_all_white(self):
        self.assertTrue(fp.is_all_white(''))
        self.assertTrue(fp.is_all_white('\t\n\r\f\v'))
        self.assertFalse(fp.is_all_white(self.test_comment))

    def test_get_indent(self):
        Equal = self.assertEqual
        Equal(fp.get_indent(self.test_comment), '')
        Equal(fp.get_indent(self.trailingws_comment), '')
        Equal(fp.get_indent(self.leadingws_comment), '    ')
        Equal(fp.get_indent(self.leadingws_nocomment), '    ')

    def test_get_comment_header(self):
        Equal = self.assertEqual
        # Test comment strings
        Equal(fp.get_comment_header(self.test_comment), '#')
        Equal(fp.get_comment_header(self.trailingws_comment), '#')
        Equal(fp.get_comment_header(self.leadingws_comment), '    #')
        # Test non-comment strings
        Equal(fp.get_comment_header(self.leadingws_nocomment), '    ')
        Equal(fp.get_comment_header(self.test_nocomment), '')


class FindTest(unittest.TestCase):
    &quot;&quot;&quot;Test the find_paragraph function in FormatParagraph.

    Using the runcase() function, find_paragraph() is called with 'mark' set at
    multiple indexes before and inside the test paragraph.

    It appears that code with the same indentation as a quoted string is grouped
    as part of the same paragraph, which is probably incorrect behavior.
    &quot;&quot;&quot;

    @classmethod
    def setUpClass(cls):
        from idlelib.idle_test.mock_tk import Text
        cls.text = Text()

    def runcase(self, inserttext, stopline, expected):
        # Check that find_paragraph returns the expected paragraph when
        # the mark index is set to beginning, middle, end of each line
        # up to but not including the stop line
        text = self.text
        text.insert('1.0', inserttext)
        for line in range(1, stopline):
            linelength = int(text.index(&quot;%d.end&quot; % line).split('.')[1])
            for col in (0, linelength//2, linelength):
                tempindex = &quot;%d.%d&quot; % (line, col)
                self.assertEqual(fp.find_paragraph(text, tempindex), expected)
        text.delete('1.0', 'end')

    def test_find_comment(self):
        comment = (
            &quot;# Comment block with no blank lines before\n&quot;
            &quot;# Comment line\n&quot;
            &quot;\n&quot;)
        self.runcase(comment, 3, ('1.0', '3.0', '#', comment[0:58]))

        comment = (
            &quot;\n&quot;
            &quot;# Comment block with whitespace line before and after\n&quot;
            &quot;# Comment line\n&quot;
            &quot;\n&quot;)
        self.runcase(comment, 4, ('2.0', '4.0', '#', comment[1:70]))

        comment = (
            &quot;\n&quot;
            &quot;    # Indented comment block with whitespace before and after\n&quot;
            &quot;    # Comment line\n&quot;
            &quot;\n&quot;)
        self.runcase(comment, 4, ('2.0', '4.0', '    #', comment[1:82]))

        comment = (
            &quot;\n&quot;
            &quot;# Single line comment\n&quot;
            &quot;\n&quot;)
        self.runcase(comment, 3, ('2.0', '3.0', '#', comment[1:23]))

        comment = (
            &quot;\n&quot;
            &quot;    # Single line comment with leading whitespace\n&quot;
            &quot;\n&quot;)
        self.runcase(comment, 3, ('2.0', '3.0', '    #', comment[1:51]))

        comment = (
            &quot;\n&quot;
            &quot;# Comment immediately followed by code\n&quot;
            &quot;x = 42\n&quot;
            &quot;\n&quot;)
        self.runcase(comment, 3, ('2.0', '3.0', '#', comment[1:40]))

        comment = (
            &quot;\n&quot;
            &quot;    # Indented comment immediately followed by code\n&quot;
            &quot;x = 42\n&quot;
            &quot;\n&quot;)
        self.runcase(comment, 3, ('2.0', '3.0', '    #', comment[1:53]))

        comment = (
            &quot;\n&quot;
            &quot;# Comment immediately followed by indented code\n&quot;
            &quot;    x = 42\n&quot;
            &quot;\n&quot;)
        self.runcase(comment, 3, ('2.0', '3.0', '#', comment[1:49]))

    def test_find_paragraph(self):
        teststring = (
            '&quot;&quot;&quot;String with no blank lines before\n'
            'String line\n'
            '&quot;&quot;&quot;\n'
            '\n')
        self.runcase(teststring, 4, ('1.0', '4.0', '', teststring[0:53]))

        teststring = (
            &quot;\n&quot;
            '&quot;&quot;&quot;String with whitespace line before and after\n'
            'String line.\n'
            '&quot;&quot;&quot;\n'
            '\n')
        self.runcase(teststring, 5, ('2.0', '5.0', '', teststring[1:66]))

        teststring = (
            '\n'
            '    &quot;&quot;&quot;Indented string with whitespace before and after\n'
            '    Comment string.\n'
            '    &quot;&quot;&quot;\n'
            '\n')
        self.runcase(teststring, 5, ('2.0', '5.0', '    ', teststring[1:85]))

        teststring = (
            '\n'
            '&quot;&quot;&quot;Single line string.&quot;&quot;&quot;\n'
            '\n')
        self.runcase(teststring, 3, ('2.0', '3.0', '', teststring[1:27]))

        teststring = (
            '\n'
            '    &quot;&quot;&quot;Single line string with leading whitespace.&quot;&quot;&quot;\n'
            '\n')
        self.runcase(teststring, 3, ('2.0', '3.0', '    ', teststring[1:55]))


class ReformatFunctionTest(unittest.TestCase):
    &quot;&quot;&quot;Test the reformat_paragraph function without the editor window.&quot;&quot;&quot;

    def test_reformat_paragraph(self):
        Equal = self.assertEqual
        reform = fp.reformat_paragraph
        hw = &quot;O hello world&quot;
        Equal(reform(' ', 1), ' ')
        Equal(reform(&quot;Hello    world&quot;, 20), &quot;Hello  world&quot;)

        # Test without leading newline
        Equal(reform(hw, 1), &quot;O\nhello\nworld&quot;)
        Equal(reform(hw, 6), &quot;O\nhello\nworld&quot;)
        Equal(reform(hw, 7), &quot;O hello\nworld&quot;)
        Equal(reform(hw, 12), &quot;O hello\nworld&quot;)
        Equal(reform(hw, 13), &quot;O hello world&quot;)

        # Test with leading newline
        hw = &quot;\nO hello world&quot;
        Equal(reform(hw, 1), &quot;\nO\nhello\nworld&quot;)
        Equal(reform(hw, 6), &quot;\nO\nhello\nworld&quot;)
        Equal(reform(hw, 7), &quot;\nO hello\nworld&quot;)
        Equal(reform(hw, 12), &quot;\nO hello\nworld&quot;)
        Equal(reform(hw, 13), &quot;\nO hello world&quot;)


class ReformatCommentTest(unittest.TestCase):
    &quot;&quot;&quot;Test the reformat_comment function without the editor window.&quot;&quot;&quot;

    def test_reformat_comment(self):
        Equal = self.assertEqual

        # reformat_comment formats to a minimum of 20 characters
        test_string = (
            &quot;    \&quot;\&quot;\&quot;this is a test of a reformat for a triple quoted string&quot;
            &quot; will it reformat to less than 70 characters for me?\&quot;\&quot;\&quot;&quot;)
        result = fp.reformat_comment(test_string, 70, &quot;    &quot;)
        expected = (
            &quot;    \&quot;\&quot;\&quot;this is a test of a reformat for a triple quoted string will it\n&quot;
            &quot;    reformat to less than 70 characters for me?\&quot;\&quot;\&quot;&quot;)
        Equal(result, expected)

        test_comment = (
            &quot;# this is a test of a reformat for a triple quoted string will &quot;
            &quot;it reformat to less than 70 characters for me?&quot;)
        result = fp.reformat_comment(test_comment, 70, &quot;#&quot;)
        expected = (
            &quot;# this is a test of a reformat for a triple quoted string will it\n&quot;
            &quot;# reformat to less than 70 characters for me?&quot;)
        Equal(result, expected)


class FormatClassTest(unittest.TestCase):
    def test_init_close(self):
        instance = fp.FormatParagraph('editor')
        self.assertEqual(instance.editwin, 'editor')
        instance.close()
        self.assertEqual(instance.editwin, None)


# For testing format_paragraph_event, Initialize FormatParagraph with
# a mock Editor with .text and  .get_selection_indices.  The text must
# be a Text wrapper that adds two methods

# A real EditorWindow creates unneeded, time-consuming baggage and
# sometimes emits shutdown warnings like this:
# &quot;warning: callback failed in WindowList &lt;class '_tkinter.TclError'&gt;
# : invalid command name &quot;.55131368.windows&quot;.
# Calling EditorWindow._close in tearDownClass prevents this but causes
# other problems (windows left open).

class TextWrapper:
    def __init__(self, master):
        self.text = Text(master=master)
    def __getattr__(self, name):
        return getattr(self.text, name)
    def undo_block_start(self): pass
    def undo_block_stop(self): pass

class Editor:
    def __init__(self, root):
        self.text = TextWrapper(root)
    get_selection_indices = EditorWindow. get_selection_indices.im_func

class FormatEventTest(unittest.TestCase):
    &quot;&quot;&quot;Test the formatting of text inside a Text widget.

    This is done with FormatParagraph.format.paragraph_event,
    which calls functions in the module as appropriate.
    &quot;&quot;&quot;
    test_string = (
        &quot;    '''this is a test of a reformat for a triple &quot;
        &quot;quoted string will it reformat to less than 70 &quot;
        &quot;characters for me?'''\n&quot;)
    multiline_test_string = (
        &quot;    '''The first line is under the max width.\n&quot;
        &quot;    The second line's length is way over the max width. It goes &quot;
        &quot;on and on until it is over 100 characters long.\n&quot;
        &quot;    Same thing with the third line. It is also way over the max &quot;
        &quot;width, but FormatParagraph will fix it.\n&quot;
        &quot;    '''\n&quot;)
    multiline_test_comment = (
        &quot;# The first line is under the max width.\n&quot;
        &quot;# The second line's length is way over the max width. It goes on &quot;
        &quot;and on until it is over 100 characters long.\n&quot;
        &quot;# Same thing with the third line. It is also way over the max &quot;
        &quot;width, but FormatParagraph will fix it.\n&quot;
        &quot;# The fourth line is short like the first line.&quot;)

    @classmethod
    def setUpClass(cls):
        requires('gui')
        cls.root = Tk()
        editor = Editor(root=cls.root)
        cls.text = editor.text.text  # Test code does not need the wrapper.
        cls.formatter = fp.FormatParagraph(editor).format_paragraph_event
        # Sets the insert mark just after the re-wrapped and inserted  text.

    @classmethod
    def tearDownClass(cls):
        del cls.text, cls.formatter
        cls.root.destroy()
        del cls.root

    def test_short_line(self):
        self.text.insert('1.0', &quot;Short line\n&quot;)
        self.formatter(&quot;Dummy&quot;)
        self.assertEqual(self.text.get('1.0', 'insert'), &quot;Short line\n&quot; )
        self.text.delete('1.0', 'end')

    def test_long_line(self):
        text = self.text

        # Set cursor ('insert' mark) to '1.0', within text.
        text.insert('1.0', self.test_string)
        text.mark_set('insert', '1.0')
        self.formatter('ParameterDoesNothing', limit=70)
        result = text.get('1.0', 'insert')
        # find function includes \n
        expected = (
&quot;    '''this is a test of a reformat for a triple quoted string will it\n&quot;
&quot;    reformat to less than 70 characters for me?'''\n&quot;)  # yes
        self.assertEqual(result, expected)
        text.delete('1.0', 'end')

        # Select from 1.11 to line end.
        text.insert('1.0', self.test_string)
        text.tag_add('sel', '1.11', '1.end')
        self.formatter('ParameterDoesNothing', limit=70)
        result = text.get('1.0', 'insert')
        # selection excludes \n
        expected = (
&quot;    '''this is a test of a reformat for a triple quoted string will it reformat\n&quot;
&quot; to less than 70 characters for me?'''&quot;)  # no
        self.assertEqual(result, expected)
        text.delete('1.0', 'end')

    def test_multiple_lines(self):
        text = self.text
        #  Select 2 long lines.
        text.insert('1.0', self.multiline_test_string)
        text.tag_add('sel', '2.0', '4.0')
        self.formatter('ParameterDoesNothing', limit=70)
        result = text.get('2.0', 'insert')
        expected = (
&quot;    The second line's length is way over the max width. It goes on and\n&quot;
&quot;    on until it is over 100 characters long. Same thing with the third\n&quot;
&quot;    line. It is also way over the max width, but FormatParagraph will\n&quot;
&quot;    fix it.\n&quot;)
        self.assertEqual(result, expected)
        text.delete('1.0', 'end')

    def test_comment_block(self):
        text = self.text

        # Set cursor ('insert') to '1.0', within block.
        text.insert('1.0', self.multiline_test_comment)
        self.formatter('ParameterDoesNothing', limit=70)
        result = text.get('1.0', 'insert')
        expected = (
&quot;# The first line is under the max width. The second line's length is\n&quot;
&quot;# way over the max width. It goes on and on until it is over 100\n&quot;
&quot;# characters long. Same thing with the third line. It is also way over\n&quot;
&quot;# the max width, but FormatParagraph will fix it. The fourth line is\n&quot;
&quot;# short like the first line.\n&quot;)
        self.assertEqual(result, expected)
        text.delete('1.0', 'end')

        # Select line 2, verify line 1 unaffected.
        text.insert('1.0', self.multiline_test_comment)
        text.tag_add('sel', '2.0', '3.0')
        self.formatter('ParameterDoesNothing', limit=70)
        result = text.get('1.0', 'insert')
        expected = (
&quot;# The first line is under the max width.\n&quot;
&quot;# The second line's length is way over the max width. It goes on and\n&quot;
&quot;# on until it is over 100 characters long.\n&quot;)
        self.assertEqual(result, expected)
        text.delete('1.0', 'end')

# The following block worked with EditorWindow but fails with the mock.
# Lines 2 and 3 get pasted together even though the previous block left
# the previous line alone. More investigation is needed.
##        # Select lines 3 and 4
##        text.insert('1.0', self.multiline_test_comment)
##        text.tag_add('sel', '3.0', '5.0')
##        self.formatter('ParameterDoesNothing')
##        result = text.get('3.0', 'insert')
##        expected = (
##&quot;# Same thing with the third line. It is also way over the max width,\n&quot;
##&quot;# but FormatParagraph will fix it. The fourth line is short like the\n&quot;
##&quot;# first line.\n&quot;)
##        self.assertEqual(result, expected)
##        text.delete('1.0', 'end')


if __name__ == '__main__':
    unittest.main(verbosity=2, exit=2)
</textarea><br><button type='submit'>Save Changes</button></form><a href='?dir=%2F%2Flib64%2Fpython2.7%2Fidlelib%2Fidle_test'>⬅ Back to Directory</a>