matrix_fft.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	# 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
	from rrompy.hfengines.base import MatrixEngineBase

	class matrixFFT(MatrixEngineBase):
	def __init__(self):
	super().__init__(verbosity = 0)
	N = 100
	np.random.seed(420)
	self.setSolver("SOLVE")
	self.npar = 1
	self.nAs, self.nbs = 2, 1
	self.mu0 = 0.
	fftB = np.fft.fft(np.eye(N)) * N**-.5
	self.As = [fftB.dot(np.multiply(np.arange(1, 1 + N), fftB.conj()).T),
	- np.eye(N)]
	self.bs = [np.random.randn(N) + 1.j * np.random.randn(N)]
	self.thAs = self.getMonomialWeights(self.nAs)
	self.thbs = self.getMonomialWeights(self.nbs)