iframe-proxy | Sunbelt Computer Software

53 lines (38 loc) · 1.32 KB

# coding: utf-8
    DeepAffects
    OpenAPI spec version: v1
from __future__ import absolute_import
import unittest
import deepaffects
from deepaffects.models.audio import Audio
from .test_base_setup import DIR
class TestDenoiseApi(unittest.TestCase):
    """ DenoiseApi unit test stubs """
    @staticmethod
    def sdr(clean, reconstructed):
        # TODO: Implement SDR here
        return 5.1
    def setUp(self):
        deepaffects.configuration.api_key['apikey'] = os.environ['DEEPAFFECTS_API_KEY']
        self.api = deepaffects.DenoiseApi()
    def tearDown(self):
        pass
    def test_async_denoise_audio(self):
        """
        Test case for async_denoise_audio
        Denoise an audio file
        """
        test_noisy_audio = os.path.normpath(os.path.join(DIR, "data/noisy.wav"))
        test_clean_audio = os.path.normpath(os.path.join(DIR, "data/clean.wav"))
        test_reconstructed_audio = os.path.normpath(os.path.join(DIR, "data/reconstructed.wav"))
        body = Audio.from_file(file_name=test_noisy_audio)
        api_response = self.api.sync_denoise_audio(body=body)
        api_response.to_file(test_reconstructed_audio)
        self.assertTrue(TestDenoiseApi.sdr(test_clean_audio, test_reconstructed_audio) > 5)
        pass
if __name__ == '__main__':
    unittest.main()

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

test_denoise_api.py

Latest commit

History

test_denoise_api.py

File metadata and controls