Add iinfo based on a patch by Albert Strasheim (ticket #250).

1 files changed, 18 insertions, 1 deletions

diff --git a/numpy/lib/tests/test_getlimits.py b/numpy/lib/tests/test_getlimits.py
index 1eaadc953..7a4fea57a 100644
--- a/numpy/lib/tests/test_getlimits.py
+++ b/numpy/lib/tests/test_getlimits.py
@@ -4,8 +4,9 @@
 from numpy.testing import *
 set_package_path()
 import numpy.lib;reload(numpy.lib)
-from numpy.lib.getlimits import finfo
+from numpy.lib.getlimits import finfo, iinfo
 from numpy import single,double,longdouble
+import numpy as N
 restore_path()
 
 ##################################################
@@ -34,5 +35,21 @@ class test_longdouble(NumpyTestCase):
         ftype2 = finfo(longdouble)
         assert_equal(id(ftype),id(ftype2))
 
+class test_iinfo(NumpyTestCase):
+    def check_basic(self):
+        dts = zip(['i1', 'i2', 'i4', 'i8',
+                   'u1', 'u2', 'u4', 'u8'],
+                  [N.int8, N.int16, N.int32, N.int64,
+                   N.uint8, N.uint16, N.uint32, N.uint64])
+        for dt1, dt2 in dts:
+            assert_equal(iinfo(dt1).min, iinfo(dt2).min)
+            assert_equal(iinfo(dt1).max, iinfo(dt2).max)
+        self.assertRaises(ValueError, iinfo, 'f4')
+
+    def check_unsigned_max(self):
+        types = N.sctypes['uint']
+        for T in types:
+            assert_equal(iinfo(T).max, T(-1))
+
 if __name__ == "__main__":
     NumpyTest().run()