KevinMidboe/tableprint
1
0
Fork 0
mirror of https://github.com/KevinMidboe/tableprint.git synced 2025-10-29 18:00:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adds a small test for the hr() function

Browse Source
This commit is contained in:
Niru Maheswaranathan
2016-10-07 11:50:50 -07:00
parent 70c533ba2c
commit a44d0dafa1
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_io.py
View File

@@ -1,5 +1,5 @@
from __future__ import unicode_literals from __future__ import unicode_literals
from tableprint import table, banner, dataframe from tableprint import table, banner, dataframe, hr
from io import StringIO from io import StringIO
import numpy as np import numpy as np
@@ -7,7 +7,7 @@ import numpy as np
def test_table(): def test_table():
output = StringIO() output = StringIO()
table([[1, 2, 3], [4, 5 ,6]], "ABC", style='round', width=5, out=output) table([[1, 2, 3], [4, 5, 6]], 'ABC', style='round', width=5, out=output)
assert output.getvalue() == '╭─────┬─────┬─────╮\n│ A │ B │ C │\n├─────┼─────┼─────┤\n│ 1│ 2│ 3│\n│ 4│ 5│ 6│\n╰─────┴─────┴─────╯\n' assert output.getvalue() == '╭─────┬─────┬─────╮\n│ A │ B │ C │\n├─────┼─────┼─────┤\n│ 1│ 2│ 3│\n│ 4│ 5│ 6│\n╰─────┴─────┴─────╯\n'
output = StringIO() output = StringIO()
@@ -41,3 +41,10 @@ def test_banner():
output = StringIO() output = StringIO()
banner('!', style='banner', width=1, out=output) banner('!', style='banner', width=1, out=output)
assert output.getvalue() == '╒═╕\n│!│\n╘═╛\n' assert output.getvalue() == '╒═╕\n│!│\n╘═╛\n'
def test_hr():
output = hr(1, width=11)
assert len(output) == 11
assert '───────────'