use correct heaviside def in morlet

aaren · aaren · commit 0286c3ff82f2 · 2015-02-14T16:27:30.000Z
diff --git a/wavelets/wavelets.py b/wavelets/wavelets.py
@@ -98,7 +98,9 @@ def frequency(self, w, s=1.0):
         """
         x = w * s
         # heaviside mock
-        Hw = 0.5 * (np.sign(x) + 1)
+        Hw = np.array(w)
+        Hw[w <= 0] = 0
+        Hw[w > 0] = 1
         return np.pi ** -.25 * Hw * np.exp((-(x - self.w0) ** 2) / 2)
 
     def coi(self, s):
