def test_complexity_sanity():
signal = np.cos(np.linspace(start=0, stop=30, num=1000))
# Entropy
assert np.allclose(nk.entropy_fuzzy(signal),
nk.entropy_sample(signal, fuzzy=True),
atol=0.000001)
# Fractal
assert np.allclose(nk.fractal_dfa(signal, windows=np.array([4, 8, 12,
20])),
2.1009048365682133,
atol=0.000001)
assert np.allclose(nk.fractal_dfa(signal),
1.957966586191164,
atol=0.000001)
assert np.allclose(nk.fractal_dfa(signal, multifractal=True),
1.957966586191164,
atol=0.000001)
assert np.allclose(nk.fractal_correlation(signal),
0.7884473170763334,
atol=0.000001)
assert np.allclose(nk.fractal_correlation(signal, r="nolds"),
nolds.corr_dim(signal, 2),
atol=0.0001)
def test_complexity_vs_Python():
signal = np.cos(np.linspace(start=0, stop=30, num=100))
# Shannon
shannon = nk.entropy_shannon(signal)
# assert scipy.stats.entropy(shannon, pd.Series(signal).value_counts())
assert np.allclose(shannon - pyentrp.shannon_entropy(signal), 0)
# Approximate
assert np.allclose(nk.entropy_approximate(signal), 0.17364897858477146)
assert np.allclose(
nk.entropy_approximate(
signal, dimension=2, r=0.2 * np.std(signal, ddof=1)) -
entropy_app_entropy(signal, 2), 0)
assert nk.entropy_approximate(
signal, dimension=2,
r=0.2 * np.std(signal, ddof=1)) != pyeeg_ap_entropy(
signal, 2, 0.2 * np.std(signal, ddof=1))
# Sample
assert np.allclose(
nk.entropy_sample(signal, dimension=2, r=0.2 * np.std(signal, ddof=1))
- entropy_sample_entropy(signal, 2), 0)
assert np.allclose(
nk.entropy_sample(signal, dimension=2, r=0.2) -
nolds.sampen(signal, 2, 0.2), 0)
assert np.allclose(
nk.entropy_sample(signal, dimension=2, r=0.2) -
entro_py_sampen(signal, 2, 0.2, scale=False), 0)
assert np.allclose(
nk.entropy_sample(signal, dimension=2, r=0.2) -
pyeeg_samp_entropy(signal, 2, 0.2), 0)
# import sampen
# sampen.sampen2(signal[0:300], mm=2, r=r)
assert nk.entropy_sample(signal,
dimension=2, r=0.2) != pyentrp.sample_entropy(
signal, 2, 0.2)[1]
assert nk.entropy_sample(
signal, dimension=2,
r=0.2 * np.sqrt(np.var(signal))) != MultiscaleEntropy_sample_entropy(
signal, 2, 0.2)[0.2][2]
# MSE
# assert nk.entropy_multiscale(signal, 2, 0.2*np.sqrt(np.var(signal))) != np.trapz(MultiscaleEntropy_mse(signal, [i+1 for i in range(10)], 2, 0.2, return_type="list"))
# assert nk.entropy_multiscale(signal, 2, 0.2*np.std(signal, ddof=1)) != np.trapz(pyentrp.multiscale_entropy(signal, 2, 0.2, 10))
# Fuzzy
assert np.allclose(
nk.entropy_fuzzy(signal, dimension=2, r=0.2, delay=1) -
entro_py_fuzzyen(signal, 2, 0.2, 1, scale=False), 0)
# DFA
assert nk.fractal_dfa(signal, windows=np.array([
4, 8, 12, 20
])) != nolds.dfa(signal, nvals=[4, 8, 12, 20], fit_exp="poly")