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)