Merge pull request #182 from nschloe/legend-columns

Legend columns

Author: nschloe

matplotlib2tikz/__about__.py

@@ -5,6 +5,6 @@
 __copyright__ = 'Copyright (c) 2010-2017, %s <%s>' % (__author__, __email__)
 __credits__ = []
 __license__ = 'License :: OSI Approved :: MIT License'
-__version__ = '0.6.10'
+__version__ = '0.6.11'
 __maintainer__ = u'Nico Schlömer'
 __status__ = 'Development Status :: 5 - Production/Stable'

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

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+#
+import helpers
+
+
+def plot():
+    import numpy as np
+    import matplotlib.pyplot as plt
+    fig, ax = plt.subplots(figsize=(17, 6))
+    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 == '8386de99666939a9', phash.get_details()
+    return
+
+
+if __name__ == '__main__':
+    helpers.compare_with_latex(plot())