def test_weighted_phase_lag_index_sets_zero_phase_signals_to_zero(): n_time_samples, n_trials, n_tapers, n_fft_samples, n_signals = (1, 30, 1, 1, 2) fourier_coefficients = np.zeros( (n_time_samples, n_trials, n_tapers, n_fft_samples, n_signals), dtype=np.complex) fourier_coefficients[..., :] = [2 * np.exp(1j * 0), 3 * np.exp(1j * 0)] expected_phase_lag_index = np.zeros((2, 2)) this_Conn = Connectivity(fourier_coefficients=fourier_coefficients) assert np.allclose(this_Conn.weighted_phase_lag_index().squeeze(), expected_phase_lag_index)
def test_weighted_phase_lag_index_is_same_as_phase_lag_index(): n_time_samples, n_trials, n_tapers, n_fft_samples, n_signals = (1, 30, 1, 1, 2) fourier_coefficients = np.zeros( (n_time_samples, n_trials, n_tapers, n_fft_samples, n_signals), dtype=np.complex) fourier_coefficients[..., :] = [ 1 * np.exp(1j * 3 * np.pi / 4), 1 * np.exp(1j * 5 * np.pi / 4) ] this_Conn = Connectivity(fourier_coefficients=fourier_coefficients) assert np.allclose(this_Conn.phase_lag_index(), this_Conn.weighted_phase_lag_index())