def test_error_minus_9(r=10):
"""Test if MUMPSError -9 is properly caught by increasing memory"""
graphene = honeycomb(norbs=1)
a, b = graphene.sublattices
def circle(pos):
x, y = pos
return x**2 + y**2 < r**2
syst = Builder()
syst[graphene.shape(circle, (0, 0))] = -0.0001
for kind in [((0, 0), b, a), ((0, 1), b, a), ((-1, 1), b, a)]:
syst[HoppingKind(*kind)] = -1
ham = syst.finalized().hamiltonian_submatrix(sparse=True)
# No need to check result, it's enough if no exception is raised
MUMPSContext().factor(ham)
def _test_lu_with_dense(dtype):
rand = _Random()
a = rand.randmat(5, 5, dtype)
bmat = rand.randmat(5, 5, dtype)
bvec = rand.randvec(5, dtype)
ctx = MUMPSContext()
ctx.factor(sp.coo_matrix(a))
xvec = ctx.solve(bvec)
xmat = ctx.solve(bmat)
assert_array_almost_equal(dtype, np.dot(a, xmat), bmat)
assert_array_almost_equal(dtype, np.dot(a, xvec), bvec)
# now "sparse" right hand side
xvec = ctx.solve(sp.csc_matrix(bvec.reshape(5, 1)))
xmat = ctx.solve(sp.csc_matrix(bmat))
assert_array_almost_equal(dtype, np.dot(a, xmat), bmat)
assert_array_almost_equal(dtype, np.dot(a, xvec), bvec.reshape(5, 1))
def _test_lu_with_dense(dtype):
rand = _Random()
a = rand.randmat(5, 5, dtype)
bmat = rand.randmat(5, 5, dtype)
bvec = rand.randvec(5, dtype)
ctx = MUMPSContext()
ctx.factor(sp.coo_matrix(a))
xvec = ctx.solve(bvec)
xmat = ctx.solve(bmat)
assert_array_almost_equal(dtype, np.dot(a, xmat), bmat)
assert_array_almost_equal(dtype, np.dot(a, xvec), bvec)
# now "sparse" right hand side
xvec = ctx.solve(sp.csc_matrix(bvec.reshape(5,1)))
xmat = ctx.solve(sp.csc_matrix(bmat))
assert_array_almost_equal(dtype, np.dot(a, xmat), bmat)
assert_array_almost_equal(dtype, np.dot(a, xvec),
bvec.reshape(5,1))
def test_factor_warning():
"""Test that a warning is raised if factor is asked without analysis."""
a = sp.identity(10, dtype=complex)
with pytest.warns(RuntimeWarning):
MUMPSContext().factor(a, reuse_analysis=True)