def test_spectral_extractors():
audio = AudioStim(join(AUDIO_DIR, 'barber.wav'))
ext = SpectralCentroidExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (1221, 5)
assert np.isclose(df['onset'][1], 0.04644)
assert np.isclose(df['duration'][0], 0.04644)
assert np.isclose(df['spectral_centroid'][0], 1144.98145)
ext2 = SpectralCentroidExtractor(n_fft=1024, hop_length=256)
df = ext2.transform(audio).to_df()
assert df.shape == (2441, 5)
assert np.isclose(df['onset'][1], 0.02322)
assert np.isclose(df['duration'][0], 0.02322)
assert np.isclose(df['spectral_centroid'][0], 866.20176)
ext = SpectralBandwidthExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (1221, 5)
assert np.isclose(df['spectral_bandwidth'][0], 1172.96090)
ext = SpectralContrastExtractor(fmin=100.0)
df = ext.transform(audio).to_df()
assert df.shape == (1221, 11)
assert np.isclose(df['spectral_contrast_band_4'][0], 25.637166)
ext = SpectralRolloffExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (1221, 5)
assert np.isclose(df['spectral_rolloff'][0], 2492.46826)
def test_spectral_extractors():
audio = AudioStim(join(AUDIO_DIR, "barber.wav"))
ext = SpectralCentroidExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (4882, 3)
assert np.isclose(df['onset'][1], 0.01161)
assert np.isclose(df['duration'][0], 0.01161)
assert np.isclose(df['spectral_centroid'][0], 817.53095)
ext2 = SpectralCentroidExtractor(n_fft=1024, hop_length=256)
df = ext2.transform(audio).to_df()
assert df.shape == (9763, 3)
assert np.isclose(df['onset'][1], 0.005805)
assert np.isclose(df['duration'][0], 0.005805)
assert np.isclose(df['spectral_centroid'][0], 1492.00515)
ext = SpectralBandwidthExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (4882, 3)
assert np.isclose(df['spectral_bandwidth'][0], 1056.66227)
ext = SpectralContrastExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (4882, 9)
assert np.isclose(df['spectral_contrast_band_4'][0], 25.09001)
ext = SpectralRolloffExtractor()
df = ext.transform(audio).to_df()
assert df.shape == (4882, 3)
assert np.isclose(df['spectral_rolloff'][0], 1550.39063)