def test_fast_expm(self):
mx = np.array([[1, 2], [2, 3]], 'd')
A = sps.csr_matrix(mx)
A, mu, m_star, s, eta = mt.expm_multiply_prep(A)
tol = 1e-6
B = np.array([1, 1], 'd')
expA = mt._custom_expm_multiply_simple_core(A, B, mu, m_star, s, tol,
eta)
nonSq = np.array([[1, 2, 4], [2, 3, 5]], 'd')
N = sps.csr_matrix(nonSq)
with self.assertRaises(ValueError):
mt.expm_multiply_prep(N)
def test_fast_expm(self):
mx = np.array([[1, 2], [2, 3]], 'd')
A = sps.csr_matrix(mx)
A, mu, m_star, s, eta = mt.expm_multiply_prep(A)
tol = 1e-6
B = np.array([1, 1], 'd')
expA = mt._custom_expm_multiply_simple_core(A, B, mu, m_star, s, tol,
eta)
def test_fast_expm(self):
mx = np.array([[1, 2], [2, 3]], 'd')
A = sps.csr_matrix(mx)
A, mu, m_star, s, eta = mt.expm_multiply_prep(A)
tol = 1e-6
B = np.array([1, 1], 'd')
expA = mt._custom_expm_multiply_simple_core(A, B, mu, m_star, s, tol,
eta)
sp_expA = np.inner(spl.expm(mx), B)
self.assertArraysAlmostEqual(expA, sp_expA)
def test_fast_expm_raises_on_non_square(self):
nonSq = np.array([[1, 2, 4], [2, 3, 5]], 'd')
N = sps.csr_matrix(nonSq)
with self.assertRaises(ValueError):
mt.expm_multiply_prep(N)