def test_HawkesSumExpKern_model_settings(self):
"""...Test HawkesSumExpKern setting of parameters
of model
"""
learner = HawkesSumExpKern(self.decays)
np.testing.assert_array_equal(learner.decays, self.decays)
np.testing.assert_array_equal(learner._model_obj.decays, self.decays)
msg = "decays is readonly in HawkesSumExpKern"
with self.assertRaisesRegex(AttributeError, msg):
learner.decays = self.decays + 1
def test_HawkesSumExpKern_model_settings(self):
"""...Test HawkesSumExpKern setting of parameters
of model
"""
# Single baseline
learner = HawkesSumExpKern(self.decays)
np.testing.assert_array_equal(learner.decays, self.decays)
np.testing.assert_array_equal(learner._model_obj.decays, self.decays)
self.assertEqual(learner.n_baselines, 1)
self.assertIsNone(learner.period_length)
# Multiple baselines
n_baselines = 3
period_length = 2.
learner = HawkesSumExpKern(self.decays,
n_baselines=n_baselines,
period_length=period_length)
self.assertEqual(learner.n_baselines, n_baselines)
self.assertEqual(learner._model_obj.n_baselines, n_baselines)
self.assertEqual(learner.period_length, period_length)
self.assertEqual(learner._model_obj.period_length, period_length)
msg = "decays is readonly in HawkesSumExpKern"
with self.assertRaisesRegex(AttributeError, msg):
learner.decays = self.decays + 1
msg = "n_baselines is readonly in HawkesSumExpKern"
with self.assertRaisesRegex(AttributeError, msg):
learner.n_baselines = n_baselines + 1
msg = "period_length is readonly in HawkesSumExpKern"
with self.assertRaisesRegex(AttributeError, msg):
learner.period_length = period_length + 1
msg = "You must fit data before getting estimated baseline"
with self.assertRaisesRegex(ValueError, msg):
learner.baseline
msg = "You must fit data before getting estimated adjacency"
with self.assertRaisesRegex(ValueError, msg):
learner.adjacency
