def test_HTMLJinjaTableBlock_constructor_formatters_list():
table = abt.HTMLJinjaTableBlock(df, formatters=None, use_default_formatters=False)
assert table.formatters == []
table = abt.HTMLJinjaTableBlock(df, formatters=None, use_default_formatters=True)
assert table.formatters == abtf.DEFAULT_FORMATTERS + abtf.DEFAULT_DECIMALS_FORMATTER
formatter = abtf.TableFormatter()
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter], use_default_formatters=True)
assert table.formatters == abtf.DEFAULT_FORMATTERS + [formatter] + abtf.DEFAULT_DECIMALS_FORMATTER
def test_HTMLJinjaTableBlock_constructor_df_formatter(mock_formatter):
mock_formatter.return_value = (df - 1)
formatter = abtf.TableFormatter()
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter], use_default_formatters=False)
mock_formatter.assert_called_with(df)
assert table.df.equals(df - 1)
# NotImplementedError should be handled
mock_formatter.side_effect = NotImplementedError
formatter = abtf.TableFormatter()
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter], use_default_formatters=False)
mock_formatter.assert_called_with(df)
def test_HTMLJinjaTableBlock_modify_cell_content_not_implemented(mock_formatter):
mock_formatter.side_effect = NotImplementedError
formatter = abtf.TableFormatter()
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter], use_default_formatters=False)
res = table.modify_cell_content(42., None, None)
assert mock_formatter.call_count == 1
assert res == 42.
def test_HTMLJinjaTableBlock_modify_cell_content(mock_formatter):
mock_formatter.return_value = 1234
formatter = abtf.TableFormatter()
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter], use_default_formatters=False)
res = table.modify_cell_content(None, None, None)
assert mock_formatter.call_count == 1
assert res == 1234
def test_HTMLJinjaTableBlock_insert_additional_html_not_implemented(mock_formatter):
mock_formatter.side_effect = NotImplementedError
formatter = abtf.TableFormatter()
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter], use_default_formatters=False)
res = table.insert_additional_html()
mock_formatter.assert_called_once_with()
assert res == ''
def test__get_header_iterable_multiindex():
df = pd.DataFrame(np.arange(12, dtype=float).reshape(3, 4),
index=['a', 'b', 'c'],
columns=['aa', 'bb', 'cc', 'aa'])
df['grouping'] = 'g'
df = df.reset_index()
df = df.groupby(['grouping', 'index']).sum().unstack()
t = abt.HTMLJinjaTableBlock(df)
result = t._get_header_iterable()
expected = [[
abt.IndexCell('aa', [('aa', 'a'), ('aa', 'b'), ('aa', 'c')], 3, 1),
abt.IndexCell('bb', [('bb', 'a'), ('bb', 'b'), ('bb', 'c')], 3, 1),
abt.IndexCell('cc', [('cc', 'a'), ('cc', 'b'), ('cc', 'c')], 3, 1),
abt.IndexCell('aa', [('aa', 'a'), ('aa', 'b'), ('aa', 'c')], 3, 1),
],
[
abt.IndexCell('a', [('aa', 'a')], 1, 1),
abt.IndexCell('b', [('aa', 'b')], 1, 1),
abt.IndexCell('c', [('aa', 'c')], 1, 1),
abt.IndexCell('a', [('bb', 'a')], 1, 1),
abt.IndexCell('b', [('bb', 'b')], 1, 1),
abt.IndexCell('c', [('bb', 'c')], 1, 1),
abt.IndexCell('a', [('cc', 'a')], 1, 1),
abt.IndexCell('b', [('cc', 'b')], 1, 1),
abt.IndexCell('c', [('cc', 'c')], 1, 1),
abt.IndexCell('a', [('aa', 'a')], 1, 1),
abt.IndexCell('b', [('aa', 'b')], 1, 1),
abt.IndexCell('c', [('aa', 'c')], 1, 1),
]]
assert result == expected
def test__aggregate_css_formatters_not_implemented():
formatter = abtf.TableFormatter()
formatter.dummy_function = MagicMock(side_effect=NotImplementedError())
table = abt.HTMLJinjaTableBlock(df,
formatters=[formatter],
use_default_formatters=False)
res = table._aggregate_css_formatters('dummy_function')
assert res == 'style=""'
def test__aggregate_css_formatters_no_args():
dummy_css = 'dummy_css'
formatter = abtf.TableFormatter()
formatter.dummy_function = MagicMock(return_value='dummy_css')
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter], use_default_formatters=False)
res = table._aggregate_css_formatters('dummy_function')
assert res == 'style="' + dummy_css + '"'
formatter.dummy_function.assert_called_once_with()
def test__aggregate_css_formatters_args():
dummy_css = 'dummy_css'
dummy_parameter = '42'
formatter = abtf.TableFormatter()
formatter.dummy_function = MagicMock(side_effect=lambda x: dummy_css + x)
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter], use_default_formatters=False)
res = table._aggregate_css_formatters('dummy_function', fmt_args=[dummy_parameter])
assert res == 'style="' + dummy_css + dummy_parameter + '"'
formatter.dummy_function.assert_called_once_with(dummy_parameter)
def test_HTMLJinjaTableBlock_insert_additional_html_concat():
dummy_html1 = '<style></style>'
formatter1 = abtf.TableFormatter()
formatter1.insert_additional_html = MagicMock(return_value=dummy_html1)
dummy_html2 = '<div></div>'
formatter2 = abtf.TableFormatter()
formatter2.insert_additional_html = MagicMock(return_value=dummy_html2)
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter1, formatter2], use_default_formatters=False)
res = table.insert_additional_html()
assert res == dummy_html1 + dummy_html2
def test__aggregate_css_formatters_concatenation():
css1 = 'aaaaa'
formatter1 = abtf.TableFormatter()
formatter1.dummy_function = MagicMock(return_value=css1)
css2 = 'bbbbb'
formatter2 = abtf.TableFormatter()
formatter2.dummy_function = MagicMock(return_value=css2)
table = abt.HTMLJinjaTableBlock(df, formatters=[formatter1, formatter2], use_default_formatters=False)
res = table._aggregate_css_formatters('dummy_function')
res = res.replace(' ', '')
res = res.replace("'", '"')
assert res == 'style="aaaaa;bbbbb"'
def test__get_header_iterable_plain_index():
columns = ['a', 'b', 'c', 'd', 'e']
p = pd.DataFrame(np.ones((4, 5)), columns=columns)
t = abt.HTMLJinjaTableBlock(p)
result = t._get_header_iterable()
expected = [[
abt.HTMLJinjaTableBlock.HeaderCell('a', ['a'], 1, 1),
abt.HTMLJinjaTableBlock.HeaderCell('b', ['b'], 1, 1),
abt.HTMLJinjaTableBlock.HeaderCell('c', ['c'], 1, 1),
abt.HTMLJinjaTableBlock.HeaderCell('d', ['d'], 1, 1),
abt.HTMLJinjaTableBlock.HeaderCell('e', ['e'], 1, 1),
]]
assert result == expected
def test__get_index_iterable_plain_index():
index = ['a', 'b', 'c', 'd', 'e']
p = pd.DataFrame(np.ones((5, 4)), index=index)
t = abt.HTMLJinjaTableBlock(p)
result = t._get_index_iterable()
expected = [
[abt.IndexCell('a', ['a'], 1, 1)],
[abt.IndexCell('b', ['b'], 1, 1)],
[abt.IndexCell('c', ['c'], 1, 1)],
[abt.IndexCell('d', ['d'], 1, 1)],
[abt.IndexCell('e', ['e'], 1, 1)],
]
assert result == expected
def test__header_depth_and_span_merge():
columns = pd.MultiIndex.from_tuples([('a', 'a', 'a'), ('a', 'a', 'a')])
p = pd.DataFrame([], columns=columns)
t = abt.HTMLJinjaTableBlock(p, merge_vertical=True)
result = t._get_header_iterable()
expected = [
[abt.IndexCell('a', [('a', 'a', 'a'), ('a', 'a', 'a')], 2, 2)],
[],
[
abt.IndexCell('a', [('a', 'a', 'a')], 1, 1),
abt.IndexCell('a', [('a', 'a', 'a')], 1, 1),
],
]
assert result == expected
def test__jinja_hides_multiindex_flattening():
df = pd.DataFrame(columns=['a', 'b', 'c'],
index=pd.MultiIndex.from_tuples([('x', 'x'), ('y', 'y'),
('z', 'z')]))
table = abt.HTMLJinjaTableBlock(df,
formatters=[abtf.FmtExpandMultiIndex()],
use_default_formatters=False)
container = MagicMock()
actual_cfg = MagicMock()
table._write_contents(container, actual_cfg)
assert "('x', 'x')" not in str(container.append.call_args_list[0][0][0])
def test__no_vertical_merge():
"""
should coalesce vertically if merge_vertical is passed, except for the
bottom cell. for example, the following multiindex:
"""
columns = pd.MultiIndex.from_tuples([('a', 'a', 'a')])
p = pd.DataFrame([], columns=columns)
t = abt.HTMLJinjaTableBlock(p)
result = t._get_header_iterable()
expected = [
[abt.HTMLJinjaTableBlock.HeaderCell('a', [('a', 'a', 'a')], 1, 1)],
[abt.HTMLJinjaTableBlock.HeaderCell('a', [('a', 'a', 'a')], 1, 1)],
[abt.HTMLJinjaTableBlock.HeaderCell('a', [('a', 'a', 'a')], 1, 1)],
]
assert result == expected
def test__no_merge_if_parents_differ():
"""
should not merge multiindex cells if their parent cells differ.
"""
columns = pd.MultiIndex.from_tuples([('a', 'b', 'c'), ('c', 'b', 'a')])
p = pd.DataFrame([], columns=columns)
t = abt.HTMLJinjaTableBlock(p)
result = t._get_header_iterable()
expected = [
[
abt.HTMLJinjaTableBlock.HeaderCell('a', [('a', 'b', 'c')], 1, 1),
abt.HTMLJinjaTableBlock.HeaderCell('c', [('c', 'b', 'a')], 1, 1),
],
[
abt.HTMLJinjaTableBlock.HeaderCell('b', [('a', 'b', 'c')], 1, 1),
abt.HTMLJinjaTableBlock.HeaderCell('b', [('c', 'b', 'a')], 1, 1),
],
[
abt.HTMLJinjaTableBlock.HeaderCell('c', [('a', 'b', 'c')], 1, 1),
abt.HTMLJinjaTableBlock.HeaderCell('a', [('c', 'b', 'a')], 1, 1),
],
]
assert result == expected
def test__jinja_header_row_name():
df = pd.DataFrame(
[["first", 2, 3], ["second", 5, 6]],
columns=['a', 'b', 'c'],
).set_index("a")
formatter = abtf.TableFormatter()
formatter._create_row_level_css = MagicMock(return_value=None)
table = abt.HTMLJinjaTableBlock(df,
formatters=[formatter],
use_default_formatters=False)
container = MagicMock()
actual_cfg = MagicMock()
table._write_contents(MagicMock(), MagicMock())
names = [
args[0][0].name
for args in formatter._create_row_level_css.call_args_list
]
assert names[0] == abtf.HEADER_ROW_NAME
assert names[1] == "first"
assert names[2] == "second"
def test__jinja_calls_formatters_correctly():
df = pd.DataFrame(columns=['a', 'b', 'c'], index=['x', 'y', 'z'])
formatter = abtf.TableFormatter()
formatter.create_cell_level_css = MagicMock(
side_effect=NotImplementedError)
table = abt.HTMLJinjaTableBlock(df,
formatters=[formatter],
use_default_formatters=False)
container = MagicMock()
actual_cfg = MagicMock()
table._write_contents(container, actual_cfg)
row_names = {
args[0][0][1]
for args in formatter.create_cell_level_css.call_args_list
}
col_names = {
args[0][0][2]
for args in formatter.create_cell_level_css.call_args_list
}
assert row_names == set(df.index) | {abtf.HEADER_ROW_NAME}
assert col_names == set(df.columns) | {abtf.INDEX_COL_NAME}
def test_HTMLJinjaTableBlock_join_css_substrings():
table = abt.HTMLJinjaTableBlock(df)
res = table._join_css_substrings(['a', 'b'], 'style')
assert res == 'style="a; b"'
def test__get_header_iterable_plain_index():
columns = ['a', 'b', 'c', 'd', 'e']
p = pd.DataFrame(np.ones((4, 5)), columns=columns)
t = abt.HTMLJinjaTableBlock(p)
result = t._get_header_iterable()
assert result == [columns]
