def test_sign_plot(self):
x = np.array([[1, 1, 1], [1, 1, 0], [1, 0, 1]])
a = splt.sign_plot(x, flat=True, labels=False)
with self.assertRaises(ValueError):
splt.sign_plot(x.astype(float), flat=True, labels=False)
self.assertTrue(isinstance(a, ma._subplots.Axes))
def test_sign_plot_nonflat(self):
x = np.array([[1., 0.00119517,
0.00278329], [0.00119517, 1., 0.18672227],
[0.00278329, 0.18672227, 1.]])
a, cbar = splt.sign_plot(x, cbar=True, labels=False)
with self.assertRaises(ValueError):
splt.sign_plot(x, cmap=[1, 1], labels=False)
with self.assertRaises(ValueError):
splt.sign_plot(x.astype(np.int64), labels=False)
self.assertTrue(
isinstance(a, ma._subplots.Axes)
and isinstance(cbar, mpl.colorbar.ColorbarBase))
def test_sign_plot_nonflat(self):
x = np.array([[-1., 0.00119517, 0.00278329],
[0.00119517, -1., 0.18672227],
[0.00278329, 0.18672227, -1.]])
a, cbar = splt.sign_plot(x)
print(type(cbar))
self.assertTrue(
isinstance(a, ma._subplots.Axes)
and isinstance(cbar, mpl.colorbar.ColorbarBase))
def test_sign_plot(self):
x = np.array([[-1, 1, 1], [1, -1, 0], [1, 0, -1]])
a = splt.sign_plot(x, flat=True, labels=False)
self.assertTrue(isinstance(a, ma._subplots.Axes))