def test_barchart_xtlfontsize():
''' bar chart xticklabel fontsize. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), xticklabelfontsize=20,
group_names=['a', 'b', 'c'])
def test_barchart_legend_opts():
''' bar chart legend options. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), legendloc='lower center', legendncol=2,
entry_names=['X', 'Y'])
def test_barchart_xtlrotation():
''' bar chart xticklabelrotation. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), xticklabelrotation=60,
group_names=['a', 'b', 'c'])
def test_invalid_cluster_bar_shrink(self):
''' Invalid cluster_bar_shrink. '''
with self.assertRaisesRegexp(ValueError,
r'\[barchart\] .*cluster_bar_shrink.*'):
barchart.draw(self.axes,
_data(),
breakdown=False,
cluster_bar_shrink=1.2)
def test_barchart_xtlfontsize():
''' bar chart xticklabel fontsize. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax,
_data(),
xticklabelfontsize=20,
group_names=['a', 'b', 'c'])
def test_barchart_xtlrotation():
''' bar chart xticklabelrotation. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax,
_data(),
xticklabelrotation=60,
group_names=['a', 'b', 'c'])
def test_barchart_legend_opts():
''' bar chart legend options. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax,
_data(),
legendloc='lower center',
legendncol=2,
entry_names=['X', 'Y'])
def test_barchart_hdls():
''' bar chart handlers. '''
fig = plt.figure()
ax = fig.gca()
hdls = barchart.draw(ax, _data())
ax.legend(hdls, ['E1', 'E2'])
def test_barchart_hdls():
''' bar chart handlers. '''
fig = plt.figure()
ax = fig.gca()
hdls = barchart.draw(ax, _data())
ax.legend(hdls, ['E1', 'E2'])
def test_barchart_edgecolor_none():
''' bar chart edgecolor. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), edgecolor=None)
def test_barchart_hatchcolor():
''' bar chart hatchcolor. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), hatchcolor='r', hatchs=['/', '//'])
def test_barchart_xticks():
''' bar chart xticks. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), xticks=[0, 2, 3])
def test_barchart_colors():
''' bar chart colors. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), colors=['r', 'b'])
def test_barchart_xticks():
''' bar chart xticks. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), xticks=[0, 2, 3])
def test_barchart_cbshrk_nobkdn():
''' bar chart cluster_bar_shrink without breakdown. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), cluster_bar_shrink=0.5, breakdown=False)
def test_invalid_group_names(self):
''' Invalid group names. '''
with self.assertRaisesRegexp(ValueError,
r'\[barchart\] .*group names.*'):
barchart.draw(self.axes, _data(), group_names=['a', 'b', 'c', 'd'])
def test_invalid_data(self):
''' Invalid data. '''
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*array.*'):
barchart.draw(self.axes, [[1, 2], [1, 2, 3]])
def test_invalid_data(self):
''' Invalid data. '''
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*array.*'):
barchart.draw(self.axes, [[1, 2], [1, 2, 3]])
def test_invalid_data_dim(self):
''' Invalid data dimension. '''
for d in [0, 1, 3, 4]:
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*dim.*'):
barchart.draw(self.axes, np.zeros([3] * d))
def test_return_num_handlers(self):
''' Returned number of handlers. '''
hdls = barchart.draw(self.axes, _data())
self.assertEqual(len(hdls), len(_data()[0]))
def test_barchart_edgecolor_none():
''' bar chart edgecolor. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), edgecolor=None)
def test_barchart_linewidth():
''' bar chart linewidth. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), linewidth=5.0)
def test_barchart_yaxis_log():
''' bar chart yaxis log. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), log=True)
def test_invalid_entry_names(self):
''' Invalid entry names. '''
with self.assertRaisesRegexp(ValueError,
r'\[barchart\] .*entry names.*'):
barchart.draw(self.axes, _data(), entry_names=['x', 'y', 'z'])
def test_barchart_xticks_nobkdn():
''' bar chart xticks without breakdown. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), xticks=[0, 2, 3], breakdown=False)
def test_invalid_cluster_bar_shrink(self):
''' Invalid cluster_bar_shrink. '''
with self.assertRaisesRegexp(ValueError,
r'\[barchart\] .*cluster_bar_shrink.*'):
barchart.draw(self.axes, _data(), breakdown=False,
cluster_bar_shrink=1.2)
def test_invalid_group_names(self):
''' Invalid group names. '''
with self.assertRaisesRegexp(ValueError,
r'\[barchart\] .*group names.*'):
barchart.draw(self.axes, _data(), group_names=['a', 'b', 'c', 'd'])
def test_invalid_xticks(self):
''' Invalid xticks. '''
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*xticks.*'):
barchart.draw(self.axes, _data(), xticks=['x'])
def test_barchart_yaxis_log():
''' bar chart yaxis log. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), log=True)
def test_not_enough_def_colors(self):
''' Not enough default colors. '''
with self.assertRaisesRegexp(ValueError,
r'\[barchart\] .*default colors.*'):
barchart.draw(self.axes, [[1] * 100])
def test_barchart_cbshrk_nobkdn():
''' bar chart cluster_bar_shrink without breakdown. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), cluster_bar_shrink=0.5, breakdown=False)
def test_invalid_colors(self):
''' Invalid colors. '''
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*colors.*'):
barchart.draw(self.axes, _data(), colors=['k'])
def test_barchart_xticks_nobkdn():
''' bar chart xticks without breakdown. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), xticks=[0, 2, 3], breakdown=False)
def test_invalid_hatchs(self):
''' Invalid hatchs. '''
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*hatchs.*'):
barchart.draw(self.axes, _data(), hatchs=['/', '//', 'xx'])
def test_barchart_colors_nobkdn():
''' bar chart colors without breakdown. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), colors=['r', 'b'], breakdown=False)
def test_barchart_base():
''' bar chart base. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data())
def test_barchart_linewidth():
''' bar chart linewidth. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), linewidth=5.0)
def test_barchart_nparray():
''' bar chart with data np.array. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, np.array(_data()))
def test_barchart_colors_nobkdn():
''' bar chart colors without breakdown. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), colors=['r', 'b'], breakdown=False)
def test_not_enough_def_colors(self):
''' Not enough default colors. '''
with self.assertRaisesRegexp(ValueError,
r'\[barchart\] .*default colors.*'):
barchart.draw(self.axes, [[1] * 100])
def test_barchart_colors():
''' bar chart colors. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), colors=['r', 'b'])
def test_invalid_colors(self):
''' Invalid colors. '''
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*colors.*'):
barchart.draw(self.axes, _data(), colors=['k'])
def test_return_num_handlers(self):
''' Returned number of handlers. '''
hdls = barchart.draw(self.axes, _data())
self.assertEqual(len(hdls), len(_data()[0]))
def test_barchart_nparray():
''' bar chart with data np.array. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, np.array(_data()))
def test_invalid_data_dim(self):
''' Invalid data dimension. '''
for d in [0, 1, 3, 4]:
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*dim.*'):
barchart.draw(self.axes, np.zeros([3] * d))
def test_barchart_group_names():
''' bar chart group names. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), group_names=['Aa', 'Bb', '$Cc$'])
def test_invalid_entry_names(self):
''' Invalid entry names. '''
with self.assertRaisesRegexp(ValueError,
r'\[barchart\] .*entry names.*'):
barchart.draw(self.axes, _data(), entry_names=['x', 'y', 'z'])
def test_barchart_width():
''' bar chart width. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), width=0.5)
def test_invalid_xticks(self):
''' Invalid xticks. '''
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*xticks.*'):
barchart.draw(self.axes, _data(), xticks=['x'])
def test_barchart_hatchcolor():
''' bar chart hatchcolor. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), hatchcolor='r', hatchs=['/', '//'])
def test_barchart_base():
''' bar chart base. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data())
def test_barchart_group_names():
''' bar chart group names. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), group_names=['Aa', 'Bb', '$Cc$'])
def test_invalid_hatchs(self):
''' Invalid hatchs. '''
with self.assertRaisesRegexp(ValueError, r'\[barchart\] .*hatchs.*'):
barchart.draw(self.axes, _data(), hatchs=['/', '//', 'xx'])
def test_barchart_entry_names():
''' bar chart entry names. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), entry_names=['x', '$x^2$'])
def test_barchart_nobkdn():
''' bar chart without breakdown. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), breakdown=False)
def test_barchart_width():
''' bar chart width. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), width=0.5)
def test_barchart_entry_names():
''' bar chart entry names. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), entry_names=['x', '$x^2$'])
def test_barchart_cbshrk():
''' bar chart cluster_bar_shrink. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), cluster_bar_shrink=0.5)
def test_barchart_width_nobkdn():
''' bar chart width without breakdown. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), width=0.5, breakdown=False)
def test_barchart_cbshrk():
''' bar chart cluster_bar_shrink. '''
fig = plt.figure()
ax = fig.gca()
barchart.draw(ax, _data(), cluster_bar_shrink=0.5)
