test/textwrap_test.py

   1 """Test the textwrap module"""
   2 import pytest
   3
   4 from cola import textwrap
   5
   6
   7 class WordWrapDefaults:
   8     def __init__(self):
   9         self.tabwidth = 8
  10         self.limit = None
  11
  12     def wrap(self, text, break_on_hyphens=True):
  13         return textwrap.word_wrap(
  14             text, self.tabwidth, self.limit, break_on_hyphens=break_on_hyphens
  15         )
  16
  17
  18 @pytest.fixture
  19 def wordwrap():
  20     """Provide default word wrap options for tests"""
  21     return WordWrapDefaults()
  22
  23
  24 def test_word_wrap(wordwrap):
  25     wordwrap.limit = 16
  26     text = """
  27 12345678901 3 56 8 01 3 5 7
  28
  29 1 3 5"""
  30     expect = """
  31 12345678901 3 56
  32 8 01 3 5 7
  33
  34 1 3 5"""
  35     assert expect == wordwrap.wrap(text)
  36
  37
  38 def test_word_wrap_dashes(wordwrap):
  39     wordwrap.limit = 4
  40     text = '123-5'
  41     expect = '123-5'
  42     assert expect == wordwrap.wrap(text)
  43
  44
  45 def test_word_wrap_leading_spaces(wordwrap):
  46     wordwrap.limit = 4
  47     expect = '1234\n5'
  48
  49     assert expect == wordwrap.wrap('1234 5')
  50     assert expect == wordwrap.wrap('1234  5')
  51     assert expect == wordwrap.wrap('1234   5')
  52     assert expect == wordwrap.wrap('1234    5')
  53     assert expect == wordwrap.wrap('1234     5')
  54
  55     expect = '123\n4'
  56     assert expect == wordwrap.wrap('123 4')
  57     assert expect == wordwrap.wrap('123  4')
  58     assert expect == wordwrap.wrap('123   4')
  59     assert expect == wordwrap.wrap('123    4')
  60     assert expect == wordwrap.wrap('123     4')
  61
  62
  63 def test_word_wrap_double_dashes(wordwrap):
  64     wordwrap.limit = 4
  65     text = '12--5'
  66     expect = '12--\n5'
  67     actual = wordwrap.wrap(text, break_on_hyphens=True)
  68     assert expect == actual
  69
  70     expect = '12--5'
  71     actual = wordwrap.wrap(text, break_on_hyphens=False)
  72     assert expect == actual
  73
  74
  75 def test_word_wrap_many_lines(wordwrap):
  76     wordwrap.limit = 2
  77     text = """
  78 aa
  79
  80
  81 bb cc dd"""
  82
  83     expect = """
  84 aa
  85
  86
  87 bb
  88 cc
  89 dd"""
  90     actual = wordwrap.wrap(text)
  91     assert expect == actual
  92
  93
  94 def test_word_python_code(wordwrap):
  95     wordwrap.limit = 78
  96     text = """
  97 if True:
  98     print "hello world"
  99 else:
 100     print "hello world"
 101
 102 """
 103     expect = text
 104     actual = wordwrap.wrap(text)
 105     assert expect == actual
 106
 107
 108 def test_word_wrap_spaces(wordwrap):
 109     wordwrap.limit = 2
 110     text = ' ' * 6
 111     expect = ''
 112     actual = wordwrap.wrap(text)
 113     assert expect == actual
 114
 115
 116 def test_word_wrap_special_tag(wordwrap):
 117     wordwrap.limit = 2
 118     text = """
 119 This test is so meta, even this sentence
 120
 121 Cheered-on-by: Avoids word-wrap
 122 C.f. This also avoids word-wrap
 123 References: This also avoids word-wrap
 124 See-also: This also avoids word-wrap
 125 Related-to: This also avoids word-wrap
 126 Link: This also avoids word-wrap
 127 """
 128
 129     expect = """
 130 This
 131 test
 132 is
 133 so
 134 meta,
 135 even
 136 this
 137 sentence
 138
 139 Cheered-on-by: Avoids word-wrap
 140 C.f. This also avoids word-wrap
 141 References: This also avoids word-wrap
 142 See-also: This also avoids word-wrap
 143 Related-to: This also avoids word-wrap
 144 Link: This also avoids word-wrap
 145 """
 146     actual = wordwrap.wrap(text)
 147     assert expect == actual
 148
 149
 150 def test_word_wrap_space_at_start_of_wrap(wordwrap):
 151     inputs = """0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 """
 152     expect = """0 1 2 3 4 5 6 7 8 9\n0 1 2 3 4 5 6 7 8"""
 153     wordwrap.limit = 20
 154
 155     actual = wordwrap.wrap(inputs)
 156     assert expect == actual
 157
 158
 159 def test_word_wrap_keeps_tabs_at_start(wordwrap):
 160     inputs = """\tfirst line\n\n\tsecond line"""
 161     expect = """\tfirst line\n\n\tsecond line"""
 162     wordwrap.limit = 20
 163
 164     actual = wordwrap.wrap(inputs)
 165     assert expect == actual
 166
 167
 168 def test_word_wrap_keeps_twospace_indents(wordwrap):
 169     inputs = """first line\n\n* branch:\n  line1\n  line2\n"""
 170     expect = """first line\n\n* branch:\n  line1\n  line2\n"""
 171     wordwrap.limit = 20
 172
 173     actual = wordwrap.wrap(inputs)
 174     assert expect == actual
 175
 176
 177 def test_word_wrap_ranges():
 178     text = 'a bb ccc dddd\neeeee'
 179     expect = 'a\nbb\nccc\ndddd\neeeee'
 180     actual = textwrap.word_wrap(text, 8, 2)
 181     assert expect == actual
 182
 183     expect = 'a bb\nccc\ndddd\neeeee'
 184     actual = textwrap.word_wrap(text, 8, 4)
 185     assert expect == actual
 186
 187     text = 'a bb ccc dddd\n\teeeee'
 188     expect = 'a bb\nccc\ndddd\n\t\neeeee'
 189     actual = textwrap.word_wrap(text, 8, 4)
 190     assert expect == actual
 191
 192
 193 def test_triplets():
 194     text = 'xx0 xx1 xx2 xx3 xx4 xx5 xx6 xx7 xx8 xx9 xxa xxb'
 195
 196     expect = 'xx0 xx1 xx2 xx3 xx4 xx5 xx6\nxx7 xx8 xx9 xxa xxb'
 197     actual = textwrap.word_wrap(text, 8, 27)
 198     assert expect == actual
 199
 200     expect = 'xx0 xx1 xx2 xx3 xx4 xx5\nxx6 xx7 xx8 xx9 xxa xxb'
 201     actual = textwrap.word_wrap(text, 8, 26)
 202     assert expect == actual
 203
 204     actual = textwrap.word_wrap(text, 8, 25)
 205     assert expect == actual
 206
 207     actual = textwrap.word_wrap(text, 8, 24)
 208     assert expect == actual
 209
 210     actual = textwrap.word_wrap(text, 8, 23)
 211     assert expect == actual
 212
 213     expect = 'xx0 xx1 xx2 xx3 xx4\nxx5 xx6 xx7 xx8 xx9\nxxa xxb'
 214     actual = textwrap.word_wrap(text, 8, 22)
 215     assert expect == actual