mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
1244e486ae
* [Feature]: Add Part3 of Hooks * [Feature]: Add Hook * add iter timer hook * update test * [Fix]: Add docstring and type hint for base hook * fix mypy * improve doc coverage and merge main Co-authored-by: seuyou <3463423099@qq.com>
30 lines
781 B
Python
30 lines
781 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from unittest.mock import Mock
|
|
|
|
from mmengine.hooks import IterTimerHook
|
|
|
|
|
|
class TestIterTimerHook:
|
|
|
|
def test_before_epoch(self):
|
|
Hook = IterTimerHook()
|
|
Runner = Mock()
|
|
Hook.before_epoch(Runner)
|
|
assert isinstance(Hook.t, float)
|
|
|
|
def test_before_iter(self):
|
|
Hook = IterTimerHook()
|
|
Runner = Mock()
|
|
Runner.log_buffer = dict()
|
|
Hook.before_epoch(Runner)
|
|
Hook.before_iter(Runner)
|
|
assert 'data_time' in Runner.log_buffer
|
|
|
|
def test_after_iter(self):
|
|
Hook = IterTimerHook()
|
|
Runner = Mock()
|
|
Runner.log_buffer = dict()
|
|
Hook.before_epoch(Runner)
|
|
Hook.after_iter(Runner)
|
|
assert 'time' in Runner.log_buffer
|