Fix flake8 for test_spectrogram.py