From 12b4e35f97d1f7041c3ed285bed01a3e0efcdac5 Mon Sep 17 00:00:00 2001
From: HugoMVale <57530119+HugoMVale@users.noreply.github.com>
Date: Sat, 27 Apr 2024 21:11:05 +0200
Subject: [PATCH] update test to sse signature

---
 tests/math/test_jcr.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/math/test_jcr.py b/tests/math/test_jcr.py
index fbeda53..233f1fa 100644
--- a/tests/math/test_jcr.py
+++ b/tests/math/test_jcr.py
@@ -86,7 +86,7 @@ def model(x, beta):
         return beta[0]*x**2 + beta[1]*x**5
 
     ndata = 101
-    X = np.linspace(0, 1., ndata)
+    X = np.linspace(0., 1., ndata)
     beta = (0.15, -0.05)
     Y = model(X, beta) + np.random.normal(0., 0.05, len(X))
 
@@ -94,10 +94,10 @@ def sse(beta) -> float:
         Ye = model(X, beta)
         return sum((Ye - Y)**2)
 
-    sol = minimize(sse, (-1., 1))
+    sol = minimize(sse, (-1., 1.))
     beta_est = sol.x
 
     jcr = confidence_region(beta_est,
-                            sse=lambda x, y: sse((x, y)),
+                            sse=sse,
                             ndata=ndata)
     assert len(jcr[0]) > 1