Page MenuHomec4science
Files F60762668

matrix.py
No OneTemporary
Actions

File Metadata

Created
Thu, May 2, 11:13

matrix.py
View Options

# Copyright (C) 2018 by the RROMPy authors
#
# This file is part of RROMPy.
#
# RROMPy is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# RROMPy is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with RROMPy. If not, see <http://www.gnu.org/licenses/>.
#
import numpy as np
import scipy.sparse as sp
from rrompy.hfengines.base import MatrixEngineBase
def test_deterministic():
solver = MatrixEngineBase(verbosity = 0)
N = 100
solver.npar = 1
solver.nAs, solver.nbs = 2, 1
solver.As = [sp.spdiags([np.arange(1, 1 + N)], [0], N, N),
- sp.eye(N)]
solver.bs = [np.exp(1.j * np.linspace(0, -np.pi, N))]
solver.thAs = solver.getMonomialWeights(solver.nAs)
solver.thbs = solver.getMonomialWeights(solver.nbs)
mu = 10. + .5j
uh = solver.solve(mu)[0]
assert np.isclose(solver.norm(solver.residual(mu, uh)[0], dual = True),
1.088e-15, rtol = 1e-1)
assert np.isclose(solver.norm(solver.residual(mu, uh)[0], dual = True),
solver.norm(solver.residual(mu, uh, duality = False)[0],
dual = True, duality = False), rtol = 1e-1)
def test_random():
solver = MatrixEngineBase(verbosity = 0)
N = 100
solver.setSolver("SOLVE")
solver.npar = 1
solver.nAs, solver.nbs = 2, 1
np.random.seed(420)
fftB = np.fft.fft(np.eye(N)) * N**-.5
solver.As = [fftB.dot(np.multiply(np.arange(1, 1 + N), fftB.conj()).T),
- np.eye(N)]
solver.bs = [np.random.randn(N) + 1.j * np.random.randn(N)]
solver.thAs = solver.getMonomialWeights(solver.nAs)
solver.thbs = solver.getMonomialWeights(solver.nbs)
mu = 1. + .5j
uh = solver.solve(mu)[0]
assert np.isclose(solver.norm(solver.residual(mu, uh)[0], dual = True),
7.18658e-14, rtol = 1e-1)

Event Timeline

c4science · Help