36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import unittest
|
|
|
|
import numpy as np
|
|
|
|
from audio_field.program_reference import suppress_program_reference
|
|
|
|
|
|
class ProgramReferenceTests(unittest.TestCase):
|
|
def test_reference_bleed_is_suppressed_from_channel(self):
|
|
sample_rate = 48000
|
|
t = np.arange(sample_rate, dtype=np.float32) / sample_rate
|
|
reference = 0.2 * np.sin(2 * np.pi * 440 * t) + 0.1 * np.sin(2 * np.pi * 1400 * t)
|
|
voice = 0.03 * np.sin(2 * np.pi * 220 * t)
|
|
field = np.zeros((sample_rate, 2), dtype=np.float32)
|
|
field[:, 0] = voice + 0.7 * reference
|
|
field[:, 1] = voice
|
|
|
|
cleaned, reports = suppress_program_reference(
|
|
field,
|
|
reference,
|
|
sample_rate,
|
|
channels=[0],
|
|
nperseg=2048,
|
|
noverlap=1536,
|
|
subtraction_strength=1.0,
|
|
)
|
|
|
|
self.assertEqual((sample_rate, 2), cleaned.shape)
|
|
self.assertLess(reports[0].output_rms, reports[0].input_rms)
|
|
self.assertLess(float(np.sqrt(np.mean((cleaned[:, 0] - voice) ** 2))), 0.01)
|
|
np.testing.assert_allclose(cleaned[:, 1], field[:, 1])
|
|
self.assertGreater(len(reports[0].phase_mapping), 0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|