Fix randint. Closes ticket #552.

1 files changed, 4 insertions, 6 deletions

diff --git a/numpy/oldnumeric/random_array.py b/numpy/oldnumeric/random_array.py
index 0b06ee959..2f5479c1f 100644
--- a/numpy/oldnumeric/random_array.py
+++ b/numpy/oldnumeric/random_array.py
@@ -12,8 +12,6 @@ ArgumentError = ValueError
 import numpy.random.mtrand as mt
 import numpy as Numeric
 
-from types import IntType
-
 def seed(x=0, y=0):
     if (x == 0 or y == 0):
         mt.seed()
@@ -42,16 +40,16 @@ def uniform(minimum, maximum, shape=[]):
 def randint(minimum, maximum=None, shape=[]):
     """randint(min, max, shape=[]) = random integers >=min, < max
     If max not given, random integers >= 0, <min"""
-    if not isinstance(minimum, IntType):
+    if not isinstance(minimum, int):
         raise ArgumentError, "randint requires first argument integer"
     if maximum is None:
         maximum = minimum
         minimum = 0
-    if not isinstance(maximum, IntType):
+    if not isinstance(maximum, int):
         raise ArgumentError, "randint requires second argument integer"
     a = ((maximum-minimum)* random(shape))
-    if isinstance(a, Numeric.ArrayType):
-        return minimum + a.astype(Numeric.Int)
+    if isinstance(a, Numeric.ndarray):
+        return minimum + a.astype(Numeric.int)
     else:
         return minimum + int(a)