def setUp(self): rospack = rospkg.RosPack() self.auxiliary_files_url = rospack.get_path("rapp_testing_tools") +\ '/test_data' self.sox_denoise_module = SoxDenoise()
class TestAudioProcessing(unittest.TestCase): def setUp(self): rospack = rospkg.RosPack() self.auxiliary_files_url = rospack.get_path("rapp_testing_tools") +\ '/test_data' self.sox_denoise_module = SoxDenoise() def tearDown(self): self.sox_denoise_module = None self.rospack = None def test_realFile(self): original_file = self.auxiliary_files_url + "/nai_sample.wav" denoised_file = self.auxiliary_files_url + "/nai_sample_sox_denoised.wav" user = '******' audio_type = 'nao_wav_1_ch' scale = 0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) # The function thinks the denoising succeded self.assertEqual(result, "true") # Check for the denoised file denoised_exists = os.path.isfile(denoised_file) self.assertEqual(denoised_exists, True) # Check if denoised energy is lower than the initial one samp_freq, signal_orig = wavfile.read(original_file) energy_orig = 0.0 for i in signal_orig: energy_orig += i * 1.0 * i samp_freq, signal_denoised = wavfile.read(denoised_file) energy_denoised = 0.0 for i in signal_denoised: energy_denoised += i * 1.0 * i self.assertGreater(energy_orig, energy_denoised) # erase the denoised file os.remove(denoised_file) def test_notExistentAudioFile(self): original_file = self.auxiliary_files_url + "/not_existent_sample.wav" denoised_file = self.auxiliary_files_url + "/nai_sample_sox_denoised.wav" user = '******' audio_type = 'nao_wav_1_ch' scale = 0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) # The function thinks the denoising succeded self.assertNotEqual(result, "true") def test_notExistentUser(self): original_file = self.auxiliary_files_url + "/nai_sample.wav" denoised_file = self.auxiliary_files_url + "/nai_sample_sox_denoised.wav" user = '******' audio_type = 'nao_wav_1_ch' scale = 0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) # The function thinks the denoising succeded self.assertNotEqual(result, "true") def test_notExistentAudioType(self): original_file = self.auxiliary_files_url + "/nai_sample.wav" denoised_file = self.auxiliary_files_url + "/nai_sample_sox_denoised.wav" user = '******' audio_type = 'nao_wav_7_ch' scale = 0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) self.assertNotEqual(result, "true") def test_negativeScale(self): original_file = self.auxiliary_files_url + "/nai_sample.wav" denoised_file = self.auxiliary_files_url + "/nai_sample_sox_denoised.wav" user = '******' audio_type = 'nao_wav_1_ch' scale = -0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) self.assertNotEqual(result, "true") def test_ogg_type_wav_1_ch_file(self): original_file = self.auxiliary_files_url + "/nao_ogg_d05_a1.ogg" denoised_file = self.auxiliary_files_url + "/nao_ogg_teest.ogg" user = '******' audio_type = 'nao_wav_1_ch' scale = -0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) self.assertNotEqual(result, "true") def test_ogg_type_wav_4_ch_file(self): original_file = self.auxiliary_files_url + "/nao_ogg_d05_a1.ogg" denoised_file = self.auxiliary_files_url + "/nao_ogg_teest.ogg" user = '******' audio_type = 'nao_wav_4_ch' scale = -0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) self.assertNotEqual(result, "true") def test_wav_1_ch_type_ogg_file(self): original_file = self.auxiliary_files_url + "/silence_sample.wav" denoised_file = self.auxiliary_files_url + "/new_file.wav" user = '******' audio_type = 'nao_ogg' scale = -0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) self.assertNotEqual(result, "true") def test_wav_1_ch_type_wav_4_ch_file(self): original_file = self.auxiliary_files_url + "/silence_sample.wav" denoised_file = self.auxiliary_files_url + "/new_file.wav" user = '******' audio_type = 'nao_wav_4_ch' scale = -0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) self.assertNotEqual(result, "true") def test_wav_4_ch_type_ogg_file(self): original_file = self.auxiliary_files_url + "/nao_wav_d05_a1.wav" denoised_file = self.auxiliary_files_url + "/new_file.wav" user = '******' audio_type = 'nao_ogg' scale = -0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) self.assertNotEqual(result, "true") def test_wav_4_ch_type_wav_1_ch_file(self): original_file = self.auxiliary_files_url + "/nao_wav_d05_a1.wav" denoised_file = self.auxiliary_files_url + "/new_file.wav" user = '******' audio_type = 'nao_wav_1_ch' scale = -0.2 result = self.sox_denoise_module.soxDenoise(\ user,\ audio_type,\ original_file,\ denoised_file,\ scale) self.assertNotEqual(result, "true")