class TestAudioProcessing(unittest.TestCase):
def setUp(self):
rospack = rospkg.RosPack()
self.auxiliary_files_url = rospack.get_path("rapp_testing_tools") +\
'/test_data'
self.energy_denoise_module = EnergyDenoise()
def tearDown(self):
self.energy_denoise_module = None
self.rospack = None
def test_energy_denois(self):
original_file = self.auxiliary_files_url + "/nai_sample.wav"
denoised_file = self.auxiliary_files_url + "/nai_sample_energy_denoised.wav"
result = self.energy_denoise_module.energyDenoise(\
original_file,\
0.2,\
denoised_file,\
False)
# 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_notExistentFile(self):
original_file = self.auxiliary_files_url + "/not_existent_sample.wav"
denoised_file = self.auxiliary_files_url + "/not_existent_denoised.wav"
result = self.energy_denoise_module.energyDenoise(\
original_file,\
0.2,\
denoised_file,\
False)
self.assertEqual(result, False)
class TestAudioProcessing(unittest.TestCase):
def setUp(self):
rospack = rospkg.RosPack()
self.auxiliary_files_url = rospack.get_path("rapp_testing_tools") +\
'/test_data'
self.energy_denoise_module = EnergyDenoise()
def tearDown(self):
self.energy_denoise_module = None
self.rospack = None
def test_energy_denois(self):
original_file = self.auxiliary_files_url + "/nai_sample.wav"
denoised_file = self.auxiliary_files_url + "/nai_sample_energy_denoised.wav"
result = self.energy_denoise_module.energyDenoise(\
original_file,\
0.2,\
denoised_file,\
False)
# 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_notExistentFile(self):
original_file = self.auxiliary_files_url + "/not_existent_sample.wav"
denoised_file = self.auxiliary_files_url + "/not_existent_denoised.wav"
result = self.energy_denoise_module.energyDenoise(\
original_file,\
0.2,\
denoised_file,\
False)
self.assertEqual(result, False)
def setUp(self):
rospack = rospkg.RosPack()
self.auxiliary_files_url = rospack.get_path("rapp_testing_tools") +\
'/test_data'
self.energy_denoise_module = EnergyDenoise()
def setUp(self):
rospack = rospkg.RosPack()
self.auxiliary_files_url = rospack.get_path("rapp_testing_tools") +\
'/test_data'
self.energy_denoise_module = EnergyDenoise()
