From 185e4638b6bc653d92bf21dff8f213a8b53b1932 Mon Sep 17 00:00:00 2001 From: huijo Date: Tue, 21 Nov 2023 17:54:55 +0100 Subject: [PATCH] update module name --- tests/test_shape_check.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/tests/test_shape_check.py b/tests/test_shape_check.py index a47abd4..8b2f34b 100644 --- a/tests/test_shape_check.py +++ b/tests/test_shape_check.py @@ -1,9 +1,12 @@ import numpy as np -from constrained_linear_regression.constrained_linear_regression import ConstrainedLinearRegression -from constrained_linear_regression.multi_constrained_linear_regression import MultiConstrainedLinearRegression +from constrainedML.constrained_linear_regression import ConstrainedLinearRegression +from constrainedML.multi_constrained_linear_regression import ( + MultiConstrainedLinearRegression, +) from sklearn.datasets import load_linnerud import pytest + @pytest.mark.xfail(raises=AssertionError) def test_constraint_min_coef(): X, Y = load_linnerud(return_X_y=True) @@ -17,6 +20,7 @@ def test_constraint_min_coef(): model.reset() assert model.global_horizon_count == 0 # Check reset function + @pytest.mark.xfail(raises=AssertionError) def test_constraint_max_coef(): X, Y = load_linnerud(return_X_y=True) @@ -30,6 +34,7 @@ def test_constraint_max_coef(): model.reset() assert model.global_horizon_count == 0 # Check reset function + @pytest.mark.xfail(raises=AssertionError) def test_multi_constraint_min_coef(): X, Y = load_linnerud(return_X_y=True) @@ -43,6 +48,7 @@ def test_multi_constraint_min_coef(): model.reset() assert model.global_horizon_count == 0 # Check reset function + @pytest.mark.xfail(raises=AssertionError) def test_multi_constraint_max_coef(): X, Y = load_linnerud(return_X_y=True)