def test_is_started(self):
sw = StopWatch()
self.assertFalse(sw.is_started)
sw.start()
self.assertTrue(sw.is_started)
sw.reset()
self.assertFalse(sw.is_started)
def test_split(self):
sw = StopWatch()
sw.start()
self.assertEqual(0, sw.splits_count)
### split 1
delay1 = 0.5
time.sleep(delay1)
sw.split()
self.assertEqual(1, sw.splits_count)
self.assertAlmostEqual(delay1,
sw.last_split_time.total_seconds(),
places=1)
self.assertAlmostEqual(delay1,
sw.mean_split_time.total_seconds(),
places=1)
self.assertEqual(sw.last_split_time, sw.total_split_time)
### split 1
delay2 = 1.0
time.sleep(delay2)
sw.split()
self.assertEqual(2, sw.splits_count)
self.assertAlmostEqual(delay2,
sw.last_split_time.total_seconds(),
places=1)
self.assertAlmostEqual((delay1 + delay2) / 2,
sw.mean_split_time.total_seconds(),
places=1)
self.assertAlmostEqual(delay1 + delay2,
sw.total_split_time.total_seconds(),
places=1)
class TestExecStopwatch(BaseResult):
"""Time test execution using a stopwatch.
Use it as a mix-in of a unittest's result class.
"""
def __init__(self, **kwds):
super().__init__(**kwds)
self.stopwatch = StopWatch()
def startTest(self, test):
if not self.stopwatch.is_started:
self.stopwatch.start()
super().startTest(test)
def stopTest(self, test):
super().stopTest(test)
self.stopwatch.split()
class Walltime(BaseResult):
"""Compute the whole execution time of the test suite.
This is not the sum of the execution time of each test. It is useful
to measure the speedup of running test in parallel.
Use it as a mix-in of a unittest's result class.
"""
def __init__(self, **kwds):
super().__init__(**kwds)
self._walltime_watch = StopWatch()
@property
def walltime(self):
return self._walltime_watch.total_time
def startTest(self, test):
if not self._walltime_watch.is_started:
self._walltime_watch.start()
super().startTest(test)
def __init__(self, **kwds):
super().__init__(**kwds)
self._walltime_watch = StopWatch()
def __init__(self, **kwds):
super().__init__(**kwds)
self.stopwatch = StopWatch()
def test_start_raises_if_already_started(self):
sw = StopWatch()
sw.start()
with self.assertRaises(RuntimeError):
sw.start()
def test_split_raises_if_not_started(self):
sw = StopWatch()
with self.assertRaises(RuntimeError):
sw.split()
def test_total_time(self):
sw = StopWatch()
sw.start()
delay = 0.5
time.sleep(delay)
self.assertAlmostEqual(delay, sw.total_time.total_seconds(), places=1)