Fix bug whereby multiplication by WindowingFunc didn't maintain origi…

…nal dtype
JohnVinyard · May 31, 2018 · 681513c · 681513c
1 parent ea8470c
commit 681513c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/zounds/spectral/sliding_window.py b/zounds/spectral/sliding_window.py
@@ -80,7 +80,7 @@ def __mul__(self, other):
         wdata = self._wdata(size)
         if wdata is None:
             return other
-        return wdata * other
+        return (wdata * other).astype(other.dtype)
 
     def __rmul__(self, other):
         return self.__mul__(other)

diff --git a/zounds/spectral/test_sliding_window.py b/zounds/spectral/test_sliding_window.py
@@ -46,6 +46,12 @@ def test_multiply_1d(self):
         result = wf * samples
         self.assertEqual(samples.shape, result.shape)
 
+    def test_maintains_dtype(self):
+        samples = np.random.random_sample(10).astype(np.float32)
+        wf = OggVorbisWindowingFunc()
+        result = wf * samples
+        self.assertEqual(np.float32, result.dtype)
+
 
 class SlidingWindowTests(unittest2.TestCase):
     def _check(self, samplerate, expected_window_size, expected_step_size):