Fix csv2rec for passing in both names and comments.

gatagat · gatagat · commit 90942c9d3e42 · 2015-05-22T14:07:13.000+02:00
diff --git a/lib/matplotlib/mlab.py b/lib/matplotlib/mlab.py
@@ -2867,12 +2867,17 @@ def get_func(name, item, func):
         'print':  'print_',
         }
 
-    def get_converters(reader):
+    def get_converters(reader, comments):
 
         converters = None
-        for i, row in enumerate(reader):
+        i = 0
+        for row in reader:
+            if (len(row) and comments is not None and
+                    row[0].startswith(comments)):
+                continue
             if i == 0:
                 converters = [mybool]*len(row)
+            i += 1
             if checkrows and i > checkrows:
                 break
             for j, (name, item) in enumerate(zip(names, row)):
@@ -2925,7 +2930,7 @@ def get_converters(reader):
             names = [n.strip() for n in names.split(',')]
 
     # get the converter functions by inspecting checkrows
-    converters = get_converters(reader)
+    converters = get_converters(reader, comments)
     if converters is None:
         raise ValueError('Could not find any valid data in CSV file')
 
diff --git a/lib/matplotlib/tests/test_mlab.py b/lib/matplotlib/tests/test_mlab.py
@@ -339,6 +339,14 @@ def test_rec2csv_bad_shape_ValueError(self):
         # the bad recarray should trigger a ValueError for having ndim > 1.
         assert_raises(ValueError, mlab.rec2csv, bad, self.fd)
 
+    def test_csv2rec_names_with_comments(self):
+        self.fd.seek(0)
+        self.fd.write('# comment\n1,2,3\n4,5,6\n')
+        self.fd.seek(0)
+        array = mlab.csv2rec(self.fd, names='a,b,c')
+        assert len(array) == 2
+        assert len(array.dtype) == 3
+
 
 class window_testcase(CleanupTestCase):
     def setUp(self):
