def __init__(self, description, category=None):
self.utc_time = utcnow()
self.machine_time = datetime.datetime.now()
self.description = description
self.category = category or DEFAULT_CATEGORY
# Will be set explicitly via containers
self.line_no = None
self.file_path = None
def test_start(self):
"""`Timer.start` should create an `Interval` that has `start` attribute set."""
timer = Timer()
assert 'my_key' not in timer
prev_now = utcnow()
time.sleep(0.001)
timer.start('my_key')
assert prev_now < timer['my_key'].start
assert timer['my_key'].end is None
def test_end(self):
"""`timer.end` should update the matching `Interval.end` for the given key."""
timer = Timer()
# Explicitly set value for testing
# don't care about start
timer['my_key'] = Interval('foo', None)
prev_now = utcnow()
time.sleep(0.001)
timer.end('my_key')
assert prev_now < timer['my_key'].end