class TestPlotHistogram(PlotTestCase):
def setUp(self):
self.darray = DataArray(np.random.randn(2, 3, 4))
def test_3d_array(self):
self.darray.plot_hist()
def test_title_no_name(self):
self.darray.plot_hist()
self.assertEqual('', plt.gca().get_title())
def test_title_uses_name(self):
self.darray.name = 'randompoints'
self.darray.plot_hist()
self.assertIn(self.darray.name, plt.gca().get_title())
def test_ylabel_is_count(self):
self.darray.plot_hist()
self.assertEqual('Count', plt.gca().get_ylabel())
def test_can_pass_in_kwargs(self):
nbins = 5
self.darray.plot_hist(bins=nbins)
self.assertEqual(nbins, len(plt.gca().patches))
def test_can_pass_in_axis(self):
self.pass_in_axis(self.darray.plot_hist)
def test_primitive_returned(self):
h = self.darray.plot_hist()
self.assertTrue(isinstance(h[-1][0], mpl.patches.Rectangle))
def test_plot_nans(self):
self.darray[0, 0, 0] = np.nan
self.darray.plot_hist()
