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()