ENH: ma: use domains also in _DomainedBinaryOperation.__call__ (previously they were used only in __array_wrap__, which wouldn't be called from __div__ et al. which call the domained operations directly)

pv · pv · commit 3ddc421aab6e · 2010-02-21T11:58:38.000Z
diff --git a/numpy/ma/core.py b/numpy/ma/core.py
@@ -1068,6 +1068,10 @@ def __call__(self, a, b, *args, **kwargs):
         m = ~umath.isfinite(result)
         m |= ma
         m |= mb
+        # Apply the domain
+        domain = ufunc_domain.get(self.f, None)
+        if domain is not None:
+            m |= filled(domain(da, db), True)
         # Take care of the scalar case first
         if (not m.ndim):
             if m:
