mirror of
https://github.com/KevinMidboe/tableprint.git
synced 2025-10-29 18:00:16 +00:00
62 lines
2.5 KiB
Python
62 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
from tableprint import table, banner, dataframe, hrule, TableContext
|
|
from io import StringIO
|
|
import numpy as np
|
|
|
|
|
|
def test_context():
|
|
"""Tests the table context manager"""
|
|
output = StringIO()
|
|
with TableContext('ABC', style='round', width=5, out=output) as t:
|
|
t([1, 2, 3])
|
|
t([4, 5, 6])
|
|
assert output.getvalue() == '╭─────┬─────┬─────╮\n│ A │ B │ C │\n├─────┼─────┼─────┤\n│ 1│ 2│ 3│\n│ 4│ 5│ 6│\n╰─────┴─────┴─────╯\n'
|
|
|
|
|
|
def test_table():
|
|
"""Tests the table function"""
|
|
output = StringIO()
|
|
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'
|
|
|
|
output = StringIO()
|
|
table(["bar"], "foo", style='grid', width=3, out=output)
|
|
assert output.getvalue() == '+---+---+---+\n| f | o | o |\n+---+---+---+\n| b| a| r|\n+---+---+---+\n'
|
|
|
|
|
|
def test_frame():
|
|
"""Tests the dataframe function"""
|
|
# mock of a pandas DataFrame
|
|
class DataFrame:
|
|
def __init__(self, data, headers):
|
|
self.data = data
|
|
self.columns = headers
|
|
|
|
def __array__(self):
|
|
return self.data
|
|
|
|
# build a mock DataFrame
|
|
df = DataFrame(np.array([[1, 2, 3]]), ['a', 'b', 'c'])
|
|
output = StringIO()
|
|
dataframe(df, width=4, style='fancy_grid', out=output)
|
|
assert output.getvalue() == '╒════╤════╤════╕\n│ a │ b │ c │\n╞════╪════╪════╡\n│ 1│ 2│ 3│\n╘════╧════╧════╛\n'
|
|
|
|
|
|
def test_banner():
|
|
"""Tests the banner function"""
|
|
output = StringIO()
|
|
banner('hello world', style='clean', width=11, out=output)
|
|
assert output.getvalue() == ' ─────────── \n hello world \n ─────────── \n'
|
|
|
|
output = StringIO()
|
|
banner('!', style='banner', width=1, out=output)
|
|
assert output.getvalue() == '╒═╕\n│!│\n╘═╛\n'
|
|
|
|
|
|
def test_hrule():
|
|
"""Tests the hrule function"""
|
|
output = hrule(1, width=11)
|
|
assert len(output) == 11
|
|
assert '───────────'
|