Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_dynamic_length(self):
sample_len = 1024
samples_in = np.random.normal(0, 1, size=sample_len).astype(np.float32)
sample_rate = 16000
augmenter = Compose(
[AddImpulseResponse(ir_path=os.path.join(DEMO_DIR, "ir"), p=1.0)]
)
self.assertEqual(len(samples_in), sample_len)
samples_out = augmenter(samples=samples_in, sample_rate=sample_rate)
self.assertEqual(samples_out.dtype, np.float32)
self.assertGreater(len(samples_out), len(samples_in))
def test_resample(self):
samples = np.zeros((512,), dtype=np.float32)
sample_rate = 16000
augmenter = Compose(
[Resample(min_sample_rate=8000, max_sample_rate=44100, p=1.0)]
)
samples = augmenter(samples=samples, sample_rate=sample_rate)
self.assertEqual(samples.dtype, np.float32)
self.assertLessEqual(
len(samples), math.ceil(len(samples) * 44100 / sample_rate)
)
self.assertGreaterEqual(
len(samples), math.ceil(len(samples) * 8000 / sample_rate)
)
def test_distort(self):
sample_len = 1024
samples_in = np.random.normal(0, 1, size=sample_len).astype(np.float32)
sample_rate = 16000
augmenter = Compose(
[
ClippingDistortion(
min_percentile_threshold=0, max_percentile_threshold=40, p=1.0
)
]
)
samples_out = augmenter(samples=samples_in, sample_rate=sample_rate)
self.assertEqual(samples_out.dtype, np.float32)
self.assertEqual(len(samples_out), sample_len)
self.assertLessEqual(sum(abs(samples_out)), sum(abs(samples_in)))
def test_dynamic_length_with_fade_short_signal(self):
sample_len = 100
samples_in = np.random.normal(0, 1, size=sample_len).astype(np.float32)
sample_rate = 16000
augmenter = Compose(
[TimeMask(min_band_part=0.2, max_band_part=0.5, fade=True, p=1.0)]
)
samples_out = augmenter(samples=samples_in, sample_rate=sample_rate)
self.assertEqual(samples_out.dtype, np.float32)
self.assertEqual(len(samples_out), sample_len)
std_in = np.mean(np.abs(samples_in))
std_out = np.mean(np.abs(samples_out))
self.assertLess(std_out, std_in)
def test_dynamic_length(self):
samples = np.zeros((20,), dtype=np.float32)
sample_rate = 16000
augmenter = Compose(
[
TimeStretch(
min_rate=0.8, max_rate=0.9, leave_length_unchanged=False, p=1.0
)
]
)
samples = augmenter(samples=samples, sample_rate=sample_rate)
self.assertEqual(samples.dtype, np.float32)
self.assertGreater(len(samples), 20)