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
|
"""Unit tests for timing facilities"""
# Copyright (C) 2017 Jan Blechta, Jack S. Hale
#
# This file is part of DOLFINx (https://www.fenicsproject.org)
#
# SPDX-License-Identifier: LGPL-3.0-or-later
from time import sleep
import pytest
from dolfinx import common
def test_timer():
"""Test that named Timer works."""
dt = 0.05
task = "test_named_str"
t = common.Timer(task)
t.start()
sleep(dt)
t.stop()
assert t.elapsed().total_seconds() > 0.9 * dt
t.resume()
sleep(dt)
t.stop()
assert t.elapsed().total_seconds() > 2 * 0.9 * dt
t.flush()
t = common.timing(task)
assert t[0] == 1
assert t[1].total_seconds() > 0.045
def test_timer_flush_stop():
"""Test stop/flush."""
t = common.Timer()
t.start()
with pytest.raises(RuntimeError):
t.flush()
t.stop()
t.resume()
with pytest.raises(RuntimeError):
t.flush()
t.stop()
t.flush()
def test_context_manager_named():
"""Test that named Timer works as context manager."""
task = "test_context_manager_named_str"
with common.Timer(task):
sleep(0.05)
delta = common.timing(task)
assert delta[1].total_seconds() > 0.045
def test_context_manager_anonymous():
"""Test that anonymous Timer works with context manager."""
timer = common.Timer()
with timer:
sleep(0.05)
assert timer.elapsed().total_seconds() > 0.045
if __name__ == "__main__":
pytest.main([__file__])
|
