def test_spectrogram_linearity(self): signal, _ = fmlin(128, 0.1, 0.4) window = kaiser(17, 3 * np.pi) tfr1, _, _ = cohen.spectrogram(signal, n_fbins=64, window=window) tfr2, _, _ = cohen.spectrogram(signal * 2, n_fbins=64, window=window) x = np.sum(np.sum(tfr2)) y = np.sum(np.sum(tfr1)) self.assertEqual(x / y, 4)
def test_spectrogram_reality(self): signal, _ = fmlin(128, 0.1, 0.4) window = kaiser(17, 3 * np.pi) tfr, _, _ = cohen.spectrogram(signal, n_fbins=64, window=window) self.assertTrue(np.all(np.isreal(tfr)))