From 7bd846058e7eb60dbc048765086aa30f236accd3 Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Tue, 10 Mar 2020 10:26:29 +0100
Subject: [PATCH] applied bug fix as mentioned in #74
---
src/model_modules/linear_model.py | 2 +-
test/test_model_modules/test_linear_model.py | 8 ++++++++
2 files changed, 9 insertions(+), 1 deletion(-)
create mode 100644 test/test_model_modules/test_linear_model.py
diff --git a/src/model_modules/linear_model.py b/src/model_modules/linear_model.py
index 3d5323e1..933a108c 100644
--- a/src/model_modules/linear_model.py
+++ b/src/model_modules/linear_model.py
@@ -31,7 +31,7 @@ class OrdinaryLeastSquaredModel:
self.y = data_y
def predict(self, data):
- data = sm.add_constant(self.reshape_xarray_to_numpy(data))
+ data = sm.add_constant(self.reshape_xarray_to_numpy(data), has_constant="add")
return np.atleast_2d(self.model.predict(data))
@staticmethod
diff --git a/test/test_model_modules/test_linear_model.py b/test/test_model_modules/test_linear_model.py
new file mode 100644
index 00000000..e4e10e9d
--- /dev/null
+++ b/test/test_model_modules/test_linear_model.py
@@ -0,0 +1,8 @@
+
+from src.model_modules.linear_model import OrdinaryLeastSquaredModel
+
+
+class TestOrdinaryLeastSquareModel:
+
+ def test_constant_input_variable(self):
+ pass
--
GitLab