63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
|
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)
|