Mercurial > hg > plosone_underreview
diff tests/test_OPMellin.py @ 85:9b2beffa5fdd branch-tests
test updates
| author | Maria Panteli <m.x.panteli@gmail.com> |
|---|---|
| date | Tue, 26 Sep 2017 21:19:19 +0100 |
| parents | 0e70021f251e |
| children |
line wrap: on
line diff
--- a/tests/test_OPMellin.py Tue Sep 26 20:42:44 2017 +0100 +++ b/tests/test_OPMellin.py Tue Sep 26 21:19:19 2017 +0100 @@ -8,21 +8,23 @@ import pytest import numpy as np +import os import scripts.OPMellin as OPMellin opm = OPMellin.OPMellin() - +TEST_AUDIO_FILE = os.path.join(os.path.dirname(__file__), os.path.pardir, + 'data', 'sample_dataset', 'Audio', 'mel_1_2_1.wav') def test_load_audiofile(): - audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' + audiofile = TEST_AUDIO_FILE opm.load_audiofile(audiofile, segment=False) assert opm.y is not None and opm.sr is not None def test_mel_spectrogram(): - audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' + audiofile = TEST_AUDIO_FILE opm.load_audiofile(audiofile, segment=False) opm.mel_spectrogram(y=opm.y, sr=opm.sr) # assume 40 mel bands @@ -30,7 +32,7 @@ def test_post_process_spec(): - audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' + audiofile = TEST_AUDIO_FILE opm.load_audiofile(audiofile, segment=False) opm.mel_spectrogram(y=opm.y, sr=opm.sr) melspec = opm.melspec @@ -40,7 +42,7 @@ def test_onset_patterns_n_frames(): - audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' + audiofile = TEST_AUDIO_FILE opm.load_audiofile(audiofile, segment=False) opm.mel_spectrogram(y=opm.y, sr=opm.sr) opm.onset_patterns(melspec=opm.melspec, melsr=opm.melsr) @@ -48,7 +50,7 @@ def test_onset_patterns_n_bins(): - audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' + audiofile = TEST_AUDIO_FILE opm.load_audiofile(audiofile, segment=False) opm.mel_spectrogram(y=opm.y, sr=opm.sr) opm.onset_patterns(melspec=opm.melspec, melsr=opm.melsr) @@ -56,7 +58,7 @@ def test_post_process_op(): - audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' + audiofile = TEST_AUDIO_FILE opm.load_audiofile(audiofile, segment=False) opm.mel_spectrogram(y=opm.y, sr=opm.sr) opm.onset_patterns(melspec=opm.melspec, melsr=opm.melsr)