def test_getSingleWaves(self):
eegData = EEGTableFileUtil().readFile(PATH + "example_32.csv")
eeg = eegData.getColumn("F3")
nEeg = len(eeg)
samplingRate = eegData.getSamplingRate()
waves = self.util.getWaves(eeg, samplingRate)
delta = self.util.getDeltaWaves(eeg, samplingRate)
self.assertEqual(len(delta), nEeg)
self.assertTrue(all([x in waves["delta"] for x in delta]))
theta = self.util.getThetaWaves(eeg, samplingRate)
self.assertEqual(len(theta), nEeg)
self.assertTrue(all([x in waves["theta"] for x in theta]))
alpha = self.util.getAlphaWaves(eeg, samplingRate)
self.assertEqual(len(alpha), nEeg)
self.assertTrue(all([x in waves["alpha"] for x in alpha]))
beta = self.util.getBetaWaves(eeg, samplingRate)
self.assertEqual(len(beta), nEeg)
self.assertTrue(all([x in waves["alpha"] for x in alpha]))
gamma = self.util.getGammaWaves(eeg, samplingRate)
self.assertEqual(len(gamma), nEeg)
self.assertTrue(all([x in waves["gamma"] for x in gamma]))
def test_getWaves(self):
eegData = EEGTableFileUtil().readFile(PATH + "example_32.csv")
eeg = eegData.getColumn("F3")
nEeg = len(eeg)
waves = self.util.getWaves(eeg, eegData.getSamplingRate())
self.assertEqual(len(waves), 5)
for _, wave in waves.iteritems():
self.assertEqual(len(wave), nEeg)
