From 3ee78cf60bc997c3e96926572f18f513fb99ae6f Mon Sep 17 00:00:00 2001
From: Egor Tensin <Egor.Tensin@gmail.com>
Date: Fri, 27 Jan 2017 02:34:44 +0300
Subject: BarChartBuilder: bugfix

---
 bin/utils/bar_chart.py | 29 ++++++++++++++++++++---------
 1 file changed, 20 insertions(+), 9 deletions(-)

(limited to 'bin/utils/bar_chart.py')

diff --git a/bin/utils/bar_chart.py b/bin/utils/bar_chart.py
index 12b43b4..d916481 100644
--- a/bin/utils/bar_chart.py
+++ b/bin/utils/bar_chart.py
@@ -27,10 +27,20 @@ class BarChartBuilder:
         return self._ax.get_xaxis()
 
     def set_categories_axis_limits(self, start=None, end=None):
-        self._ax.set_ylim(bottom=start, top=end)
+        bottom, top = self._ax.get_ylim()
+        if start is not None:
+            bottom = start
+        if end is not None:
+            top = end
+        self._ax.set_ylim(bottom=bottom, top=top)
 
     def set_values_axis_limits(self, start=None, end=None):
-        self._ax.set_xlim(left=start, right=end)
+        left, right = self._ax.get_xlim()
+        if start is not None:
+            left = start
+        if end is not None:
+            right = end
+        self._ax.set_xlim(left=left, right=right)
 
     def enable_grid_for_categories(self):
         self._get_categories_axis().grid()
@@ -103,6 +113,8 @@ class BarChartBuilder:
 
         if min(values) >= 0:
             self.set_values_axis_limits(start=0)
+            if np.isclose(max(values), 0.):
+                self.set_values_axis_limits(end=1)
         elif max(values) < 0:
             self.set_values_axis_limits(end=0)
 
@@ -119,15 +131,14 @@ if __name__ == '__main__':
     import argparse
     parser = argparse.ArgumentParser()
 
-    parser.add_argument('--categories', nargs='*', default=[],
-                        help='categories')
-    parser.add_argument('--values', nargs='*', type=float, default=[],
-                        help='values')
+    parser.add_argument('--categories', nargs='*', metavar='LABEL',
+                        default=[])
+    parser.add_argument('--values', nargs='*', metavar='N',
+                        default=[], type=float)
 
-    parser.add_argument('--output', '-o',
-                        help='set output file path')
+    parser.add_argument('--output', '-o', help='set output file path')
 
-    parser.add_argument('--middle', action='store_true',
+    parser.add_argument('--align-middle', action='store_true',
                         dest='labels_align_middle',
                         help='align labels to the middle of the bars')
 
-- 
cgit v1.2.3