def test_chroma_extractors():
audio = AudioStim(join(AUDIO_DIR, "barber.wav"))
ext = ChromaSTFTExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (4882, 14)
assert np.isclose(df['onset'][1], 0.01161)
assert np.isclose(df['duration'][0], 0.01161)
assert np.isclose(df['chroma_2'][0], 0.417595)
ext2 = ChromaSTFTExtractor(n_chroma=6, n_fft=1024, hop_length=256)
df = ext2.transform(audio).to_df()
assert df.shape == (9763, 8)
assert np.isclose(df['onset'][1], 0.005805)
assert np.isclose(df['duration'][0], 0.005805)
assert np.isclose(df['chroma_5'][0], 0.732480)
ext = ChromaCQTExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (4882, 14)
assert np.isclose(df['chroma_cqt_2'][0], 0.286443)
ext = ChromaCENSExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (4882, 14)
assert np.isclose(df['chroma_cens_2'][0], 0.217814)
def test_chroma_extractors():
audio = AudioStim(join(AUDIO_DIR, 'barber.wav'))
ext = ChromaSTFTExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (1221, 16)
assert np.isclose(df['onset'][1], 0.04644)
assert np.isclose(df['duration'][0], 0.04644)
assert np.isclose(df['chroma_2'][0], 0.53129)
ext2 = ChromaSTFTExtractor(n_chroma=6, n_fft=1024, hop_length=256)
df = ext2.transform(audio).to_df()
assert df.shape == (2441, 10)
assert np.isclose(df['onset'][1], 0.02322)
assert np.isclose(df['duration'][0], 0.02322)
assert np.isclose(df['chroma_5'][0], 0.86870)
ext = ChromaCQTExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (1221, 16)
assert np.isclose(df['chroma_cqt_2'][0], 0.336481)
ext = ChromaCENSExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (1221, 16)
assert np.isclose(df['chroma_cens_2'][0], 0.136409)
def test_chroma_extractors():
answers = SemVerDict({
'>=0.6.3,<0.8.0': {
'cqt': 0.336481,
'cens': 0.136409
},
'>=0.8.0': {
'cqt': 0.508431,
'cens': 0.102370
}
})
audio = AudioStim(join(AUDIO_DIR, 'barber.wav'))
ext = ChromaSTFTExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (1221, 16)
assert np.isclose(df['onset'][1], 0.04644)
assert np.isclose(df['duration'][0], 0.04644)
assert np.isclose(df['chroma_2'][0], 0.53129)
ext2 = ChromaSTFTExtractor(n_chroma=6, n_fft=1024, hop_length=256)
df = ext2.transform(audio).to_df()
assert df.shape == (2441, 10)
assert np.isclose(df['onset'][1], 0.02322)
assert np.isclose(df['duration'][0], 0.02322)
assert np.isclose(df['chroma_5'][0], 0.86870)
ext = ChromaCQTExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (1221, 16)
assert np.isclose(df['chroma_cqt_2'][0], answers[LIBROSA_VERSION]['cqt'])
ext = ChromaCENSExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (1221, 16)
assert np.isclose(df['chroma_cens_2'][0], answers[LIBROSA_VERSION]['cens'])