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)