Merge pull request #3447 from efiring/ma_mvoid_len

BUG: add __len__ method to ma.mvoid; closes #576
numpy · Jun 17, 2013 · 5e977d8 · 5e977d8
2 parents 2a5c2c8 + 216d8cb
commit 5e977d8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/numpy/ma/core.py b/numpy/ma/core.py
@@ -5643,6 +5643,9 @@ def __iter__(self):
                 else:
                     yield d
 
+    def __len__(self):
+        return self._data.__len__()
+
     def filled(self, fill_value=None):
         """
         Return a copy with masked fields filled with a given value.

diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py
@@ -3527,6 +3527,10 @@ def test_setitem(self):
         a[0]['a'] = 2
         assert_equal(a.mask, control)
 
+    def test_element_len(self):
+        # check that len() works for mvoid (Github issue #576)
+        for rec in self.data['base']:
+            assert_equal(len(rec), len(self.data['ddtype']))
 
 #------------------------------------------------------------------------------