matrix_solver.py
No OneTemporary
Actions

Subscribers

None

File Metadata

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

	test = 2
	N = 100
	verb = 0

	solver = MEB(verbosity = verb)
	solver.nAs = 2
	mu = 5.001

	if test == 1:
	solver.As = [sp.spdiags([np.arange(1, 1 + N)], [0], N, N),
	- sp.eye(N)]
	elif test == 2:
	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 = [np.random.randn(N) + 1.j * np.random.randn(N)]

	uh = solver.solve(mu)
	print(solver.norm(uh))
	solver.plot(uh)
	solver.plot(solver.residual(uh, mu), 'res')