def test_get_plot(self, model):
result = model.plot(color=[
"r",
"b",
])
assert len(result.get_default_bbox_extra_artists()) == 9
plt.close("all")
model_test = LocalOutlierFactor("model_test_plot", cursor=model.cursor)
model_test.drop()
model_test.fit(
"public.titanic",
[
"age",
],
)
result = model_test.plot(color=[
"r",
"b",
])
assert len(result.get_default_bbox_extra_artists()) == 9
model_test.drop()
model_test.fit(
"public.titanic",
[
"age",
"fare",
"pclass",
],
)
result = model_test.plot(color=[
"r",
"b",
])
assert len(result.get_default_bbox_extra_artists()) == 3
model_test.drop()
def model(titanic_vd):
create_verticapy_schema()
model_class = LocalOutlierFactor("lof_model_test",)
model_class.drop()
model_class.fit("public.titanic", ["age", "fare"])
yield model_class
model_class.drop()
def test_model_from_vDF(self, titanic_vd):
model_test = LocalOutlierFactor("lof_from_vDF_tmp",)
model_test.drop()
model_test.fit(titanic_vd, ["age", "fare"])
assert model_test.predict()["lof_score"].mean() == pytest.approx(
1.17226637499694, abs=1e-6
)
model_test.drop()
def model(base, titanic_vd):
try:
create_verticapy_schema(base.cursor)
except:
pass
model_class = LocalOutlierFactor("lof_model_test", cursor=base.cursor)
model_class.drop()
model_class.fit(
"public.titanic",
[
"age",
"fare",
],
)
yield model_class
model_class.drop()
def test_drop(self):
model_test = LocalOutlierFactor("model_test_drop",)
model_test.drop()
model_test.fit("public.titanic", ["age", "fare"])
current_cursor().execute(
"SELECT model_name FROM verticapy.models WHERE model_name IN ('model_test_drop', '\"model_test_drop\"')"
)
assert current_cursor().fetchone()[0] in (
"model_test_drop",
'"model_test_drop"',
)
model_test.drop()
current_cursor().execute(
"SELECT model_name FROM verticapy.models WHERE model_name IN ('model_test_drop', '\"model_test_drop\"')"
)
assert current_cursor().fetchone() is None
def test_drop(self, base):
model_test = LocalOutlierFactor("model_test_drop", cursor=base.cursor)
model_test.drop()
model_test.fit(
"public.titanic",
[
"age",
"fare",
],
)
base.cursor.execute(
"SELECT model_name FROM verticapy.models WHERE model_name IN ('model_test_drop', '\"model_test_drop\"')"
)
assert base.cursor.fetchone()[0] in ('model_test_drop',
'"model_test_drop"')
model_test.drop()
base.cursor.execute(
"SELECT model_name FROM verticapy.models WHERE model_name IN ('model_test_drop', '\"model_test_drop\"')"
)
assert base.cursor.fetchone() is None