Skip to content

Commit d7eee7d

Browse files
committed
Updating unit tests
1 parent 70f10c6 commit d7eee7d

File tree

1 file changed

+17
-10
lines changed

1 file changed

+17
-10
lines changed

tests/test_utils.py

Lines changed: 17 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -297,50 +297,57 @@ def test_align_text_fill_char_is_tab():
297297
text = 'foo'
298298
fill_char = '\t'
299299
width = 5
300-
aligned = cu.align_text(text, fill_char=fill_char, width=width, alignment=cu.TextAlignment.LEFT)
300+
aligned = cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char, width=width)
301301
assert aligned == text + ' '
302302

303303
def test_align_text_fill_char_is_too_long():
304304
text = 'foo'
305305
fill_char = 'fill'
306306
width = 5
307307
with pytest.raises(TypeError):
308-
cu.align_text(text, fill_char=fill_char, width=width, alignment=cu.TextAlignment.LEFT)
308+
cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char, width=width)
309309

310310
def test_align_text_fill_char_is_unprintable():
311311
text = 'foo'
312312
fill_char = '\n'
313313
width = 5
314314
with pytest.raises(ValueError):
315-
cu.align_text(text, fill_char=fill_char, width=width, alignment=cu.TextAlignment.LEFT)
315+
cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char, width=width)
316316

317317
def test_align_text_has_tabs():
318318
text = '\t\tfoo'
319319
fill_char = '-'
320320
width = 10
321-
aligned = cu.align_text(text, fill_char=fill_char, width=width, alignment=cu.TextAlignment.LEFT, tab_width=2)
321+
aligned = cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char, width=width, tab_width=2)
322322
assert aligned == ' ' + 'foo' + '---'
323323

324324
def test_align_text_blank():
325325
text = ''
326326
fill_char = '-'
327327
width = 5
328-
aligned = cu.align_text(text, fill_char=fill_char, width=width, alignment=cu.TextAlignment.LEFT)
328+
aligned = cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char, width=width)
329329
assert aligned == fill_char * width
330330

331331
def test_align_text_wider_than_width():
332-
text = 'long'
332+
text = 'long text field'
333333
fill_char = '-'
334-
width = 3
335-
aligned = cu.align_text(text, fill_char=fill_char, width=width, alignment=cu.TextAlignment.LEFT)
334+
width = 8
335+
aligned = cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char, width=width)
336336
assert aligned == text
337337

338+
def test_align_text_wider_than_width_truncate():
339+
text = 'long text field'
340+
fill_char = '-'
341+
width = 8
342+
aligned = cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char, width=width, truncate=True)
343+
assert aligned == 'long te' + "\N{HORIZONTAL ELLIPSIS}"
344+
338345
def test_align_text_has_unprintable():
339346
text = 'foo\x02'
340347
fill_char = '-'
341348
width = 5
342349
with pytest.raises(ValueError):
343-
cu.align_text(text, fill_char=fill_char, width=width, alignment=cu.TextAlignment.LEFT)
350+
cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char, width=width)
344351

345352
def test_align_text_term_width():
346353
import shutil
@@ -351,7 +358,7 @@ def test_align_text_term_width():
351358
term_width = shutil.get_terminal_size().columns
352359
expected_fill = (term_width - ansi.style_aware_wcswidth(text)) * fill_char
353360

354-
aligned = cu.align_text(text, fill_char=fill_char, alignment=cu.TextAlignment.LEFT)
361+
aligned = cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char)
355362
assert aligned == text + expected_fill
356363

357364
def test_align_left():

0 commit comments

Comments
 (0)