add txtbars

tests/test_output.py

@@ -0,0 +1,62 @@
+import unittest
+
+import bumblebee.output
+
+
+class TestHBar(unittest.TestCase):
+    """tests for bumblebee.output.HBar"""
+    def setUp(self):
+        self.value = 1
+        self.values = [10, 20, 30, 40, 55, 65, 80, 90]
+        self.hbar = bumblebee.output.HBar(self.value)
+
+    def test___init__(self):
+        """bumblebee.output.HBar.__init__()"""
+        self.assertEqual(
+            self.hbar.step, bumblebee.output.MAX_PERCENTS / bumblebee.output.CHARS)
+
+    def test_get_char(self):
+        """bumblebee.output.HBar.get_char()"""
+        for i in range(bumblebee.output.CHARS):
+            hbar = bumblebee.output.HBar(self.values[i])
+            self.assertEqual(hbar.get_char(), bumblebee.output.HBARS[i])
+        # edge case for 100%
+        hbar = bumblebee.output.HBar(100)
+        self.assertEqual(hbar.get_char(), bumblebee.output.HBARS[-1])
+
+
+class TestVBar(unittest.TestCase):
+    """tests for bumblebee.output.VBar"""
+    def setUp(self):
+        self.value = 1
+        self.values = [10, 20, 30, 40, 55, 65, 80, 90]
+        self.vbar = bumblebee.output.VBar(self.value)
+
+    def test___init__(self):
+        """bumblebee.output.VBar.__init__()"""
+        self.assertEqual(
+            self.vbar.step, bumblebee.output.MAX_PERCENTS / bumblebee.output.CHARS)
+
+    def test_get_chars(self):
+        """bumblebee.output.VBar.get_char()"""
+        for i in range(bumblebee.output.CHARS):
+            vbar = bumblebee.output.VBar(self.values[i])
+            self.assertEqual(vbar.get_chars(), bumblebee.output.VBARS[i])
+        # edge case for 100%
+        vbar = bumblebee.output.VBar(100)
+        self.assertEqual(vbar.get_chars(), bumblebee.output.VBARS[-1])
+        # 0.x chars filled
+        value = 0.1
+        vbar = bumblebee.output.VBar(value, 3)
+        expected_chars = vbar.bars[0] + "  "
+        self.assertEqual(vbar.get_chars(), expected_chars)
+        # 1.x chars filled
+        value = 35
+        vbar = bumblebee.output.VBar(value, 3)
+        expected_chars = vbar.bars[-1] + vbar.bars[0] + " "
+        self.assertEqual(vbar.get_chars(), expected_chars)
+        # 2.x chars filled
+        value = 67
+        vbar = bumblebee.output.VBar(value, 3)
+        expected_chars = vbar.bars[-1] * 2 + vbar.bars[0]
+        self.assertEqual(vbar.get_chars(), expected_chars)