Add unit-test for record-arrays with object field.

2 files changed, 23 insertions, 2 deletions

diff --git a/numpy/core/records.py b/numpy/core/records.py
index 333bc486e..aebf3b3c2 100644
--- a/numpy/core/records.py
+++ b/numpy/core/records.py
@@ -176,7 +176,7 @@ class recarray(sb.ndarray):
         fielddict = sb.ndarray.__getattribute__(self,'dtype').fields
         try:
             res = fielddict[attr][:2]
-        except KeyError:
+        except (TypeError, KeyError):
             raise AttributeError, "record array has no attribute %s" % attr
         obj = self.getfield(*res)
         # if it has fields return a recarray, otherwise return
@@ -195,10 +195,16 @@ class recarray(sb.ndarray):
         fielddict = sb.ndarray.__getattribute__(self,'dtype').fields
         try:
             res = fielddict[attr][:2]
-        except KeyError:
+        except (TypeError,KeyError):
             raise AttributeError, "record array has no attribute %s" % attr 
         return self.setfield(val,*res)
 
+    def __getitem__(self, indx):
+        obj = sb.ndarray.__getitem__(self, indx)
+        if (isinstance(obj, sb.ndarray) and obj.dtype.isbuiltin):
+            return obj.view(sb.ndarray)
+        return obj
+
     def field(self,attr, val=None):
         fielddict = sb.ndarray.__getattribute__(self,'dtype').fields
 
diff --git a/numpy/core/tests/test_records.py b/numpy/core/tests/test_records.py
index 4c0e8831b..b431201b5 100644
--- a/numpy/core/tests/test_records.py
+++ b/numpy/core/tests/test_records.py
@@ -39,5 +39,20 @@ class test_fromrecords(NumpyTestCase):
         fd.seek(2880*2)
         r = rec.fromfile(fd, formats='f8,i4,a5', shape=3, byteorder='big')
 
+    def check_recarray_from_obj(self):
+        count = 10
+        a = zeros(count, dtype='O')
+        b = zeros(count, dtype='f8')
+        c = zeros(count, dtype='f8')
+        for i in range(len(a)):
+            a[i] = range(1,10)
+
+        mine = numpy.rec.fromarrays([a,b,c],
+                                    names='date,data1,data2')
+        for i in range(len(a)):
+            assert(mine.date[i]==range(1,10))
+            assert(mine.data1[i]==0.0)
+            assert(mine.data2[i]==0.0)
+
 if __name__ == "__main__":
     NumpyTest().run()