From 4936346a72668191b02c795eec1c4718b8ad64ff Mon Sep 17 00:00:00 2001
From: Sosnowsky <jlo044@uit.no>
Date: Fri, 23 Feb 2024 18:22:29 +0100
Subject: [PATCH] Fix test

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

diff --git a/tests/test_pulse_shape.py b/tests/test_pulse_shape.py
index 2c2dc95..a79a948 100644
--- a/tests/test_pulse_shape.py
+++ b/tests/test_pulse_shape.py
@@ -47,9 +47,9 @@ def test__get_double_exponential_shape():
     theta = np.array([-1, 0, 1])
     lam = 0.5
     expected_result = np.array([0.13533528, 1.0, 0.13533528])
-    assert np.allclose(
-        BlobShapeImpl._get_double_exponential_shape(theta, lam=lam), expected_result
-    )
+    ps = BlobShapeImpl("2-exp", "2-exp")
+    values = ps.get_blob_shape_perp(theta, lam=lam)
+    assert np.max(np.abs(values - expected_result)) < 1e-5, "Wrong shape"
 
 
 def test__get_secant_shape():