matrix.py
90a58bfecbd5master

matrix.py
View Options

			import os
			import numpy as np
			import scipy.sparse as sp
			from rrompy.hfengines.base import MatrixEngineBase as MEB

			def test_deterministic():
			N = 100
			verb = 0

			solver = MEB(verbosity = verb)
			solver.nAs = 2
			mu = 10. + .5j

			solver.As = [sp.spdiags([np.arange(1, 1 + N)], [0], N, N),
			- sp.eye(N)]
			solver.nbs = 1
			solver.bs = {False: [np.exp(1.j * np.linspace(0, -np.pi, N))]}
			uh = solver.solve(mu)
			solver.plot(uh, save = "outDeterministic")

			assert np.isclose(np.linalg.norm(solver.residual(uh, mu)), 1.088e-15)
			filesOut = [x for x in os.listdir(".") if (
			x[-4:] == ".eps" and x[:16] == "outDeterministic")]
			assert len(filesOut) == 1
			os.remove(filesOut[0])

			def test_random():
			N = 100
			verb = 0

			solver = MEB(verbosity = verb)
			solver.nAs = 2
			mu = 1. + .5j

			np.random.seed(420)

			solver.setSolver("SOLVE")
			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.nbs = 1
			solver.bs = {False: [np.random.randn(N) + 1.j * np.random.randn(N)]}

			uh = solver.solve(mu)

			assert np.isclose(np.linalg.norm(solver.residual(uh, mu)), 7.18658e-14)