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")