update test to sse signature

HugoMVale · Apr 27, 2024 · 12b4e35 · 12b4e35
1 parent 355c2cc
commit 12b4e35
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/math/test_jcr.py b/tests/math/test_jcr.py
@@ -86,18 +86,18 @@ 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))
 
     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