Author: Matti Picus <[email protected]>
Branch: ndarray-round
Changeset: r64987:737e0485a18b
Date: 2013-06-25 22:33 +0300
http://bitbucket.org/pypy/pypy/changeset/737e0485a18b/
Log: implement round for scalars
diff --git a/pypy/module/micronumpy/interp_boxes.py
b/pypy/module/micronumpy/interp_boxes.py
--- a/pypy/module/micronumpy/interp_boxes.py
+++ b/pypy/module/micronumpy/interp_boxes.py
@@ -235,6 +235,11 @@
w_values = space.newtuple([self])
return convert_to_array(space, w_values)
+ @unwrap_spec(decimals=int)
+ def descr_round(self, space, decimals=0):
+ v = self.convert_to(self.get_dtype(space))
+ return self.get_dtype(space).itemtype.round(v, decimals)
+
class W_BoolBox(W_GenericBox, PrimitiveBox):
descr__new__, _get_dtype, descr_reduce = new_dtype_getter("bool")
@@ -501,6 +506,7 @@
any = interp2app(W_GenericBox.descr_any),
all = interp2app(W_GenericBox.descr_all),
ravel = interp2app(W_GenericBox.descr_ravel),
+ round = interp2app(W_GenericBox.descr_round),
)
W_BoolBox.typedef = TypeDef("bool_", W_GenericBox.typedef,
_______________________________________________
pypy-commit mailing list
[email protected]
http://mail.python.org/mailman/listinfo/pypy-commit
