def test_HawkesKernelSumExp_pickle(self): """...Test pickling ability of HawkesKernelSumExp """ obj = HawkesKernelSumExp(decays=np.arange(1., 2., 0.2), intensities=np.arange(0.3, 2.3, .4)) pickled = pickle.loads(pickle.dumps(obj)) self.assertTrue(str(obj) == str(pickled)) self.assertTrue(np.array_equal(obj.decays, pickled.decays)) self.assertTrue(np.array_equal(obj.intensities, pickled.intensities)) np.testing.assert_array_equal(obj.get_values(self.random_times), obj.get_values(self.random_times))
class Test(unittest.TestCase): def setUp(self): self.decays = np.array([1., 2., 0.2]) self.intensities = np.array([0.3, 4., 2.]) self.hawkes_kernel_sumexp = HawkesKernelSumExp(self.intensities, self.decays) def test_HawkesKernelSumExp_decays(self): """...Test HawkesKernelSumExp decays """ np.testing.assert_array_equal(self.hawkes_kernel_sumexp.decays, self.decays) def test_HawkesKernelSumExp_intensities(self): """...Test HawkesKernelSumExp intensities """ np.testing.assert_array_equal(self.hawkes_kernel_sumexp.intensities, self.intensities) def test_HawkesKernelSumExp_n_decays(self): """...Test HawkesKernelSumExp decay """ self.assertEqual(self.hawkes_kernel_sumexp.n_decays, len(self.decays)) def test_HawkesKernelSumExp_str(self): """...Test HawkesKernelSumExp string representation """ self.assertEqual( str(self.hawkes_kernel_sumexp), "0.3 * 1 * exp(- 1 * t) + 4 * 2 * exp(- 2 * t) + " "2 * 0.2 * exp(- 0.2 * t)") self.assertEqual( str(self.hawkes_kernel_sumexp), "0.3 * 1 * exp(- 1 * t) + 4 * 2 * exp(- 2 * t) + " "2 * 0.2 * exp(- 0.2 * t)") self.decays[1] = 0 self.intensities[2] = 0 hawkes_kernel_sumexp = HawkesKernelSumExp(self.intensities, self.decays) self.assertEqual(str(hawkes_kernel_sumexp), "0.3 * 1 * exp(- 1 * t) + 4 + 0") def test_HawkesKernelSumExp_repr(self): """...Test HawkesKernelSumExp string in list representation """ self.assertEqual( str([self.hawkes_kernel_sumexp]), "[0.3*1*exp(-1*t) + 4*2*exp(-2*t) + " "2*0.2*exp(-0.2*t)]") def test_HawkesKernelSumExp_strtex(self): """...Test HawkesKernelSumExp latex string representation """ self.assertEqual(self.hawkes_kernel_sumexp.__strtex__(), "$0.3 e^{- t}$ + $8 e^{-2 t}$ + $0.4 e^{-0.2 t}$")
def test_HawkesKernelSumExp_str(self): """...Test HawkesKernelSumExp string representation """ self.assertEqual(str(self.hawkes_kernel_sumexp), "0.3 * 1 * exp(- 1 * t) + 4 * 2 * exp(- 2 * t) + " "2 * 0.2 * exp(- 0.2 * t)") self.assertEqual(str(self.hawkes_kernel_sumexp), "0.3 * 1 * exp(- 1 * t) + 4 * 2 * exp(- 2 * t) + " "2 * 0.2 * exp(- 0.2 * t)") self.decays[1] = 0 self.intensities[2] = 0 hawkes_kernel_sumexp = HawkesKernelSumExp(self.intensities, self.decays) self.assertEqual(str(hawkes_kernel_sumexp), "0.3 * 1 * exp(- 1 * t) + 4 + 0")
def setUp(self): self.decays = np.array([1., 2., 0.2]) self.intensities = np.array([0.3, 4., 2.]) self.hawkes_kernel_sumexp = HawkesKernelSumExp(self.intensities, self.decays)