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_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_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_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_called_without_fit(self):
u = self.data
se_model = SteeringEntropyModel()
with self.assertRaises(Exception):
se_model.get_entropy()
def test_called_without_fit(self):
u = self.data
se_model = SteeringEntropyModel()
with self.assertRaises(Exception):
se_model.get_entropy()
