implemented basic unit tests

cadyxia · cadyxia · commit f0dc7ea69a66 · 2025-09-18T18:45:25.000-07:00
diff --git a/bashplotlib/scatterplot.py b/bashplotlib/scatterplot.py
@@ -31,7 +31,7 @@ def get_scale(series, is_y=False, steps=20):
     return scaled_series
 
 
-def _plot_scatter(xs, ys, size, pch, colour, title, cs, x_title, y_title):
+def _plot_scatter(xs, ys, size, pch, title, cs, x_title, y_title):
     plotted = set()
     graph = ""
 
@@ -63,7 +63,7 @@ def _plot_scatter(xs, ys, size, pch, colour, title, cs, x_title, y_title):
                     point = "-"
             graph += point + " "
         graph += " |\n"
-    graph += "  +" + "-" * (2 * (len(get_scale(xs, False, size)) + 2) - 2) + "+"
+    graph += "  +" + "-" * (2 * (len(get_scale(xs, False, size)) + 2) - 2) + "+\n"
     graph += "  " + x_title
     return graph
 
@@ -99,7 +99,7 @@ def plot_scatter(f, xs, ys, size, pch, colour, title, x_title, y_title):
         with open(ys) as fh:
             ys = [float(str(row).strip()) for row in fh]
 
-    graph = _plot_scatter(xs, ys, size, pch, colour, title, cs, x_title, y_title)
+    graph = _plot_scatter(xs, ys, size, pch, title, cs, x_title, y_title)
     print(graph)
     
 
diff --git a/test.py b/test.py
@@ -0,0 +1,69 @@
+# test.py
+from bashplotlib.scatterplot import _plot_scatter
+import unittest
+
+class TestScatterPlot(unittest.TestCase):
+  def test_no_points(self):
+    self.assertEqual(_plot_scatter([], [], 10, "x",
+                                  "My Test Graph", "default", "My X Axis", "My Y Axis"), 
+                    """  +------------------------+\n  |     My Test Graph      |
+  +------------------------+
+  +------------------------+
+M |           |            |
+y |           |            |
+  |           |            |
+Y |           |            |
+  |           |            |
+A | - - - - - o - - - - -  |
+x |           |            |
+i |           |            |
+s |           |            |
+  |           |            |
+  |           |            |
+  +------------------------+
+  My X Axis""")
+    
+  def test_points(self):
+    self.assertEqual(_plot_scatter([-10, -3, 20,30], [-10, 6, 20,-2], 10, "*",
+                                  "Points!", "default", "Something x", "Another y"), 
+                    """  +--------------------------+\n  |         Points!          |
+  +--------------------------+
+  +--------------------------+
+A |       |           *      |
+n |       |                  |
+o |       |                  |
+t |       |                  |
+h |       |                  |
+e |     * |                  |
+r |       |                  |
+  | - - - o - - - - - - - -  |
+y |       |                  |
+  |       |               *  |
+  |       |                  |
+  | *     |                  |
+  +--------------------------+
+  Something x""")
+
+  def test_points_on_axes(self):
+    self.assertEqual(_plot_scatter([0, -10, 0, 20,0], [0, 0, 6, 0,-2], 10, "#",
+                                  "Points!", "default", "x", "y"),
+                    """  +--------------------------+\n  |         Points!          |
+  +--------------------------+
+  +--------------------------+
+y |         #                |
+  |         |                |
+  |         |                |
+  |         |                |
+  |         |                |
+  |         |                |
+  |         |                |
+  |         |                |
+  | # - - - # - - - - - - #  |
+  |         |                |
+  |         |                |
+  |         #                |
+  +--------------------------+
+  x""")
+
+if __name__ == "__main__":
+  unittest.main()
