Author: Justin Peel <notmuchtot...@gmail.com> Branch: numpy-dtype Changeset: r46218:83b6f9331702 Date: 2011-08-02 13:29 -0600 http://bitbucket.org/pypy/pypy/changeset/83b6f9331702/
Log: Add num and kind attributes to dtype diff --git a/pypy/module/micronumpy/interp_dtype.py b/pypy/module/micronumpy/interp_dtype.py --- a/pypy/module/micronumpy/interp_dtype.py +++ b/pypy/module/micronumpy/interp_dtype.py @@ -1,7 +1,7 @@ from pypy.interpreter.baseobjspace import Wrappable from pypy.interpreter.error import OperationError from pypy.interpreter.gateway import interp2app -from pypy.interpreter.typedef import TypeDef +from pypy.interpreter.typedef import TypeDef, GetSetProperty from pypy.rlib.rarithmetic import r_int, r_uint, LONG_BIT, LONGLONG_BIT _letters_to_nums = [-1]*128 @@ -58,6 +58,12 @@ self.num = num self.kind = kind + def descr_num(self, space): + return space.wrap(self.num) + + def descr_kind(self, space): + return space.wrap(self.kind) + def conv_float(space, val): return space.float(val) @@ -120,4 +126,7 @@ Dtype.typedef = TypeDef( 'numpy.dtype', __new__ = interp2app(descr_new_dtype), + + num = GetSetProperty(Dtype.descr_num), + kind = GetSetProperty(Dtype.descr_kind), ) diff --git a/pypy/module/micronumpy/test/test_dtypes.py b/pypy/module/micronumpy/test/test_dtypes.py --- a/pypy/module/micronumpy/test/test_dtypes.py +++ b/pypy/module/micronumpy/test/test_dtypes.py @@ -6,3 +6,5 @@ def test_dtype(self): from numpy import dtype d = dtype('l') + assert d.num == 7 + assert d.kind == 'i' _______________________________________________ pypy-commit mailing list pypy-commit@python.org http://mail.python.org/mailman/listinfo/pypy-commit