def test_csv01(self):
mydata = [[11,12],[21,22]]
myheaders = [ "Column 1", "Column 2" ]
mystubs = [ "Row 1", "Row 2" ]
tbl = SimpleTable(mydata, myheaders, mystubs, title="Title")
actual = '%s' % tbl.as_csv().strip()
desired = """
Title
,Column 1,Column 2
Row 1,11 ,12
Row 2,21 ,22
""".strip()
self.assertEqual(actual, desired)
def test_csv01(self):
mydata = [[11, 12], [21, 22]]
myheaders = ["Column 1", "Column 2"]
mystubs = ["Row 1", "Row 2"]
tbl = SimpleTable(mydata, myheaders, mystubs, title="Title")
actual = '%s' % tbl.as_csv().strip()
desired = """
Title
,Column 1,Column 2
Row 1,11 ,12
Row 2,21 ,22
""".strip()
self.assertEqual(actual, desired)
def _df_to_simpletable(df,
align='r',
float_format="%.4f",
header=True,
index=True,
table_dec_above='-',
table_dec_below=None,
header_dec_below='-',
pad_col=0,
pad_index=0):
dat = df.copy()
dat = dat.applymap(lambda x: _formatter(x, float_format))
if header:
headers = [str(x) for x in dat.columns.tolist()]
else:
headers = None
if index:
stubs = [str(x) + int(pad_index) * ' ' for x in dat.index.tolist()]
else:
dat.ix[:, 0] = [str(x) + int(pad_index) * ' ' for x in dat.ix[:, 0]]
stubs = None
st = SimpleTable(np.array(dat),
headers=headers,
stubs=stubs,
ltx_fmt=fmt_latex,
txt_fmt=fmt_txt)
st.output_formats['latex']['data_aligns'] = align
st.output_formats['txt']['data_aligns'] = align
st.output_formats['txt']['table_dec_above'] = table_dec_above
st.output_formats['txt']['table_dec_below'] = table_dec_below
st.output_formats['txt']['header_dec_below'] = header_dec_below
st.output_formats['txt']['colsep'] = ' ' * int(pad_col + 1)
return st
def _df_to_simpletable(
df,
align="r",
float_format="%.4f",
header=True,
index=True,
table_dec_above="-",
table_dec_below=None,
header_dec_below="-",
pad_col=0,
pad_index=0,
):
dat = df.copy()
dat = dat.applymap(lambda x: _formatter(x, float_format))
if header:
headers = [str(x) for x in dat.columns.tolist()]
else:
headers = None
if index:
stubs = [str(x) + int(pad_index) * " " for x in dat.index.tolist()]
else:
dat.ix[:, 0] = [str(x) + int(pad_index) * " " for x in dat.ix[:, 0]]
stubs = None
st = SimpleTable(np.array(dat), headers=headers, stubs=stubs, ltx_fmt=fmt_latex, txt_fmt=fmt_txt)
st.output_formats["latex"]["data_aligns"] = align
st.output_formats["txt"]["data_aligns"] = align
st.output_formats["txt"]["table_dec_above"] = table_dec_above
st.output_formats["txt"]["table_dec_below"] = table_dec_below
st.output_formats["txt"]["header_dec_below"] = header_dec_below
st.output_formats["txt"]["colsep"] = " " * int(pad_col + 1)
return st
def test_customlabel(self): """Limited test of custom custom labeling""" if has_numpy: tbl = SimpleTable(table1data, test1header, test1stubs, txt_fmt=txt_fmt1) tbl[1][1].data = np.nan tbl.label_cells(custom_labeller) print([[c.datatype for c in row] for row in tbl]) desired = """ ***************************** * * header1 * header2 * ***************************** * stub1 * -- * 1 * * stub2 * 2.00 * 3 * ***************************** """ actual = '\n%s\n' % tbl.as_text(missing='--') print(actual) print(desired) self.assertEqual(actual, desired)
def test_customlabel(self):
"""Limited test of custom custom labeling"""
if has_numpy:
tbl = SimpleTable(table1data,
test1header,
test1stubs,
txt_fmt=txt_fmt1)
tbl[1][1].data = np.nan
tbl.label_cells(custom_labeller)
print([[c.datatype for c in row] for row in tbl])
desired = """
*****************************
* * header1 * header2 *
*****************************
* stub1 * -- * 1 *
* stub2 * 2.00 * 3 *
*****************************
"""
actual = '\n%s\n' % tbl.as_text(missing='--')
print(actual)
print(desired)
self.assertEqual(actual, desired)
title_align='r',
header_align='r',
data_aligns="r",
stubs_align="l",
fmt='txt')
cell0data = 0.0000
cell1data = 1
row0data = [cell0data, cell1data]
row1data = [2, 3.333]
table1data = [row0data, row1data]
test1stubs = ('stub1', 'stub2')
test1header = ('header1', 'header2')
#test1header = ('header1\nheader1a', 'header2\nheader2a')
tbl = SimpleTable(table1data,
test1header,
test1stubs,
txt_fmt=txt_fmt1,
ltx_fmt=ltx_fmt1,
html_fmt=html_fmt1)
def custom_labeller(cell):
if cell.data is np.nan:
return 'missing'
class test_Cell(unittest.TestCase):
def test_celldata(self):
celldata = cell0data, cell1data, row1data[0], row1data[1]
cells = [
Cell(datum, datatype=i % 2) for i, datum in enumerate(celldata)
]