implement unary ops

gdementen · gdementen · commit 17e500a2e1a0 · 2015-01-30T10:57:44.000+01:00
diff --git a/larray/core.py b/larray/core.py
@@ -1399,11 +1399,21 @@ def opmethod(self, other):
     __or__ = _binop('or')
     __ror__ = _binop('ror')
 
+    # element-wise method factory
+    def _unaryop(opname):
+        fullname = '__%s__' % opname
+        super_method = getattr(np.ndarray, fullname)
+
+        def opmethod(self):
+            return LArray(super_method(self.data), self.axes)
+        opmethod.__name__ = fullname
+        return opmethod
+
     # unary ops do not need broadcasting so do not need to be overridden
-    # __neg__ = _unaryop('neg')
-    # __pos__ = _unaryop('pos')
-    # __abs__ = _unaryop('abs')
-    # __invert__ = _unaryop('invert')
+    __neg__ = _unaryop('neg')
+    __pos__ = _unaryop('pos')
+    __abs__ = _unaryop('abs')
+    __invert__ = _unaryop('invert')
 
     def append(self, **kwargs):
         label = kwargs.pop('label', None)
