fix test warnings

Author: Aggrathon

slise/plot.py

@@ -310,8 +310,10 @@ def plot_2d(
     else:
         ax.plot(x_limits, y_limits, "-", color=SLISE_ORANGE, label="Model")
     formula = ""
-    if isinstance(model, float) or len(model) == 1:
-        formula = f"{float(model):.{decimals}f} * {label_x}"
+    if isinstance(model, float):
+        formula = f"{model:.{decimals}f} * {label_x}"
+    elif len(model.flat) == 1:
+        formula = f"{model.flat[0]:.{decimals}f} * {label_x}"
     elif np.abs(model[0]) > 1e-8:
         sign = "-" if model[1] < 0.0 else "+"
         formula = f"{model[0]:.{decimals}f} {sign} {abs(model[1]):.{decimals}f} $\\cdot$ {label_x}"

tests/test_optim.py

@@ -250,9 +250,8 @@ def test_weights():
         regularised_regression(X, Y, 1e-4, 1e-4, weight=w2),
     )
     assert np.allclose(
-        regularised_regression(X2, Y2, 1e-4, 1e-4, weight=w1),
-        regularised_regression(X, Y, 1e-4, 1e-4, weight=w3),
-        atol=1e-4,
+        regularised_regression(X2, Y2, 1e-4, 1e-4, weight=w1, max_iterations=300),
+        regularised_regression(X, Y, 1e-4, 1e-4, weight=w3, max_iterations=300),
     )