def test_contour(self, base, winequality_vd):
model_test = Lasso("model_contour", cursor=base.cursor)
model_test.drop()
model_test.fit(
winequality_vd,
["residual_sugar", "alcohol",],
"quality",
)
result = model_test.contour()
assert len(result.get_default_bbox_extra_artists()) == 10
model_test.drop()
def test_contour(self, winequality_vd):
model_test = Lasso("model_contour", )
model_test.drop()
model_test.fit(
winequality_vd,
["residual_sugar", "alcohol"],
"quality",
)
with warnings.catch_warnings(record=True) as w:
result = model_test.contour()
assert len(result.get_default_bbox_extra_artists()) == 10
model_test.drop()