FIX: compute str column width

tacaswell · tacaswell · commit 190037239931 · 2016-05-16T10:41:37.000-04:00
The division was always returning 1, use length of first element
instead.
diff --git a/lib/matplotlib/mlab.py b/lib/matplotlib/mlab.py
@@ -3188,9 +3188,7 @@ 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)
+            length = max(len(colname), len(column[0]))
             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,9 +403,9 @@ 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 ').splitlines()
+        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)
 
 
