Merge pull request #6423 from tacaswell/tst_fix_windows_print_test

jenshnielsen · tacaswell · commit cab8cf7c4239 · 2016-05-22T21:28:03.000-04:00
TST: splitlines in rec2txt test
diff --git a/lib/matplotlib/mlab.py b/lib/matplotlib/mlab.py
@@ -3188,9 +3188,8 @@ def get_justify(colname, column, precision):
         ntype = column.dtype
 
         if np.issubdtype(ntype, str) or np.issubdtype(ntype, bytes):
-            # The division below handles unicode stored in array, which could
-            # have 4 bytes per char
-            length = max(len(colname), column.itemsize // column[0].itemsize)
+            fixed_width = int(ntype.str[2:])
+            length = max(len(colname), fixed_width)
             return 0, length+padding, "%s"  # left justify
 
         if np.issubdtype(ntype, np.int):
diff --git a/lib/matplotlib/tests/test_mlab.py b/lib/matplotlib/tests/test_mlab.py
@@ -403,10 +403,10 @@ def test_csv2txt_basic(self):
                                      (str('y'), np.int8),
                                      (str('s'), str, 3),
                                      (str('s2'), str, 4)]))
-        truth = ('       x   y   s   s2\n'
-                 '   1.000   2   foo bing \n'
-                 '   2.000   3   bar blah ')
-        assert_equal(mlab.rec2txt(a), truth)
+        truth = ('       x   y   s     s2\n'
+                 '   1.000   2   foo   bing   \n'
+                 '   2.000   3   bar   blah   ').splitlines()
+        assert_equal(mlab.rec2txt(a).splitlines(), truth)
 
 
 class window_testcase(CleanupTestCase):
