def model(base, winequality_vd):
base.cursor.execute("DROP MODEL IF EXISTS pca_model_test")
model_class = PCA("pca_model_test", cursor=base.cursor)
model_class.fit("public.winequality",
["citric_acid", "residual_sugar", "alcohol"])
yield model_class
model_class.drop()
def test_model_from_vDF(self, base, winequality_vd):
base.cursor.execute("DROP MODEL IF EXISTS pca_vDF")
model_test = PCA("pca_vDF", cursor=base.cursor)
model_test.fit(winequality_vd, ["alcohol", "quality"])
base.cursor.execute(
"SELECT model_name FROM models WHERE model_name = 'pca_vDF'")
assert base.cursor.fetchone()[0] == "pca_vDF"
model_test.drop()
def test_model_from_vDF(self, winequality_vd):
current_cursor().execute("DROP MODEL IF EXISTS pca_vDF")
model_test = PCA("pca_vDF",)
model_test.fit(winequality_vd, ["alcohol", "quality"])
current_cursor().execute(
"SELECT model_name FROM models WHERE model_name = 'pca_vDF'"
)
assert current_cursor().fetchone()[0] == "pca_vDF"
model_test.drop()
def test_set_cursor(self, base):
model_test = PCA("pca_cursor_test", cursor=base.cursor)
# TODO: creat a new cursor
model_test.set_cursor(base.cursor)
model_test.drop()
model_test.fit("public.winequality", ["alcohol"])
base.cursor.execute(
"SELECT model_name FROM models WHERE model_name = 'pca_cursor_test'"
)
assert base.cursor.fetchone()[0] == "pca_cursor_test"
model_test.drop()
def test_drop(self):
current_cursor().execute("DROP MODEL IF EXISTS pca_model_test_drop")
model_test = PCA("pca_model_test_drop",)
model_test.fit("public.winequality", ["alcohol", "quality"])
current_cursor().execute(
"SELECT model_name FROM models WHERE model_name = 'pca_model_test_drop'"
)
assert current_cursor().fetchone()[0] == "pca_model_test_drop"
model_test.drop()
current_cursor().execute(
"SELECT model_name FROM models WHERE model_name = 'pca_model_test_drop'"
)
assert current_cursor().fetchone() is None
def test_drop(self, base):
base.cursor.execute("DROP MODEL IF EXISTS pca_model_test_drop")
model_test = PCA("pca_model_test_drop", cursor=base.cursor)
model_test.fit("public.winequality", ["alcohol", "quality"])
base.cursor.execute(
"SELECT model_name FROM models WHERE model_name = 'pca_model_test_drop'"
)
assert base.cursor.fetchone()[0] == "pca_model_test_drop"
model_test.drop()
base.cursor.execute(
"SELECT model_name FROM models WHERE model_name = 'pca_model_test_drop'"
)
assert base.cursor.fetchone() is None
def model(winequality_vd):
model_class = PCA("pca_model_test",)
model_class.drop()
model_class.fit("public.winequality", ["citric_acid", "residual_sugar", "alcohol"])
yield model_class
model_class.drop()