def test_colorbar(self): arr = self.biarr for mode in ["density", "hist2d", "hexbin"]: with self.subTest(mode=mode): out = density(arr, mode=mode, colorbar=True) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all")
def test_multiple(self): """Test generation of plot with multiple records.""" for arr in [self.biarr, self.triarr]: with self.subTest(arr=arr): out = density(arr, vmin=0) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all")
def test_none(self): """Test generation of plot with no data.""" for arr in [np.empty(0)]: with self.subTest(arr=arr): out = density(arr) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all")
def test_colorbar(self): for arr in [self.biarr, self.triarr]: for mode in ["density", "hist2d"]: # hexbin won't work for triarr with self.subTest(mode=mode): out = density(arr, mode=mode, colorbar=True) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all")
def test_cmap(self): arr = self.biarr cmap = "viridis" for mode in ["density", "hist2d", "hexbin"]: with self.subTest(mode=mode): out = density(arr, mode=mode, cmap=cmap) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all")
def test_contours(self): arr = self.biarr contours = [0.9, 0.5] for mode in ["density"]: with self.subTest(mode=mode): out = density(arr, mode=mode, contours=contours) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all")
def test_one(self): """Test generation of plot with one record.""" for arr in [self.biarr[0, :], self.triarr[0, :]]: with self.subTest(arr=arr): out = density(arr) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all")
def test_contours_levels(self): arr = self.biarr mode = "density" for levels in [3, None]: with self.subTest(levels=levels): out = density(arr, mode=mode, contours=True, percentiles=False) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all")
def test_modes(self): # """Tests different ploting modes.""" for arr in [self.biarr, self.triarr]: with self.subTest(arr=arr): for mode in ["density", "hist2d", "hexbin"]: with self.subTest(mode=mode): try: out = density(arr, mode=mode, vmin=0) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all") except NotImplementedError: # some are not implemented for 3D pass
def test_bivariate_logscale(self): # """Tests logscale for different ploting modes using bivariate data.""" arr = self.biarr with np.errstate(invalid="ignore"): # ignore for tests arr[arr < 0] = np.nan for logspacing in [(True, True), (False, False), (False, True), (True, False)]: lx, ly = logspacing with self.subTest(logx=lx, logy=ly): for mode in ["density", "hist2d", "hexbin"]: with self.subTest(mode=mode): out = density(arr, mode=mode, logx=lx, logy=ly) self.assertTrue(isinstance(out, matplotlib.axes.Axes)) plt.close("all")