diff --git a/test/test_operators/test_adjoint.py b/test/test_operators/test_adjoint.py index 459952de6cdb770f9846301d70bb02a810825edb..e3fdddacef4de3585793bae1d8b8b9641b805bfc 100644 --- a/test/test_operators/test_adjoint.py +++ b/test/test_operators/test_adjoint.py @@ -326,3 +326,20 @@ 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) + +@pmp('sp1', [0, 2]) +@pmp('sp2', [1]) +@pmp('seed', [12, 3]) +def testSwitchSpacesOperator(sp1, sp2, seed): + with ift.random.Context(seed): + dom1 = ift.RGSpace(1) + dom2 = ift.RGSpace((2, 2)) + dom3 = ift.RGSpace(3) + dom = ift.DomainTuple.make([dom1, dom2, dom3]) + op = ift.SwitchSpacesOperator(dom, sp1, sp2) + + tgt = list(dom) + tgt[sp1] = dom[sp2] + tgt[sp2] = dom[sp1] + assert op.target == ift.DomainTuple.make(tgt) + ift.extra.consistency_check(op)