test_essentia_analyzer.py

from __future__ import absolute_import
import unittest
import settings
import sound_file
import time
import essentia_analyzer
import experiment


class TestEssentiaAnalyzer(unittest.TestCase):
    def setUp(self):
        settings.INPUT_DIRECTORY = 'test_audio'
        self.sounds = [
            sound_file.SoundFile('drums.wav'),
            sound_file.SoundFile('noise.wav')
        ]
        self.essentia_analyzer = essentia_analyzer.EssentiaAnalyzer(['spectral_centroid'])
        experiment.Experiment.folder_name = 'test'

    def test_analysis(self):
        self.start_time = time.time()

        self.essentia_analyzer.analyze_multiple(self.sounds)

        for sound in self.sounds:
            self.assertTrue('spectral_centroid' in sound.analysis['series'])
            self.assertGreaterEqual(
                len(sound.analysis['series']['spectral_centroid']),
                100
            )

        print("Execution time: {0} seconds".format(
            time.time() - self.start_time)
        )


if __name__ == '__main__':
    unittest.main()