def test_fit_baseline_coeffs(self):
u = self.data
se_model = SteeringEntropyModel()
se_model.fit_baseline(u[:2*60*60])
r = np.array([ 1. , -0.34963801, 0.20299164, -0.20005458])
assert_array_almost_equal(se_model.b_pe, r)
def test_fit_baseline_coeffs(self):
u = self.data
se_model = SteeringEntropyModel()
se_model.fit_baseline(u[:2 * 60 * 60])
r = np.array([1., -0.34963801, 0.20299164, -0.20005458])
assert_array_almost_equal(se_model.b_pe, r)
def test_repr_fit(self):
u = self.data
se_model = SteeringEntropyModel()
se_model.fit_baseline(u[:2 * 60 * 60])
se_model2 = eval(repr(se_model))
self.assertAlmostEqual(se_model.resample_fs, se_model2.resample_fs)
self.assertAlmostEqual(se_model.alpha, se_model2.alpha)
self.assertAlmostEqual(se_model.M, se_model2.M)
assert_array_almost_equal(se_model.b_pe, se_model2.b_pe)
assert_array_almost_equal(se_model.pkbas, se_model2.pkbas)
assert_array_almost_equal(se_model.bin_edges, se_model2.bin_edges)
self.assertAlmostEqual(se_model.hbas, se_model2.hbas)
self.assertEqual(repr(se_model.cdf), repr(se_model2.cdf))
def test_pedistplot(self):
u = self.data[-2 * 60 * 60:]
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2 * 60 * 60])
h, fig = se_model.get_entropy(u[-60 * 60:], _pedistplot=True)
fig.savefig('./output/pedist.png', dpi=300)
plt.close('all')
def test_get_entropy(self):
u = self.data
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2 * 60 * 60])
h = se_model.get_entropy(u[-60 * 60:])
self.assertAlmostEqual(h, 2.3165502390766974)
def test_get_entropy_empty_list(self):
u = self.data
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2 * 60 * 60])
h = se_model.get_entropy([])
self.assertAlmostEqual(h, 0.)
def test_get_entropy(self):
u = self.data
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2*60*60])
h = se_model.get_entropy(u[-60*60:])
self.assertAlmostEqual(h, 2.3165502390766974)
def test_pecdf(self):
u = self.data[-2 * 60 * 60:]
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2 * 60 * 60])
fig = se_model.cdf.plot()
fig.savefig('./output/pe_cdf.png')
plt.close('all')
def test_pedistplot(self):
u = self.data[-2*60*60:]
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2*60*60])
h, fig = se_model.get_entropy(u[-60*60:], _pedistplot=True)
fig.savefig('./output/pedist.png', dpi=300)
plt.close('all')
def test_pecdf(self):
u = self.data[-2*60*60:]
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2*60*60])
fig = se_model.cdf.plot()
fig.savefig('./output/pe_cdf.png')
plt.close('all')
def test_get_entropy_empty_list(self):
u = self.data
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2*60*60])
h = se_model.get_entropy([])
self.assertAlmostEqual(h, 0.)
def test_repr_fit(self):
u = self.data
se_model = SteeringEntropyModel()
se_model.fit_baseline(u[:2*60*60])
se_model2 = eval(repr(se_model))
self.assertAlmostEqual(se_model.resample_fs, se_model2.resample_fs)
self.assertAlmostEqual(se_model.alpha, se_model2.alpha)
self.assertAlmostEqual(se_model.M, se_model2.M)
assert_array_almost_equal(se_model.b_pe, se_model2.b_pe)
assert_array_almost_equal(se_model.pkbas, se_model2.pkbas)
assert_array_almost_equal(se_model.bin_edges,
se_model2.bin_edges)
self.assertAlmostEqual(se_model.hbas, se_model2.hbas)
self.assertEqual(repr(se_model.cdf), repr(se_model2.cdf))
def test_get_10pctlargeincrease(self):
u = self.data
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2 * 60 * 60])
h = se_model.get_entropy(u[-60 * 60:])
self.assertAlmostEqual(hbas, 2.2739162914073892)
self.assertAlmostEqual(h, 2.3165502390766974)
def test_get_10pctlargeincrease(self):
u = self.data
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2*60*60])
h = se_model.get_entropy(u[-60*60:])
self.assertAlmostEqual(hbas, 2.2739162914073892)
self.assertAlmostEqual(h, 2.3165502390766974)
def test_fit_baseline_resamplefs_adj(self):
u = self.data
se_model = SteeringEntropyModel(resample_fs=5.)
hbas = se_model.fit_baseline(u[:2 * 60 * 60])
self.assertAlmostEqual(hbas, 2.492228125216982)
def test_fit_baseline_m_adj(self):
u = self.data
se_model = SteeringEntropyModel(M=7)
hbas = se_model.fit_baseline(u[:2 * 60 * 60])
self.assertAlmostEqual(hbas, 2.2938478599767125)
def test_fit_baseline_hbas(self):
u = self.data
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2 * 60 * 60])
self.assertAlmostEqual(hbas, 2.2739162914073892)
def test_fit_baseline_m_adj(self):
u = self.data
se_model = SteeringEntropyModel(M=7)
hbas = se_model.fit_baseline(u[:2*60*60])
self.assertAlmostEqual(hbas, 2.2938478599767125)
def test_baseline_tsplot(self):
u = self.data[-2*60*60:]
se_model = SteeringEntropyModel()
h,fig = se_model.fit_baseline(u, _tsplot=True)
fig.savefig('./output/baseline_ts.png', dpi=300)
plt.close('all')
def test_baseline_tsplot(self):
u = self.data[-2 * 60 * 60:]
se_model = SteeringEntropyModel()
h, fig = se_model.fit_baseline(u, _tsplot=True)
fig.savefig('./output/baseline_ts.png', dpi=300)
plt.close('all')
def test_fit_baseline_hbas(self):
u = self.data
se_model = SteeringEntropyModel()
hbas = se_model.fit_baseline(u[:2*60*60])
self.assertAlmostEqual(hbas, 2.2739162914073892)
def test_fit_baseline_resamplefs_adj(self):
u = self.data
se_model = SteeringEntropyModel(resample_fs=5.)
hbas = se_model.fit_baseline(u[:2*60*60])
self.assertAlmostEqual(hbas, 2.492228125216982)
def test_fit_baseline_alpha_adj(self):
u = self.data
se_model = SteeringEntropyModel(alpha=0.05)
hbas = se_model.fit_baseline(u[:2 * 60 * 60])
self.assertAlmostEqual(hbas, 1.6410033779603466)
def test_fit_baseline_alpha_adj(self):
u = self.data
se_model = SteeringEntropyModel(alpha=0.05)
hbas = se_model.fit_baseline(u[:2*60*60])
self.assertAlmostEqual(hbas, 1.6410033779603466)
