From cf3fc29ef35b51a9d10a9fa8254ac08cb1c5b4ae Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Wed, 6 Nov 2019 12:09:50 +0100
Subject: [PATCH] Test _SlowFieldAdapter
---
nifty5/operators/simple_linear_operators.py | 4 +---
test/test_operators/test_adjoint.py | 7 +++++++
2 files changed, 8 insertions(+), 3 deletions(-)
diff --git a/nifty5/operators/simple_linear_operators.py b/nifty5/operators/simple_linear_operators.py
index ba7c3eecc..f8c7c75d9 100644
--- a/nifty5/operators/simple_linear_operators.py
+++ b/nifty5/operators/simple_linear_operators.py
@@ -187,9 +187,7 @@ class _SlowFieldAdapter(LinearOperator):
self._check_input(x, mode)
if isinstance(x, MultiField):
return x[self._name]
- else:
- return MultiField.from_dict({self._name: x},
- domain=self._tgt(mode))
+ return MultiField.from_dict({self._name: x}, domain=self._tgt(mode))
def __repr__(self):
return '_SlowFieldAdapter'
diff --git a/test/test_operators/test_adjoint.py b/test/test_operators/test_adjoint.py
index f89fdedcd..9e37ee82a 100644
--- a/test/test_operators/test_adjoint.py
+++ b/test/test_operators/test_adjoint.py
@@ -319,3 +319,10 @@ def testPartialExtractor(seed):
tgt = ift.MultiDomain.make(tgt)
op = ift.PartialExtractor(dom, tgt)
ift.extra.consistency_check(op)
+
+
+@pmp('seed', [12, 3])
+def testSlowFieldAdapter(seed):
+ dom = {'a': ift.RGSpace(1), 'b': ift.RGSpace(2)}
+ op = ift.operators.simple_linear_operators._SlowFieldAdapter(dom, 'a')
+ ift.extra.consistency_check(op)
--
GitLab