handle legend columns

Author: nschloe

matplotlib2tikz/legend.py

@@ -179,13 +179,18 @@ def draw_legend(data, obj):
         if alignment != child_alignment:
             warnings.warn(
                 'Varying horizontal alignments in the legend. Using default.'
-            )
+                )
             alignment = None
             break
 
     if alignment:
         data['extra axis options'].add(
-            'legend cell align={%s}' % alignment
+            'legend cell align={{{}}}'.format(alignment)
+            )
+
+    if obj._ncol != 1:
+        data['extra axis options'].add(
+            'legend columns={}'.format(obj._ncol)
             )
 
     # Write styles to data

test/test_legend_columns.py

@@ -7,15 +7,15 @@ def plot():
     import numpy as np
     import matplotlib.pyplot as plt
     fig, ax = plt.subplots(figsize=(17, 6))
-    ax.plot(np.array([1, 2, 3, 4, 5]), label='Test 1')
-    ax.plot(np.array([5, 4, 3, 2, 1]), label='Test 2')
+    ax.plot(np.array([1, 5]), label='Test 1')
+    ax.plot(np.array([5, 1]), label='Test 2')
     ax.legend(ncol=2, loc='upper center')
     return fig
 
 
 def test():
     phash = helpers.Phash(plot())
-    assert phash.phash == 'a3267e19cad99998', phash.get_details()
+    assert phash.phash == '8386de99666939a9', phash.get_details()
     return