284 lines
8.5 KiB
Python
284 lines
8.5 KiB
Python
|
import datetime
|
||
|
|
from unittest.mock import Mock
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
from matplotlib.path import Path
|
||
|
|
from matplotlib.table import CustomCell, Table
|
||
|
|
from matplotlib.testing.decorators import image_comparison, check_figures_equal
|
||
|
|
from matplotlib.transforms import Bbox
|
||
|
|
import matplotlib.units as munits
|
||
|
|
|
||
|
|
|
||
|
|
def test_non_square():
|
||
|
|
# Check that creating a non-square table works
|
||
|
|
cellcolors = ['b', 'r']
|
||
|
|
plt.table(cellColours=cellcolors)
|
||
|
|
|
||
|
|
|
||
|
|
@image_comparison(['table_zorder.png'], remove_text=True)
|
||
|
|
def test_zorder():
|
||
|
|
data = [[66386, 174296],
|
||
|
|
[58230, 381139]]
|
||
|
|
|
||
|
|
colLabels = ('Freeze', 'Wind')
|
||
|
|
rowLabels = ['%d year' % x for x in (100, 50)]
|
||
|
|
|
||
|
|
cellText = []
|
||
|
|
yoff = np.zeros(len(colLabels))
|
||
|
|
for row in reversed(data):
|
||
|
|
yoff += row
|
||
|
|
cellText.append(['%1.1f' % (x/1000.0) for x in yoff])
|
||
|
|
|
||
|
|
t = np.linspace(0, 2*np.pi, 100)
|
||
|
|
plt.plot(t, np.cos(t), lw=4, zorder=2)
|
||
|
|
|
||
|
|
plt.table(cellText=cellText,
|
||
|
|
rowLabels=rowLabels,
|
||
|
|
colLabels=colLabels,
|
||
|
|
loc='center',
|
||
|
|
zorder=-2,
|
||
|
|
)
|
||
|
|
|
||
|
|
plt.table(cellText=cellText,
|
||
|
|
rowLabels=rowLabels,
|
||
|
|
colLabels=colLabels,
|
||
|
|
loc='upper center',
|
||
|
|
zorder=4,
|
||
|
|
)
|
||
|
|
plt.yticks([])
|
||
|
|
|
||
|
|
|
||
|
|
@image_comparison(['table_labels.png'])
|
||
|
|
def test_label_colours():
|
||
|
|
dim = 3
|
||
|
|
|
||
|
|
c = np.linspace(0, 1, dim)
|
||
|
|
colours = plt.cm.RdYlGn(c)
|
||
|
|
cellText = [['1'] * dim] * dim
|
||
|
|
|
||
|
|
fig = plt.figure()
|
||
|
|
|
||
|
|
ax1 = fig.add_subplot(4, 1, 1)
|
||
|
|
ax1.axis('off')
|
||
|
|
ax1.table(cellText=cellText,
|
||
|
|
rowColours=colours,
|
||
|
|
loc='best')
|
||
|
|
|
||
|
|
ax2 = fig.add_subplot(4, 1, 2)
|
||
|
|
ax2.axis('off')
|
||
|
|
ax2.table(cellText=cellText,
|
||
|
|
rowColours=colours,
|
||
|
|
rowLabels=['Header'] * dim,
|
||
|
|
loc='best')
|
||
|
|
|
||
|
|
ax3 = fig.add_subplot(4, 1, 3)
|
||
|
|
ax3.axis('off')
|
||
|
|
ax3.table(cellText=cellText,
|
||
|
|
colColours=colours,
|
||
|
|
loc='best')
|
||
|
|
|
||
|
|
ax4 = fig.add_subplot(4, 1, 4)
|
||
|
|
ax4.axis('off')
|
||
|
|
ax4.table(cellText=cellText,
|
||
|
|
colColours=colours,
|
||
|
|
colLabels=['Header'] * dim,
|
||
|
|
loc='best')
|
||
|
|
|
||
|
|
|
||
|
|
@image_comparison(['table_cell_manipulation.png'], style='mpl20')
|
||
|
|
def test_diff_cell_table(text_placeholders):
|
||
|
|
cells = ('horizontal', 'vertical', 'open', 'closed', 'T', 'R', 'B', 'L')
|
||
|
|
cellText = [['1'] * len(cells)] * 2
|
||
|
|
colWidths = [0.1] * len(cells)
|
||
|
|
|
||
|
|
_, axs = plt.subplots(nrows=len(cells), figsize=(4, len(cells)+1), layout='tight')
|
||
|
|
for ax, cell in zip(axs, cells):
|
||
|
|
ax.table(
|
||
|
|
colWidths=colWidths,
|
||
|
|
cellText=cellText,
|
||
|
|
loc='center',
|
||
|
|
edges=cell,
|
||
|
|
)
|
||
|
|
ax.axis('off')
|
||
|
|
|
||
|
|
|
||
|
|
def test_customcell():
|
||
|
|
types = ('horizontal', 'vertical', 'open', 'closed', 'T', 'R', 'B', 'L')
|
||
|
|
codes = (
|
||
|
|
(Path.MOVETO, Path.LINETO, Path.MOVETO, Path.LINETO, Path.MOVETO),
|
||
|
|
(Path.MOVETO, Path.MOVETO, Path.LINETO, Path.MOVETO, Path.LINETO),
|
||
|
|
(Path.MOVETO, Path.MOVETO, Path.MOVETO, Path.MOVETO, Path.MOVETO),
|
||
|
|
(Path.MOVETO, Path.LINETO, Path.LINETO, Path.LINETO, Path.CLOSEPOLY),
|
||
|
|
(Path.MOVETO, Path.MOVETO, Path.MOVETO, Path.LINETO, Path.MOVETO),
|
||
|
|
(Path.MOVETO, Path.MOVETO, Path.LINETO, Path.MOVETO, Path.MOVETO),
|
||
|
|
(Path.MOVETO, Path.LINETO, Path.MOVETO, Path.MOVETO, Path.MOVETO),
|
||
|
|
(Path.MOVETO, Path.MOVETO, Path.MOVETO, Path.MOVETO, Path.LINETO),
|
||
|
|
)
|
||
|
|
|
||
|
|
for t, c in zip(types, codes):
|
||
|
|
cell = CustomCell((0, 0), visible_edges=t, width=1, height=1)
|
||
|
|
code = tuple(s for _, s in cell.get_path().iter_segments())
|
||
|
|
assert c == code
|
||
|
|
|
||
|
|
|
||
|
|
@image_comparison(['table_auto_column.png'])
|
||
|
|
def test_auto_column():
|
||
|
|
fig, (ax1, ax2, ax3, ax4) = plt.subplots(4, 1)
|
||
|
|
|
||
|
|
# iterable list input
|
||
|
|
ax1.axis('off')
|
||
|
|
tb1 = ax1.table(
|
||
|
|
cellText=[['Fit Text', 2],
|
||
|
|
['very long long text, Longer text than default', 1]],
|
||
|
|
rowLabels=["A", "B"],
|
||
|
|
colLabels=["Col1", "Col2"],
|
||
|
|
loc="center")
|
||
|
|
tb1.auto_set_font_size(False)
|
||
|
|
tb1.set_fontsize(12)
|
||
|
|
tb1.auto_set_column_width([-1, 0, 1])
|
||
|
|
|
||
|
|
# iterable tuple input
|
||
|
|
ax2.axis('off')
|
||
|
|
tb2 = ax2.table(
|
||
|
|
cellText=[['Fit Text', 2],
|
||
|
|
['very long long text, Longer text than default', 1]],
|
||
|
|
rowLabels=["A", "B"],
|
||
|
|
colLabels=["Col1", "Col2"],
|
||
|
|
loc="center")
|
||
|
|
tb2.auto_set_font_size(False)
|
||
|
|
tb2.set_fontsize(12)
|
||
|
|
tb2.auto_set_column_width((-1, 0, 1))
|
||
|
|
|
||
|
|
# 3 single inputs
|
||
|
|
ax3.axis('off')
|
||
|
|
tb3 = ax3.table(
|
||
|
|
cellText=[['Fit Text', 2],
|
||
|
|
['very long long text, Longer text than default', 1]],
|
||
|
|
rowLabels=["A", "B"],
|
||
|
|
colLabels=["Col1", "Col2"],
|
||
|
|
loc="center")
|
||
|
|
tb3.auto_set_font_size(False)
|
||
|
|
tb3.set_fontsize(12)
|
||
|
|
tb3.auto_set_column_width(-1)
|
||
|
|
tb3.auto_set_column_width(0)
|
||
|
|
tb3.auto_set_column_width(1)
|
||
|
|
|
||
|
|
# 4 this used to test non-integer iterable input, which did nothing, but only
|
||
|
|
# remains to avoid re-generating the test image.
|
||
|
|
ax4.axis('off')
|
||
|
|
tb4 = ax4.table(
|
||
|
|
cellText=[['Fit Text', 2],
|
||
|
|
['very long long text, Longer text than default', 1]],
|
||
|
|
rowLabels=["A", "B"],
|
||
|
|
colLabels=["Col1", "Col2"],
|
||
|
|
loc="center")
|
||
|
|
tb4.auto_set_font_size(False)
|
||
|
|
tb4.set_fontsize(12)
|
||
|
|
|
||
|
|
|
||
|
|
def test_table_cells():
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
table = Table(ax)
|
||
|
|
|
||
|
|
cell = table.add_cell(1, 2, 1, 1)
|
||
|
|
assert isinstance(cell, CustomCell)
|
||
|
|
assert cell is table[1, 2]
|
||
|
|
|
||
|
|
cell2 = CustomCell((0, 0), 1, 2, visible_edges=None)
|
||
|
|
table[2, 1] = cell2
|
||
|
|
assert table[2, 1] is cell2
|
||
|
|
|
||
|
|
# make sure getitem support has not broken
|
||
|
|
# properties and setp
|
||
|
|
table.properties()
|
||
|
|
plt.setp(table)
|
||
|
|
|
||
|
|
|
||
|
|
@check_figures_equal(extensions=["png"])
|
||
|
|
def test_table_bbox(fig_test, fig_ref):
|
||
|
|
data = [[2, 3],
|
||
|
|
[4, 5]]
|
||
|
|
|
||
|
|
col_labels = ('Foo', 'Bar')
|
||
|
|
row_labels = ('Ada', 'Bob')
|
||
|
|
|
||
|
|
cell_text = [[f"{x}" for x in row] for row in data]
|
||
|
|
|
||
|
|
ax_list = fig_test.subplots()
|
||
|
|
ax_list.table(cellText=cell_text,
|
||
|
|
rowLabels=row_labels,
|
||
|
|
colLabels=col_labels,
|
||
|
|
loc='center',
|
||
|
|
bbox=[0.1, 0.2, 0.8, 0.6]
|
||
|
|
)
|
||
|
|
|
||
|
|
ax_bbox = fig_ref.subplots()
|
||
|
|
ax_bbox.table(cellText=cell_text,
|
||
|
|
rowLabels=row_labels,
|
||
|
|
colLabels=col_labels,
|
||
|
|
loc='center',
|
||
|
|
bbox=Bbox.from_extents(0.1, 0.2, 0.9, 0.8)
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@check_figures_equal(extensions=['png'])
|
||
|
|
def test_table_unit(fig_test, fig_ref):
|
||
|
|
# test that table doesn't participate in unit machinery, instead uses repr/str
|
||
|
|
|
||
|
|
class FakeUnit:
|
||
|
|
def __init__(self, thing):
|
||
|
|
pass
|
||
|
|
def __repr__(self):
|
||
|
|
return "Hello"
|
||
|
|
|
||
|
|
fake_convertor = munits.ConversionInterface()
|
||
|
|
# v, u, a = value, unit, axis
|
||
|
|
fake_convertor.convert = Mock(side_effect=lambda v, u, a: 0)
|
||
|
|
# not used, here for completeness
|
||
|
|
fake_convertor.default_units = Mock(side_effect=lambda v, a: None)
|
||
|
|
fake_convertor.axisinfo = Mock(side_effect=lambda u, a: munits.AxisInfo())
|
||
|
|
|
||
|
|
munits.registry[FakeUnit] = fake_convertor
|
||
|
|
|
||
|
|
data = [[FakeUnit("yellow"), FakeUnit(42)],
|
||
|
|
[FakeUnit(datetime.datetime(1968, 8, 1)), FakeUnit(True)]]
|
||
|
|
|
||
|
|
fig_test.subplots().table(data)
|
||
|
|
fig_ref.subplots().table([["Hello", "Hello"], ["Hello", "Hello"]])
|
||
|
|
fig_test.canvas.draw()
|
||
|
|
fake_convertor.convert.assert_not_called()
|
||
|
|
|
||
|
|
munits.registry.pop(FakeUnit)
|
||
|
|
assert not munits.registry.get_converter(FakeUnit)
|
||
|
|
|
||
|
|
|
||
|
|
def test_table_dataframe(pd):
|
||
|
|
# Test if Pandas Data Frame can be passed in cellText
|
||
|
|
|
||
|
|
data = {
|
||
|
|
'Letter': ['A', 'B', 'C'],
|
||
|
|
'Number': [100, 200, 300]
|
||
|
|
}
|
||
|
|
|
||
|
|
df = pd.DataFrame(data)
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
table = ax.table(df, loc='center')
|
||
|
|
|
||
|
|
for r, (index, row) in enumerate(df.iterrows()):
|
||
|
|
for c, col in enumerate(df.columns if r == 0 else row.values):
|
||
|
|
assert table[r if r == 0 else r+1, c].get_text().get_text() == str(col)
|
||
|
|
|
||
|
|
|
||
|
|
def test_table_fontsize():
|
||
|
|
# Test that the passed fontsize propagates to cells
|
||
|
|
tableData = [['a', 1], ['b', 2]]
|
||
|
|
fig, ax = plt.subplots()
|
||
|
|
test_fontsize = 20
|
||
|
|
t = ax.table(cellText=tableData, loc='top', fontsize=test_fontsize)
|
||
|
|
cell_fontsize = t[(0, 0)].get_fontsize()
|
||
|
|
assert cell_fontsize == test_fontsize, f"Actual:{test_fontsize},got:{cell_fontsize}"
|
||
|
|
cell_fontsize = t[(1, 1)].get_fontsize()
|
||
|
|
assert cell_fontsize == test_fontsize, f"Actual:{test_fontsize},got:{cell_fontsize}"
|