mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
82a313d09b
* rename global accessible and intergration get_sintance and create_instance * move ManagerMixin to utils * fix as docstring and seporate get_instance to get_instance and get_current_instance * fix lint * fix docstring, rename and move test_global_meta * rename LogBuffer to HistoryBuffer, rename MessageHub methods, MessageHub support resume * refine MMLogger timestamp, update unit test * MMLogger add logger_name arguments * Fix docstring * change default logger_name to mmengine * Fix docstring comment and unitt test * fix docstring fix docstring * fix docstring * Fix lint * Fix hook unit test * Fix comments * should not accept other arguments if corresponding instance has been created * fix logging ddp file saving * fix logging ddp file saving * fix docstring * fix unit test * fix docstring as comment
30 lines
815 B
Python
30 lines
815 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, 0)
|
|
runner.message_hub.update_scalar.assert_called()
|
|
|
|
def test_after_iter(self):
|
|
hook = IterTimerHook()
|
|
runner = Mock()
|
|
runner.log_buffer = dict()
|
|
hook._before_epoch(runner)
|
|
hook._after_iter(runner, 0)
|
|
runner.message_hub.update_scalar.assert_called()
|