def test_get_plot(self, winequality_vd):
current_cursor().execute("DROP MODEL IF EXISTS model_test_plot")
model_test = XGBoostRegressor("model_test_plot", )
model_test.fit(winequality_vd, ["alcohol"], "quality")
result = model_test.plot()
assert len(result.get_default_bbox_extra_artists()) in (9, 12)
plt.close("all")
model_test.drop()
def test_get_plot(self, base, winequality_vd):
base.cursor.execute("DROP MODEL IF EXISTS model_test_plot")
model_test = XGBoostRegressor("model_test_plot", cursor=base.cursor)
model_test.fit(winequality_vd, ["alcohol"], "quality")
result = model_test.plot()
assert len(result.get_default_bbox_extra_artists()) == 9
plt.close("all")
model_test.drop()