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