def test_get_task_de_noise_freq_filter(self): self.cfg.task = "de_noise" self.cfg.de_noise = "freq_filter" fx, source_directory, target_directory = get_task(self.cfg) self.assertEqual(fx, DtlnDeNoise) self.assertEqual(source_directory.split("/")[-1], "freq_filter") self.assertEqual( target_directory.split("/")[-1], "de_noised-freq_filter")
def test_get_task_de_noise_reverberate(self): self.cfg.task = "de_noise" self.cfg.de_noise = "reverberate" fx, source_directory, target_directory = get_task(self.cfg) self.assertEqual(fx, DtlnDeNoise) self.assertEqual(source_directory.split("/")[-1], "reverb") self.assertEqual( target_directory.split("/")[-1], "de_noised-reverberate")
def test_get_task_add_reverberate(self): self.cfg.task = "reverberate" fx, source_directory, target_directory = get_task(self.cfg) self.assertEqual(fx, Reverberate) self.assertEqual(source_directory.split("/")[-1], "test-clean") self.assertEqual(target_directory.split("/")[-1], "reverb")
def test_get_task_add_freq_filter(self): self.cfg.task = "freq_filter" fx, source_directory, target_directory = get_task(self.cfg) self.assertEqual(fx, FreqFilter) self.assertEqual(source_directory.split("/")[-1], "test-clean") self.assertEqual(target_directory.split("/")[-1], "freq_filter")
def test_get_task_add_noise(self): self.cfg.task = "add_noise" fx, source_directory, target_directory = get_task(self.cfg) self.assertEqual(fx, AddNoise) self.assertEqual(source_directory.split("/")[-1], "test-clean") self.assertEqual(target_directory.split("/")[-1], "noise")