def test_SimpleTable_4(self):
# Basic test, test_SimpleTable_4 test uses custom txt_fmt
txt_fmt1 = dict(data_fmts=['%3.2f', '%d'],
empty_cell=' ',
colwidths=1,
colsep=' * ',
row_pre='* ',
row_post=' *',
table_dec_above='*',
table_dec_below='*',
header_dec_below='*',
header_fmt='%s',
stub_fmt='%s',
title_align='r',
header_align='r',
data_aligns="r",
stubs_align="l",
fmt='txt')
ltx_fmt1 = default_latex_fmt.copy()
html_fmt1 = default_html_fmt.copy()
cell0data = 0.0000
cell1data = 1
row0data = [cell0data, cell1data]
row1data = [2, 3.333]
table1data = [row0data, row1data]
test1stubs = ('stub1', 'stub2')
test1header = ('header1', 'header2')
tbl = SimpleTable(table1data,
test1header,
test1stubs,
txt_fmt=txt_fmt1,
ltx_fmt=ltx_fmt1,
html_fmt=html_fmt1)
def test_txt_fmt1(self):
# Limited test of custom txt_fmt
desired = """
*****************************
* * header1 * header2 *
*****************************
* stub1 * 0.00 * 1 *
* stub2 * 2.00 * 3 *
*****************************
"""
actual = '\n%s\n' % tbl.as_text()
#print(actual)
#print(desired)
self.assertEqual(actual, desired)
def test_ltx_fmt1(self):
# Limited test of custom ltx_fmt
desired = r"""
\begin{tabular}{lcc}
\toprule
& \textbf{header1} & \textbf{header2} \\
\midrule
\textbf{stub1} & 0.0 & 1 \\
\textbf{stub2} & 2 & 3.333 \\
\bottomrule
\end{tabular}
"""
actual = '\n%s\n' % tbl.as_latex_tabular(center=False)
#print(actual)
#print(desired)
self.assertEqual(actual, desired)
# Test "center=True" (the default):
desired_centered = r"""
\begin{center}
%s
\end{center}
""" % desired[1:-1]
actual_centered = '\n%s\n' % tbl.as_latex_tabular()
self.assertEqual(actual_centered, desired_centered)
def test_html_fmt1(self):
# Limited test of custom html_fmt
desired = """
<table class="simpletable">
<tr>
<td></td> <th>header1</th> <th>header2</th>
</tr>
<tr>
<th>stub1</th> <td>0.0</td> <td>1</td>
</tr>
<tr>
<th>stub2</th> <td>2</td> <td>3.333</td>
</tr>
</table>
"""
actual = '\n%s\n' % tbl.as_html()
self.assertEqual(actual, desired)
test_txt_fmt1(self)
test_ltx_fmt1(self)
test_html_fmt1(self)
def test_simple_table_4(self):
# Basic test, test_simple_table_4 test uses custom txt_fmt
txt_fmt1 = dict(data_fmts = ['%3.2f', '%d'],
empty_cell = ' ',
colwidths = 1,
colsep=' * ',
row_pre = '* ',
row_post = ' *',
table_dec_above='*',
table_dec_below='*',
header_dec_below='*',
header_fmt = '%s',
stub_fmt = '%s',
title_align='r',
header_align = 'r',
data_aligns = "r",
stubs_align = "l",
fmt = 'txt'
)
ltx_fmt1 = default_latex_fmt.copy()
html_fmt1 = default_html_fmt.copy()
cell0data = 0.0000
cell1data = 1
row0data = [cell0data, cell1data]
row1data = [2, 3.333]
table1data = [ row0data, row1data ]
test1stubs = ('stub1', 'stub2')
test1header = ('header1', 'header2')
tbl = SimpleTable(table1data, test1header, test1stubs,txt_fmt=txt_fmt1,
ltx_fmt=ltx_fmt1, html_fmt=html_fmt1)
def test_txt_fmt1(self):
# Limited test of custom txt_fmt
desired = """
*****************************
* * header1 * header2 *
*****************************
* stub1 * 0.00 * 1 *
* stub2 * 2.00 * 3 *
*****************************
"""
actual = '\n%s\n' % tbl.as_text()
#print(actual)
#print(desired)
assert_equal(actual, desired)
def test_ltx_fmt1(self):
# Limited test of custom ltx_fmt
desired = r"""
\begin{tabular}{lcc}
\toprule
& \textbf{header1} & \textbf{header2} \\
\midrule
\textbf{stub1} & 0.0 & 1 \\
\textbf{stub2} & 2 & 3.333 \\
\bottomrule
\end{tabular}
"""
actual = '\n%s\n' % tbl.as_latex_tabular(center=False)
#print(actual)
#print(desired)
assert_equal(actual, desired)
# Test "center=True" (the default):
desired_centered = r"""
\begin{center}
%s
\end{center}
""" % desired[1:-1]
actual_centered = '\n%s\n' % tbl.as_latex_tabular()
assert_equal(actual_centered, desired_centered)
def test_html_fmt1(self):
# Limited test of custom html_fmt
desired = """
<table class="simpletable">
<tr>
<td></td> <th>header1</th> <th>header2</th>
</tr>
<tr>
<th>stub1</th> <td>0.0</td> <td>1</td>
</tr>
<tr>
<th>stub2</th> <td>2</td> <td>3.333</td>
</tr>
</table>
""" # noqa:W291
actual = '\n%s\n' % tbl.as_html()
assert_equal(actual, desired)
test_txt_fmt1(self)
test_ltx_fmt1(self)
test_html_fmt1(self)
import numpy as np
import unittest
from statsmodels.iolib.table import SimpleTable, default_txt_fmt
from statsmodels.iolib.table import default_latex_fmt
from statsmodels.iolib.table import default_html_fmt
import pandas
from statsmodels.regression.linear_model import OLS
ltx_fmt1 = default_latex_fmt.copy()
html_fmt1 = default_html_fmt.copy()
class TestSimpleTable(unittest.TestCase):
def test_SimpleTable_1(self):
# Basic test, test_SimpleTable_1
desired = '''
=====================
header1 header2
---------------------
stub1 1.30312 2.73999
stub2 1.95038 2.65765
---------------------
'''
test1data = [[1.30312, 2.73999], [1.95038, 2.65765]]
test1stubs = ('stub1', 'stub2')
test1header = ('header1', 'header2')
actual = SimpleTable(test1data,
test1header,
test1stubs,
txt_fmt=default_txt_fmt)
actual = '\n%s\n' % actual.as_text()
import numpy as np
from numpy.testing import assert_equal
from statsmodels.iolib.table import SimpleTable, default_txt_fmt
from statsmodels.iolib.table import default_latex_fmt
from statsmodels.iolib.table import default_html_fmt
import pandas
from statsmodels.regression.linear_model import OLS
ltx_fmt1 = default_latex_fmt.copy()
html_fmt1 = default_html_fmt.copy()
class TestSimpleTable(object):
def test_simple_table_1(self):
# Basic test, test_simple_table_1
desired = '''
=====================
header1 header2
---------------------
stub1 1.30312 2.73999
stub2 1.95038 2.65765
---------------------
'''
test1data = [[1.30312, 2.73999],[1.95038, 2.65765]]
test1stubs = ('stub1', 'stub2')
test1header = ('header1', 'header2')
actual = SimpleTable(test1data, test1header, test1stubs,
txt_fmt=default_txt_fmt)
actual = '\n%s\n' % actual.as_text()
assert_equal(desired, str(actual))