Expose BandPassFilter, HighPassFilter and LowPassFilter in the public…

… API
iver56 · Aug 5, 2021 · 528f0ca · 528f0ca
1 parent cbcb8e1
commit 528f0ca
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 33 deletions.
diff --git a/audiomentations/__init__.py b/audiomentations/__init__.py
@@ -6,11 +6,14 @@
     AddImpulseResponse,
     ApplyImpulseResponse,
     AddShortNoises,
+    BandPassFilter,
     Clip,
     ClippingDistortion,
     FrequencyMask,
     Gain,
+    HighPassFilter,
     LoudnessNormalization,
+    LowPassFilter,
     Mp3Compression,
     Normalize,
     PitchShift,

diff --git a/demo/demo.py b/demo/demo.py
@@ -24,8 +24,6 @@
     Mp3Compression,
     LoudnessNormalization,
     Trim,
-)
-from audiomentations.augmentations.transforms import (
     LowPassFilter,
     HighPassFilter,
     BandPassFilter,

diff --git a/tests/test_band_pass_filter.py b/tests/test_band_pass_filter.py
@@ -4,31 +4,28 @@
 
 import matplotlib.pyplot as plt
 
-from audiomentations.augmentations.transforms import BandPassFilter
+from audiomentations import BandPassFilter
 
 DEBUG = False
 
+
 class TestBandPassFilter(unittest.TestCase):
     def test_band_pass_filter(self):
         sample_rate = 16000
-        t = .25                # signal duration in sec
-        f = 500                # signal frequency in Hz
+        t = 0.25  # signal duration in sec
+        f = 500  # signal frequency in Hz
         samples = np.arange(t * f, dtype=np.float32) / sample_rate
         samples = np.sin(2 * np.pi * f * samples)
 
-        augment = BandPassFilter(min_center_freq=100, 
-                                 max_center_freq=5000,
-                                 min_q=1.0,
-                                 max_q=2.0,
-                                 p=1.0)
-        processed_samples = augment(
-            samples=samples, sample_rate=sample_rate
+        augment = BandPassFilter(
+            min_center_freq=100, max_center_freq=5000, min_q=1.0, max_q=2.0, p=1.0
         )
+        processed_samples = augment(samples=samples, sample_rate=sample_rate)
 
         self.assertEqual(processed_samples.shape, samples.shape)
         self.assertEqual(processed_samples.dtype, np.float32)
-        
-        if DEBUG:        
+
+        if DEBUG:
             plt.plot(samples)
-            plt.plot(processed_samples, '-.')
+            plt.plot(processed_samples, "-.")
             plt.show()
diff --git a/tests/test_high_pass_filter.py b/tests/test_high_pass_filter.py
@@ -4,27 +4,26 @@
 
 import matplotlib.pyplot as plt
 
-from audiomentations.augmentations.transforms import HighPassFilter
+from audiomentations import HighPassFilter
 
 DEBUG = False
 
+
 class TestHighPassFilter(unittest.TestCase):
     def test_high_pass_filter(self):
         sample_rate = 16000
-        t = .25                # signal duration in sec
-        f = 500                # signal frequency in Hz
+        t = 0.25  # signal duration in sec
+        f = 500  # signal frequency in Hz
         samples = np.arange(t * f, dtype=np.float32) / sample_rate
         samples = np.sin(2 * np.pi * f * samples)
 
         augment = HighPassFilter(min_cutoff_freq=100, max_cutoff_freq=200, p=1.0)
-        processed_samples = augment(
-            samples=samples, sample_rate=sample_rate
-        )
+        processed_samples = augment(samples=samples, sample_rate=sample_rate)
 
         self.assertEqual(processed_samples.shape, samples.shape)
         self.assertEqual(processed_samples.dtype, np.float32)
-        
-        if DEBUG:        
+
+        if DEBUG:
             plt.plot(samples)
-            plt.plot(processed_samples, '-.')
+            plt.plot(processed_samples, "-.")
             plt.show()
diff --git a/tests/test_low_pass_filter.py b/tests/test_low_pass_filter.py
@@ -4,27 +4,26 @@
 
 import matplotlib.pyplot as plt
 
-from audiomentations.augmentations.transforms import LowPassFilter
+from audiomentations import LowPassFilter
 
 DEBUG = False
 
+
 class TestLowPassFilter(unittest.TestCase):
     def test_low_pass_filter(self):
         sample_rate = 16000
-        t = .25                # signal duration in sec
-        f = 500                # signal frequency in Hz
+        t = 0.25  # signal duration in sec
+        f = 500  # signal frequency in Hz
         samples = np.arange(t * f, dtype=np.float32) / sample_rate
         samples = np.sin(2 * np.pi * f * samples)
 
         augment = LowPassFilter(min_cutoff_freq=100, max_cutoff_freq=200, p=1.0)
-        processed_samples = augment(
-            samples=samples, sample_rate=sample_rate
-        )
+        processed_samples = augment(samples=samples, sample_rate=sample_rate)
 
         self.assertEqual(processed_samples.shape, samples.shape)
         self.assertEqual(processed_samples.dtype, np.float32)
-        
-        if DEBUG:        
+
+        if DEBUG:
             plt.plot(samples)
-            plt.plot(processed_samples, '-.')
+            plt.plot(processed_samples, "-.")
             plt.show()