def test_1D_continuous_bar_plot(self):
ale_eff = aleplot_1D_discrete(X=self.X,
model=self.model,
feature="x4",
include_CI=True)
fig, ax, ax2 = plot_1D_discrete_eff(ale_eff, self.X)
self.assertCountEqual(ale_eff['size'],
[bar.get_height() for bar in ax2.patches])
def test_1D_continuous_line_plot(self):
ale_eff = aleplot_1D_discrete(X=self.X,
model=self.model,
feature="x4",
include_CI=True)
fig, ax, ax2 = plot_1D_discrete_eff(ale_eff, self.X)
self.assertCountEqual(ax.lines[0].get_xydata()[:, 0], ale_eff.index)
self.assertCountEqual(ax.lines[0].get_xydata()[:, 1], ale_eff.eff)
def test_1D_continuous_ci_plot(self):
ale_eff = aleplot_1D_discrete(X=self.X,
model=self.model,
feature="x4",
include_CI=True)
fig, ax, ax2 = plot_1D_discrete_eff(ale_eff, self.X)
self.assertCountEqual(np.round(ax.lines[1].get_xydata()[1:, 1], 8),
np.round(ale_eff['lowerCI_95%'][1:], 8))
self.assertCountEqual(np.round(ax.lines[2].get_xydata()[1:, 1], 8),
np.round(ale_eff['upperCI_95%'][1:], 8))