bumblebee-status/tests/test_util.py
Tobi-wan Kenobi 846d28714a [tests] Add tests for bumblebee.util
Add tests for the utility functions and fix a few embarrassing bugs in
the process.
2016-12-17 09:00:49 +01:00

25 lines
846 B
Python

# pylint: disable=C0103,C0111
import unittest
from bumblebee.util import *
class TestUtil(unittest.TestCase):
def test_bytefmt(self):
value = 10
display = 10
units = [ "B", "KiB", "MiB", "GiB" ]
for unit in units:
self.assertEquals(bytefmt(value), "{:.2f}{}".format(display, unit))
value *= 1024
self.assertEquals(bytefmt(value), "{:.2f}GiB".format(display*1024))
def test_durationfmt(self):
self.assertEquals(durationfmt(00), "00:00")
self.assertEquals(durationfmt(25), "00:25")
self.assertEquals(durationfmt(60), "01:00")
self.assertEquals(durationfmt(119), "01:59")
self.assertEquals(durationfmt(3600), "01:00:00")
self.assertEquals(durationfmt(7265), "02:01:05")
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4