def model(winequality_vd):
model_class = SVD("SVD_model_test", )
model_class.drop()
model_class.fit("public.winequality",
["citric_acid", "residual_sugar", "alcohol"])
yield model_class
model_class.drop()
def model(base, winequality_vd):
base.cursor.execute("DROP MODEL IF EXISTS SVD_model_test")
model_class = SVD("SVD_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, winequality_vd):
current_cursor().execute("DROP MODEL IF EXISTS SVD_vDF")
model_test = SVD("SVD_vDF", )
model_test.fit(winequality_vd, ["alcohol", "quality"])
current_cursor().execute(
"SELECT model_name FROM models WHERE model_name = 'SVD_vDF'")
assert current_cursor().fetchone()[0] == "SVD_vDF"
model_test.drop()
def test_model_from_vDF(self, base, winequality_vd):
base.cursor.execute("DROP MODEL IF EXISTS SVD_vDF")
model_test = SVD("SVD_vDF", cursor=base.cursor)
model_test.fit(winequality_vd, ["alcohol", "quality"])
base.cursor.execute(
"SELECT model_name FROM models WHERE model_name = 'SVD_vDF'")
assert base.cursor.fetchone()[0] == "SVD_vDF"
model_test.drop()
def test_set_cursor(self, base):
model_test = SVD("SVD_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 = 'SVD_cursor_test'"
)
assert base.cursor.fetchone()[0] == "SVD_cursor_test"
model_test.drop()
def test_drop(self):
current_cursor().execute("DROP MODEL IF EXISTS SVD_model_test_drop")
model_test = SVD("SVD_model_test_drop", )
model_test.fit("public.winequality", ["alcohol", "quality"])
current_cursor().execute(
"SELECT model_name FROM models WHERE model_name = 'SVD_model_test_drop'"
)
assert current_cursor().fetchone()[0] == "SVD_model_test_drop"
model_test.drop()
current_cursor().execute(
"SELECT model_name FROM models WHERE model_name = 'SVD_model_test_drop'"
)
assert current_cursor().fetchone() is None
def test_drop(self, base):
base.cursor.execute("DROP MODEL IF EXISTS SVD_model_test_drop")
model_test = SVD("SVD_model_test_drop", cursor=base.cursor)
model_test.fit("public.winequality", ["alcohol", "quality"])
base.cursor.execute(
"SELECT model_name FROM models WHERE model_name = 'SVD_model_test_drop'"
)
assert base.cursor.fetchone()[0] == "SVD_model_test_drop"
model_test.drop()
base.cursor.execute(
"SELECT model_name FROM models WHERE model_name = 'SVD_model_test_drop'"
)
assert base.cursor.fetchone() is None