Commit bb390ed7 authored by Martin Reinecke's avatar Martin Reinecke
Browse files

tweaks

parent 7b463c37
......@@ -23,12 +23,13 @@ class WienerFilterEnergy(Energy):
The prior signal covariance in harmonic space.
"""
def __init__(self, position, d, R, N, S):
def __init__(self, position, d, R, N, S, inverter):
super(WienerFilterEnergy, self).__init__(position=position)
self.d = d
self.R = R
self.N = N
self.S = S
self._inverter = inverter
def at(self, position):
return self.__class__(position=position, d=self.d, R=self.R, N=self.N,
......@@ -47,7 +48,8 @@ class WienerFilterEnergy(Energy):
@property
@memo
def curvature(self):
return WienerFilterCurvature(R=self.R, N=self.N, S=self.S)
return WienerFilterCurvature(R=self.R, N=self.N, S=self.S,
inverter=self._inverter)
@property
@memo
......
......@@ -78,16 +78,6 @@ def get_weight_configs():
]
def get_hermitian_configs():
np.random.seed(42)
h_0_res_real = np.random.rand(32, 16, 6).astype(np.complex128)
h_0_res_imag = np.random.rand(32, 16, 6).astype(np.complex128)
h_0_x = h_0_res_real + h_0_res_imag * 1j
return [
[h_0_x, h_0_res_real, h_0_res_imag]
]
class LMSpaceInterfaceTests(unittest.TestCase):
@expand([['lmax', int],
['mmax', int],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment