def test_timefrequency_map(): """Compute the power in three specific bands""" # Load dataset: x, y, channel, sf = test_load_dataset() # Define power propreties: poObj = power(sf, [2, 200], x.shape[1], method='wavelet', baseline=(250, 750), norm=3, step=50, width=100, split=(None, None, 20) ) # Compute power : po = poObj.tf(x, f=(2, 200, 20, 10)) # Plot power : poObj.tfplot(po, interpolation='none', interp=(0.1, 1))
def test_power_in_band(): """Compute the power in three specific bands""" # Load dataset: x, y, channel, sf = test_load_dataset() # Define the frequency for the power : f = [[2, 4], [8, 13], [60, 200]] # Define power propreties: poObj = power(sf, f, x.shape[1], method='wavelet', baseline=(250, 750), norm=3, step=50, width=100, split=(None, None, 20) ) # Compute power : po = poObj.get(x) # Plot power : poObj.plot(po, linewidth=2)