コード例 #1
0
 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()
コード例 #2
0
ファイル: test_lasso.py プロジェクト: vertica/VerticaPy
 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()