diff --git a/tests/test_core.py b/tests/test_core.py index 928be14..1bc93af 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -1063,7 +1063,7 @@ def test_error_transfer_matrix(self): with self.assertRaises(TypeError): ff.error_transfer_matrix(cumulant_function=[1, 2, 3]) - with self.assertRaises(ValueError): + with self.assertRaises((ValueError, np.linalg.LinAlgError)): ff.error_transfer_matrix(cumulant_function=testutil.rng.standard_normal((2, 3, 4))) def test_infidelity_convergence(self):