Mercurial > hg > plosone_underreview

comparison tests/test_OPMellin.py @ 85:9b2beffa5fdd branch-tests

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
test updates
author Maria Panteli <m.x.panteli@gmail.com>
date Tue, 26 Sep 2017 21:19:19 +0100
parents 0e70021f251e
children
comparison
equal deleted inserted replaced
84:027945e93211 85:9b2beffa5fdd
6 """ 6 """
7 7
8 import pytest 8 import pytest
9 9
10 import numpy as np 10 import numpy as np
11 import os
11 12
12 import scripts.OPMellin as OPMellin 13 import scripts.OPMellin as OPMellin
13 14
14 15
15 opm = OPMellin.OPMellin() 16 opm = OPMellin.OPMellin()
16 17 TEST_AUDIO_FILE = os.path.join(os.path.dirname(__file__), os.path.pardir,
18 'data', 'sample_dataset', 'Audio', 'mel_1_2_1.wav')
17 19
18 def test_load_audiofile(): 20 def test_load_audiofile():
19 audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' 21 audiofile = TEST_AUDIO_FILE
20 opm.load_audiofile(audiofile, segment=False) 22 opm.load_audiofile(audiofile, segment=False)
21 assert opm.y is not None and opm.sr is not None 23 assert opm.y is not None and opm.sr is not None
22 24
23 25
24 def test_mel_spectrogram(): 26 def test_mel_spectrogram():
25 audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' 27 audiofile = TEST_AUDIO_FILE
26 opm.load_audiofile(audiofile, segment=False) 28 opm.load_audiofile(audiofile, segment=False)
27 opm.mel_spectrogram(y=opm.y, sr=opm.sr) 29 opm.mel_spectrogram(y=opm.y, sr=opm.sr)
28 # assume 40 mel bands 30 # assume 40 mel bands
29 assert opm.melspec.shape[0] == 40 31 assert opm.melspec.shape[0] == 40
30 32
31 33
32 def test_post_process_spec(): 34 def test_post_process_spec():
33 audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' 35 audiofile = TEST_AUDIO_FILE
34 opm.load_audiofile(audiofile, segment=False) 36 opm.load_audiofile(audiofile, segment=False)
35 opm.mel_spectrogram(y=opm.y, sr=opm.sr) 37 opm.mel_spectrogram(y=opm.y, sr=opm.sr)
36 melspec = opm.melspec 38 melspec = opm.melspec
37 opm.post_process_spec(melspec=melspec) 39 opm.post_process_spec(melspec=melspec)
38 proc_melspec = opm.melspec 40 proc_melspec = opm.melspec
39 assert melspec.shape == proc_melspec.shape 41 assert melspec.shape == proc_melspec.shape
40 42
41 43
42 def test_onset_patterns_n_frames(): 44 def test_onset_patterns_n_frames():
43 audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' 45 audiofile = TEST_AUDIO_FILE
44 opm.load_audiofile(audiofile, segment=False) 46 opm.load_audiofile(audiofile, segment=False)
45 opm.mel_spectrogram(y=opm.y, sr=opm.sr) 47 opm.mel_spectrogram(y=opm.y, sr=opm.sr)
46 opm.onset_patterns(melspec=opm.melspec, melsr=opm.melsr) 48 opm.onset_patterns(melspec=opm.melspec, melsr=opm.melsr)
47 assert opm.op.shape[2] == np.round(((opm.melspec.shape[1] / opm.melsr) - opm.win2sec) * 2.) 49 assert opm.op.shape[2] == np.round(((opm.melspec.shape[1] / opm.melsr) - opm.win2sec) * 2.)
48 50
49 51
50 def test_onset_patterns_n_bins(): 52 def test_onset_patterns_n_bins():
51 audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' 53 audiofile = TEST_AUDIO_FILE
52 opm.load_audiofile(audiofile, segment=False) 54 opm.load_audiofile(audiofile, segment=False)
53 opm.mel_spectrogram(y=opm.y, sr=opm.sr) 55 opm.mel_spectrogram(y=opm.y, sr=opm.sr)
54 opm.onset_patterns(melspec=opm.melspec, melsr=opm.melsr) 56 opm.onset_patterns(melspec=opm.melspec, melsr=opm.melsr)
55 assert opm.op.shape[0] == 40 57 assert opm.op.shape[0] == 40
56 58
57 59
58 def test_post_process_op(): 60 def test_post_process_op():
59 audiofile = 'data/sample_dataset/Audio/mel_1_2_1.wav' 61 audiofile = TEST_AUDIO_FILE
60 opm.load_audiofile(audiofile, segment=False) 62 opm.load_audiofile(audiofile, segment=False)
61 opm.mel_spectrogram(y=opm.y, sr=opm.sr) 63 opm.mel_spectrogram(y=opm.y, sr=opm.sr)
62 opm.onset_patterns(melspec=opm.melspec, melsr=opm.melsr) 64 opm.onset_patterns(melspec=opm.melspec, melsr=opm.melsr)
63 op = opm.op 65 op = opm.op
64 opm.post_process_op() 66 opm.post_process_op()