diff options
| author | historia <historiavg@proton.me> | 2026-09-07 06:47:47 -0400 |
|---|---|---|
| committer | historia <historiavg@proton.me> | 2026-09-07 06:47:47 -0400 |
| commit | 84dd2d068317998f6fb59400c534ef5be6b51b53 (patch) | |
| tree | 025293e9d9229e02960374771ae522d9de2628ce /lib/tests/test_dsp.py | |
| parent | 39b0f2bbed74f6487a41b82501ae3c6799e4b5c4 (diff) | |
| download | producer-main.tar.gz | |
Diffstat (limited to 'lib/tests/test_dsp.py')
| -rw-r--r-- | lib/tests/test_dsp.py | 130 |
1 files changed, 0 insertions, 130 deletions
diff --git a/lib/tests/test_dsp.py b/lib/tests/test_dsp.py deleted file mode 100644 index b9508ab..0000000 --- a/lib/tests/test_dsp.py +++ /dev/null @@ -1,130 +0,0 @@ -import numpy as np -from conftest import band_db, sine, speechish - -from producer import dsp, meters - - -def test_hpf_removes_rumble(sr): - x = sine(40, 3.0, sr, -20.0) + sine(200, 3.0, sr, -20.0) - y = dsp.hpf(x, sr, 80.0) - assert band_db(x, sr, 35, 45) - band_db(y, sr, 35, 45) > 10.0 - assert abs(band_db(x, sr, 190, 210) - band_db(y, sr, 190, 210)) < 0.5 - - -def test_peak_eq_mud_cut(sr): - x = sine(300, 3.0, sr, -20.0) + sine(1000, 3.0, sr, -20.0) - y = dsp.peak_eq(x, sr, 300.0, -3.0, 1.0) - d300 = band_db(x, sr, 280, 320) - band_db(y, sr, 280, 320) - d1k = band_db(x, sr, 950, 1050) - band_db(y, sr, 950, 1050) - assert 2.4 < d300 < 3.6 - assert abs(d1k) < 0.3 - - -def test_shelf(sr): - x = sine(60, 3.0, sr, -20.0) + sine(3000, 3.0, sr, -20.0) - y = dsp.shelf(x, sr, 150.0, 1.5, low=True) - d60 = band_db(y, sr, 50, 70) - band_db(x, sr, 50, 70) - assert 1.1 < d60 < 1.9 - - -def test_compressor_steady_sine(sr): - x = sine(440, 4.0, sr, peak_dbfs=-10.0) - y = dsp.compressor(x, sr, -20.0, 3.0, 15.0, 150.0, 6.0) - in_rms = meters.rms_db(x) - out_rms = meters.rms_db(y) - expected_gr = (1.0 - 1.0 / 3.0) * (in_rms - (-20.0)) - assert abs((in_rms - out_rms) - expected_gr) < 0.4 - - -def test_compressor_quiet_signal_unaffected(sr): - x = sine(440, 4.0, sr, peak_dbfs=-45.0) - y = dsp.compressor(x, sr, -20.0, 3.0, 15.0, 150.0, 6.0) - assert abs(meters.rms_db(x) - meters.rms_db(y)) < 0.1 - - -def test_deesser(sr): - x = sine(1000, 4.0, sr, -20.0) + sine(6500, 4.0, sr, -30.0) - y = dsp.deesser(x, sr, 5500.0, 8000.0, 8.0) - d_sib = band_db(x, sr, 5800, 7200) - band_db(y, sr, 5800, 7200) - d_mid = band_db(x, sr, 900, 1100) - band_db(y, sr, 900, 1100) - assert 1.0 < d_sib < 3.0 - assert abs(d_mid) < 0.4 - - -def test_deesser_bypass(sr): - x = sine(1000, 2.0, sr, -20.0) - y = dsp.deesser(x, sr, 5500.0, 8000.0, 0.0) - assert np.allclose(x, y) - - -def test_expander_attenuates_pauses(sr): - speech = speechish(10.0, sr, level_dbfs=-20.0) - rng = np.random.default_rng(3) - noise = rng.standard_normal(speech.size).astype(np.float64) - noise *= (10 ** (-52.0 / 20.0)) / np.sqrt(np.mean(np.square(noise))) - x = (speech + noise).astype(np.float32) - y = dsp.expander(x, sr, max_drop_db=2.1) - frame = int(0.02 * sr) - nf = x.size // frame - frms_x = np.sqrt(np.mean(np.square(x[: nf * frame].reshape(nf, frame)), axis=1)) - frms_y = np.sqrt(np.mean(np.square(y[: nf * frame].reshape(nf, frame)), axis=1)) - fdb_x = 20 * np.log10(frms_x + 1e-12) - loud = fdb_x > np.percentile(fdb_x, 75) - quiet = fdb_x < np.percentile(fdb_x, 15) - stable_loud = np.convolve(loud.astype(int), np.ones(5, dtype=int), mode="same") == 5 - d_y = 20 * np.log10(frms_y + 1e-12) - assert np.mean(fdb_x[stable_loud] - d_y[stable_loud]) < 0.6 - assert np.mean(fdb_x[quiet] - d_y[quiet]) > 1.5 - - -def test_limit_hits_ceiling(sr): - x = speechish(6.0, sr, level_dbfs=-3.0) - y = dsp.limit(x, sr, -3.0) - assert meters.true_peak_db(y, sr) <= -3.0 + 0.1 - assert meters.rms_db(x) - meters.rms_db(y) < 1.0 - - -def test_limit_below_ceiling_transparent(sr): - x = speechish(6.0, sr, level_dbfs=-20.0) - y = dsp.limit(x, sr, -3.0) - assert np.max(np.abs(x - y)) < 1e-6 - - -def test_resample_roundtrip(sr): - x = speechish(4.0, sr) - y = dsp.resample(dsp.resample(x, sr, 48000), 48000, sr) - assert abs(y.size - x.size) <= 2 - assert abs(meters.rms_db(x) - meters.rms_db(y)) < 0.2 - - -def test_tape_bypass(sr): - x = speechish(3.0, sr) - assert np.array_equal(x, dsp.tape(x, sr, 0.0)) - - -def test_tape_adds_even_harmonics(sr): - x = sine(220, 3.0, sr, -6.0) - y = dsp.tape(x, sr, 1.0) - assert np.all(np.isfinite(y)) - d_h2 = band_db(y, sr, 430, 460) - band_db(x, sr, 430, 460) - d_h3 = band_db(y, sr, 655, 690) - band_db(x, sr, 655, 690) - assert d_h2 > 1.0 - assert d_h3 > 1.0 - assert abs(meters.rms_db(x) - meters.rms_db(y)) < 2.0 - - -def test_soothe_bypass(sr): - x = speechish(3.0, sr) - assert np.array_equal(x, dsp.soothe(x, sr, 0.0)) - - -def test_soothe_reduces_resonant_bands(sr): - x = sine(300, 4.0, sr, -6.0) + sine(3500, 4.0, sr, -6.0) + sine(1000, 4.0, sr, -30.0) - y = dsp.soothe(x, sr, 1.0) - d_low = band_db(x, sr, 260, 350) - band_db(y, sr, 260, 350) - d_harsh = band_db(x, sr, 3200, 3800) - band_db(y, sr, 3200, 3800) - assert 2.0 < d_low <= 3.6 - assert 2.0 < d_harsh <= 5.2 - # frequencies outside the bands stay untouched - d_mid = band_db(x, sr, 900, 1200) - band_db(y, sr, 900, 1200) - assert abs(d_mid) < 0.4 |
