class Timings(object):
"""A time aggregation class"""
def __init__(self):
self.aggregate = []
self.timer = Timer()
self.last_time = 0
def timer_start(self):
try:
self.timer = Timer()
self.timer.start()
except ArithmeticError:
self.timer.start_time = 0
def timer_pause(self):
try:
self.timer.pause()
except RuntimeError:
pass
def timer_restart(self):
try:
self.timer.restart()
except RuntimeError:
pass
def timer_stop(self):
try:
self.last_time = self.timer.stop()
self.aggregate.append(self.last_time)
except ArithmeticError:
pass
def remove_timer(self, timer):
try:
self.aggregate.remove(self.aggregate[timer])
except RuntimeError:
pass
class TimeTests(unittest.TestCase):
def test_now(self):
self.timer = Timer()
self.assertGreaterEqual(self.timer.serial_time(), 0, "?")
def test_timer_with_wait(self):
self.timer = Timer()
self.timer.start()
time.sleep(1)
self.timer.stop()
self.assertGreaterEqual(self.timer.end_time, self.timer.start_time, "?")
def test_timer_pause(self):
self.timer = Timer()
self.timer.start()
time.sleep(1)
self.timer.pause()
time.sleep(1)
self.timer.restart()
time.sleep(1)
self.timer.stop()
self.assertGreaterEqual(self.timer.duration, 2, "?")