From e29d3456affb31f52cce9ec6bc0abd0ee296867b Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Thu, 28 Jun 2018 21:12:30 +0200
Subject: [PATCH] Fix test
---
test/test_energies/test_map.py | 9 +++------
1 file changed, 3 insertions(+), 6 deletions(-)
diff --git a/test/test_energies/test_map.py b/test/test_energies/test_map.py
index 1b7ca9226..67c2f317e 100644
--- a/test/test_energies/test_map.py
+++ b/test/test_energies/test_map.py
@@ -67,7 +67,7 @@ class Energy_Tests(unittest.TestCase):
ift.RGSpace([32, 32], distances=.789)],
[ift.Tanh, ift.Exponential, ift.Linear],
[4, 78, 23]))
- def testNonlinearMap(self, space, nonlinearity, seed):
+ def testGaussianEnergy(self, space, nonlinearity, seed):
np.random.seed(seed)
f = nonlinearity()
dim = len(space.shape)
@@ -85,14 +85,11 @@ class Energy_Tests(unittest.TestCase):
n = ift.Field.from_random(domain=space, random_type='normal')
s = ht(ift.makeOp(A)(xi0_var))
R = ift.ScalingOperator(10., space)
- sqrtN = ift.ScalingOperator(1., space)
+ N = ift.ScalingOperator(1., space)
d_model = R(ift.LocalModel(s, nonlinearity()))
d = d_model.value + n
- IC = ift.GradientNormController(iteration_limit=100,
- tol_abs_gradnorm=1e-5)
- energy = ift.NonlinearWienerFilterEnergy(
- d, d_model, sqrtN, IC)
+ energy = ift.GaussianEnergy(d_model, d, N)
if isinstance(nonlinearity, ift.Linear):
ift.extra.check_value_gradient_curvature_consistency(
energy, ntries=10)
--
GitLab