i3pystatus/tests/test_battery.py

32 lines
1.0 KiB
Python

#!/usr/bin/env python
import unittest
from i3pystatus import battery
def factory(path, format, expected):
def test():
bc = battery.BatteryChecker(path=path, format=format)
bc.run()
assert bc.output["full_text"] == expected
test.description = path + ":" + format
return test
def basic_test_generator():
cases = [
("test_battery_basic1", "FULL", "0.000", "0h:00m"),
("test_battery_basic2", "FULL", "0.000", "0h:00m"),
("test_battery_basic3", "DIS", "15.624", "4h:04m"),
("test_battery_basic4", "DIS", "17.510", "1h:46m"),
("test_battery_basic5", "DIS", "11.453", "4h:52m"),
("test_battery_basic6", "CHR", "30.764", "0h:20m"),
("test_battery_basic7", "DIS", "27.303", "1h:44m"),
("test_battery_broken1", "FULL", "0.000", "0h:00m"),
]
for path, status, consumption, remaining in cases:
yield factory(path, "{status}", status)
yield factory(path, "{consumption:.3f}", consumption)
yield factory(path, "{remaining:%hh:%Mm}", remaining)