Commit 779cae8b authored by Rouven Lemmerz's avatar Rouven Lemmerz
Browse files

Added wrapper for safe usage

parent a1fcf437
Pipeline #71735 passed with stages
in 25 minutes and 21 seconds
......@@ -197,6 +197,21 @@ def optimize_operator(op):
# Insert trees before leaves
for key in key_list_tree:
op = op.partial_insert(same_tree[key][1].adjoint(same_tree[key][0]))
for key in reversed(key_list_op):
for key in key_list_op:
op = op.partial_insert(same_op[key][1].adjoint(same_op[key][0]))
return op
from copy import deepcopy
from nifty6 import from_random, MultiField
from numpy import allclose
def optimize_operator_safe(op):
op_optimized = deepcopy(op)
op_optimized = optimize_operator(op_optimized)
test_field = from_random('normal', op.domain)
if isinstance(op(test_field), MultiField):
for key in op(test_field).keys():
assert allclose(op(test_field).val[key], op_optimized(test_field).val[key])
else:
assert allclose(op(test_field).val, op_optimized(test_field).val)
return op_optimized
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