From 9d50a20f23f7f4288560f59a98f02124469538cf Mon Sep 17 00:00:00 2001 From: Luca Moretti Date: Tue, 27 Apr 2021 21:51:34 +0200 Subject: [PATCH] Updated test cim --- PyCTBN/tests/structure_graph/test_cim.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/PyCTBN/tests/structure_graph/test_cim.py b/PyCTBN/tests/structure_graph/test_cim.py index 7731a6c..f26d9b8 100644 --- a/PyCTBN/tests/structure_graph/test_cim.py +++ b/PyCTBN/tests/structure_graph/test_cim.py @@ -32,10 +32,10 @@ class TestConditionalIntensityMatrix(unittest.TestCase): c2[i, j] = (c2[i, j] + 1) / (self.state_res_times[i] + 1) c1.compute_cim_coefficients() for i in range(0, len(c1.state_residence_times)): - self.assertTrue(np.isclose(np.sum(c1.cim[i]), 0.0, 1e01, 1e-01)) + self.assertTrue(np.isclose(np.sum(c1.cim[i]), 0.0, 1e01)) for i in range(0, len(self.state_res_times)): for j in range(0, len(self.state_res_times)): - self.assertTrue(np.isclose(c1.cim[i, j], c2[i, j], 1e01, 1e-01)) + self.assertTrue(np.isclose(c1.cim[i, j], c2[i, j], 1e01)) def test_repr(self): c1 = ConditionalIntensityMatrix(self.state_res_times, self.state_transition_matrix)