mmengine/tests/test_hook/test_empty_cache_hook.py

# Copyright (c) OpenMMLab. All rights reserved.
from unittest.mock import Mock

from mmengine.hooks import EmptyCacheHook


class TestEmptyCacheHook:

    def test_emtpy_cache_hook(self):
        Hook = EmptyCacheHook(True, True, True)
        Runner = Mock()
        Hook._after_iter(Runner)
        Hook._before_epoch(Runner)
        Hook._after_epoch(Runner)