Commit 7c104388 authored by Christian Marius Lillelund's avatar Christian Marius Lillelund
Browse files

fixed unit tests

parent c234ff23
......@@ -14,7 +14,7 @@ def test_complete_loader():
dl = data_loader.CompleteDataLoader(pt.TESTS_FILES_DIR,
"complete_emb.csv", settings).load_data()
X, y = dl.get_data()
skut.check_X_y(X, y.values.ravel())
skut.check_X_y(X, y)
X, y = dl.prepare_data()
skut.check_X_y(X, y)
......@@ -28,21 +28,21 @@ def test_compliance_loader():
dl = data_loader.ComplianceDataLoader(pt.TESTS_FILES_DIR,
"compliance_emb.csv", settings).load_data()
X, y = dl.get_data()
skut.check_X_y(X, y.values.ravel())
skut.check_X_y(X, y)
def test_fall_loader():
settings = load_config(pt.TESTS_FILES_DIR, "test_config.yaml")
dl = data_loader.FallDataLoader(pt.TESTS_FILES_DIR,
"fall_emb.csv", settings).load_data()
X, y = dl.get_data()
skut.check_X_y(X, y.values.ravel())
skut.check_X_y(X, y)
def test_risk_loader():
settings = load_config(pt.TESTS_FILES_DIR, "test_config.yaml")
dl = data_loader.RiskDataLoader(pt.TESTS_FILES_DIR,
"risk_emb.csv", settings).load_data()
X, y = dl.get_data()
skut.check_X_y(X, y.values.ravel())
skut.check_X_y(X, y)
def test_alarm_loader():
settings = load_config(pt.TESTS_FILES_DIR, "test_config.yaml")
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment