def test_coherence_partial():
""" Test partial coherence"""
x = np.sin(t) + np.sin(2*t) + np.sin(3*t) + np.random.rand(t.shape[-1])
y = x + np.random.rand(t.shape[-1])
z = y + np.random.rand(t.shape[-1])
for method in methods:
if (method is None) or method['this_method']=='welch':
f,c = tsa.coherence_partial(np.vstack([x,y]),z,csd_method=method)
npt.assert_array_almost_equal(c[0,1],c[1,0].conjugate())
def test_coherence_partial():
""" Test partial coherence"""
x = np.sin(t) + np.sin(2 * t) + np.sin(3 * t) + np.random.rand(t.shape[-1])
y = x + np.random.rand(t.shape[-1])
z = y + np.random.rand(t.shape[-1])
for method in methods:
if (method is None) or method['this_method'] == 'welch':
f, c = tsa.coherence_partial(np.vstack([x, y]),
z,
csd_method=method)
npt.assert_array_almost_equal(c[0, 1], c[1, 0].conjugate())
def test_coherence_partial():
""" Test partial coherence"""
t = np.linspace(0,16*np.pi,1024)
x = np.sin(t) + np.sin(2*t) + np.sin(3*t) + np.random.rand(t.shape[-1])
y = x + np.random.rand(t.shape[-1])
z = x + np.random.rand(t.shape[-1])
method = {"this_method":'welch',
"NFFT":256,
"Fs":2*np.pi}
f,c = tsa.coherence_partial(np.vstack([x,y]),z,csd_method=method)
f_theoretical = ut.get_freqs(method['Fs'],method['NFFT'])
npt.assert_array_almost_equal(f,f_theoretical)
npt.assert_array_almost_equal(c[0,1],c[1,0])
