Skip to content
Snippets Groups Projects
Commit 024b3725 authored by Philipp Arras's avatar Philipp Arras
Browse files

Improvements for domain checks

parent acb3d258
No related branches found
No related tags found
1 merge request!369More operator checks
Pipeline #63109 passed
Stage: build_docker
Stage: test
Stage: demo_runs
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
nifty5/extra.py
+ 2
− 2
View file @ 024b3725
...@@ -95,7 +95,7 @@ def _actual_domain_check(op, domain_dtype=None, inp=None): ...@@ -95,7 +95,7 @@ def _actual_domain_check(op, domain_dtype=None, inp=None):
assert_(op(inp).domain is op.target) assert_(op(inp).domain is op.target)
def _actual_domain_check_nonlinear(op, loc, target_dtype=np.float64): def _actual_domain_check_nonlinear(op, loc):
assert isinstance(loc, (Field, MultiField)) assert isinstance(loc, (Field, MultiField))
assert_(loc.domain is op.domain) assert_(loc.domain is op.domain)
lin = Linearization.make_var(loc, False) lin = Linearization.make_var(loc, False)
...@@ -112,7 +112,7 @@ def _actual_domain_check_nonlinear(op, loc, target_dtype=np.float64): ...@@ -112,7 +112,7 @@ def _actual_domain_check_nonlinear(op, loc, target_dtype=np.float64):
assert_(reslin.jac.domain is reslin.domain) assert_(reslin.jac.domain is reslin.domain)
assert_(reslin.jac.target is reslin.target) assert_(reslin.jac.target is reslin.target)
_actual_domain_check(reslin.jac, inp=loc) _actual_domain_check(reslin.jac, inp=loc)
_actual_domain_check(reslin.jac.adjoint, domain_dtype=target_dtype) _actual_domain_check(reslin.jac.adjoint, inp=reslin.jac(loc))
def _domain_check(op): def _domain_check(op):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment