File size: 1,999 Bytes
079c32c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
import pytest
import numpy as np
import time
from ding.utils.time_helper import build_time_helper, WatchDog, TimeWrapperTime, EasyTimer
@pytest.mark.unittest
class TestTimeHelper:
def test_naive(self):
class NaiveObject(object):
pass
cfg = NaiveObject()
setattr(cfg, 'common', NaiveObject())
setattr(cfg.common, 'time_wrapper_type', 'time')
with pytest.raises(RuntimeError):
time_handle = build_time_helper()
with pytest.raises(KeyError):
build_time_helper(cfg=None, wrapper_type="not_implement")
time_handle = build_time_helper(cfg)
time_handle = build_time_helper(wrapper_type='cuda')
# wrapper_type='cuda' but cuda is not available
assert issubclass(time_handle, TimeWrapperTime)
time_handle = build_time_helper(wrapper_type='time')
@time_handle.wrapper
def func1(x):
return x + 1
def func2(x):
return x + 1
# usage 1
ret, t = func1(3)
assert np.isscalar(t)
assert func1(4)[0] == func2(4)
# usage 2
time_handle.start_time()
_ = func2(3)
t = time_handle.end_time()
assert np.isscalar(t)
#test time_lag and restart
time_handle.start_time()
time.sleep(0.5)
time_handle.start_time()
time.sleep(1)
t = time_handle.end_time()
assert np.isscalar(t)
# time_lag is bigger than 1e-3
# assert abs(t-1) < 1e-3
assert abs(t - 1) < 1e-2
timer = EasyTimer()
with timer:
tmp = np.random.random(size=(4, 100))
tmp = tmp ** 2
value = timer.value
assert isinstance(value, float)
@pytest.mark.unittest
class TestWatchDog:
def test_naive(self):
watchdog = WatchDog(3)
watchdog.start()
time.sleep(2)
with pytest.raises(TimeoutError):
time.sleep(2)
watchdog.stop()
|