Merge pull request #24 from Anselmoo/ranges

Ranges

Author: Anselmoo

bashplot/bashplot.py

@@ -151,8 +151,13 @@ def plot(data, args, label):
         if args["y_limits"]:
             fig.set_y_limits(min_=args["y_limits"][0], max_=args["y_limits"][1])
         else:
-            fig.set_y_limits(min_=np.min(data[:, 1]), max_=np.max(data[:, 1]))
-
+            try:
+                fig.set_y_limits(min_=np.min(data[:, 1:]), max_=np.max(data[:, 1:]))
+            except ValueError:
+                min_max = np.mean(data[:, 1:])
+                fig.set_y_limits(
+                    min_=min_max - min_max * 0.1, max_=min_max + min_max * 0.1
+                )
         if args["color"]:
             fig.color_mode = "rgb"
 

test/sample_generator.py

@@ -46,3 +46,8 @@ def generate(fname="test_data"):
     y = np.random.normal(mu, sigma, 50)
     data = np.array([x, y]).T
     np.savetxt(Path(f"./{fname}_random_gaussian.txt"), data)
+
+    np.savetxt(
+        Path(f"./{fname}_equaldata.out"),
+        np.array([[-1.0, 2.0], [0.0, 2.0], [1.0, 2.0]]),
+    )

test/test_bashplot.py

@@ -14,6 +14,7 @@
 
 test_txt = list(Path.cwd().glob("test*.txt"))
 test_dat = list(Path.cwd().glob("test*.dat"))
+test_out = list(Path.cwd().glob("test*.out"))
 
 args_1 = {
     "infile": test_txt,
@@ -61,6 +62,21 @@
     "legend": True,
     "version": True,
 }
+args_4 = {
+    "infile": test_out,
+    "comments": None,
+    "delimiter": None,
+    "skip_header": 0,
+    "skip_footer": 0,
+    "usecols": (0, 1),
+    "size": [30, 20],
+    "x_limits": None,
+    "y_limits": None,
+    "scatter": False,
+    "color": False,
+    "legend": True,
+    "version": False,
+}
 
 
 def test_fnames():
@@ -136,6 +152,11 @@ def test_customize_run_2():
     assert True
 
 
+def test_customize_run_3():
+    bashplot.bashplot(fnames=test_out, args=args_4)
+    assert True
+
+
 @mock.patch("bashplot.bashplot.command_line_runner")
 def test_command_line_mock(command_line_runner):
     bashplot.command_line_runner()