def test_with_fake_timer(monkeypatch, use_cython):
"""Integration test with a fake timer.
"""
reporter = SilentReporter(
[cols.name, cols.total_us, cols.total_inner_us, cols.count])
mock_clock = mock.Mock()
mock_clock.side_effect = list(range(0, 100, 1))
@func()
def foo():
with region('a'):
for i in iter_proxy([1, 2, 3], 'iter'):
with region('b'):
pass
with region('b'):
pass
with fresh_region_profiler(monkeypatch):
install_profiler(reporter=reporter,
timer_cls=lambda: get_timer_cls(use_cython)
(mock_clock))
foo()
with region('x'):
pass
foo()
expected = [['name', 'total_us', 'total_inner_us', 'count'],
[RegionProfiler.ROOT_NODE_NAME, '54000000', '5000000', '1'],
['foo()', '48000000', '4000000', '2'],
['a', '44000000', '26000000', '2'],
['b', '12000000', '12000000', '12'],
['iter', '6000000', '6000000', '6'],
['x', '1000000', '1000000', '1']]
assert reporter.rows == expected
def test_automatic_naming(monkeypatch, use_cython):
"""Integration test with regions with automatic naming."""
reporter = SilentReporter([cols.name])
mock_clock = mock.Mock()
mock_clock.side_effect = list(range(0, 100, 1))
@func()
def foo():
with region():
for i in iter_proxy([1, 2, 3]):
pass
with fresh_region_profiler(monkeypatch):
install_profiler(reporter=reporter,
timer_cls=lambda: get_timer_cls(use_cython)
(mock_clock))
foo()
expected = [
["name"],
[RegionProfiler.ROOT_NODE_NAME],
["foo()"],
["foo() <test_module.py:198>"],
["foo() <test_module.py:199>"],
]
assert reporter.rows == expected
def test_reload_works(monkeypatch, multiple_runs):
"""Test that ``fresh_module`` fixture properly
resets ``region_profiler`` module.
"""
reporter = SilentReporter([cols.name])
with fresh_region_profiler(monkeypatch):
assert region_profiler.global_instance._profiler is None
install_profiler(reporter)
assert isinstance(region_profiler.global_instance._profiler,
RegionProfiler)
assert reporter.rows == [['name'], [RegionProfiler.ROOT_NODE_NAME]]
def test_with_real_timer(monkeypatch, use_cython):
"""Integration test with a real timer."""
reporter = SilentReporter(
[cols.name, cols.total_us, cols.total_inner_us, cols.count])
def slow_iter(iterable):
for x in iterable:
time.sleep(0.1)
yield x
@func()
def foo():
time.sleep(0.02)
with region("a"):
time.sleep(0.02)
for i in iter_proxy(slow_iter([0.1, 0.2, 0.3]), "iter"):
with region("b"):
time.sleep(i)
with fresh_region_profiler(monkeypatch):
install_profiler(reporter)
foo()
with region("x"):
time.sleep(0.5)
foo()
expected: List[List[Any]] = [
[RegionProfiler.ROOT_NODE_NAME, 2380000, 0, "1"],
["foo()", 1880000, 40000, "2"],
["a", 1840000, 40000, "2"],
["b", 1200000, 1200000, "6"],
["iter", 600000, 600000, "6"],
["x", 500000, 500000, "1"],
]
# (fresh_region_profiler calls dump_profiler)
rows = reporter.rows[1:] # type: ignore[index]
lower = 0.99
upper = 1.03
upper_delta = 5000
assert len(rows) == len(expected)
print(rows)
for i, (r, e) in enumerate(zip(rows, expected)):
assert r[0] == e[0]
assert r[3] == e[3]
if i == 0:
assert int(r[1]) > e[1]
else:
assert e[1] * lower <= int(r[1]) <= e[1] * upper + upper_delta
assert e[2] * lower <= int(r[2]) <= e[2] * upper + upper_delta
def test_with_global_regions(monkeypatch, use_cython):
"""Integration test with regions marked as globals."""
reporter = SilentReporter(
[cols.name, cols.total_us, cols.total_inner_us, cols.count])
mock_clock = mock.Mock()
mock_clock.side_effect = list(range(0, 100, 1))
@func(asglobal=True)
def bar():
with region("a"):
with region("bar_global", asglobal=True):
for i in iter_proxy([1, 2, 3], "iter", asglobal=True):
pass
@func()
def foo():
with region("a"):
for i in iter_proxy([1, 2, 3], "iter"):
with region("b"):
pass
with region("b"):
pass
bar()
with fresh_region_profiler(monkeypatch):
install_profiler(reporter=reporter,
timer_cls=lambda: get_timer_cls(use_cython)
(mock_clock))
foo()
with region("x"):
pass
foo()
expected = [
["name", "total_us", "total_inner_us", "count"],
[RegionProfiler.ROOT_NODE_NAME, "84000000", "0", "1"],
["foo()", "78000000", "4000000", "2"],
["a", "74000000", "56000000", "2"],
["b", "12000000", "12000000", "12"],
["iter", "6000000", "6000000", "6"],
["bar()", "28000000", "4000000", "2"],
["a", "24000000", "24000000", "2"],
["bar_global", "20000000", "20000000", "2"],
["iter", "6000000", "6000000", "6"],
["x", "1000000", "1000000", "1"],
]
assert reporter.rows == expected
def test_with_real_timer(monkeypatch, use_cython):
"""Integration test with a real timer.
"""
reporter = SilentReporter(
[cols.name, cols.total_us, cols.total_inner_us, cols.count])
def slow_iter(iterable):
for x in iterable:
time.sleep(0.1)
yield x
@func()
def foo():
time.sleep(0.02)
with region('a'):
time.sleep(0.02)
for i in iter_proxy(slow_iter([0.1, 0.2, 0.3]), 'iter'):
with region('b'):
time.sleep(i)
with fresh_region_profiler(monkeypatch):
install_profiler(reporter)
foo()
with region('x'):
time.sleep(0.5)
foo()
expected = [[RegionProfiler.ROOT_NODE_NAME, 2380000, 0, '1'],
['foo()', 1880000, 40000, '2'], ['a', 1840000, 40000, '2'],
['b', 1200000, 1200000, '6'], ['iter', 600000, 600000, '6'],
['x', 500000, 500000, '1']]
rows = reporter.rows[1:]
lower = 0.99
upper = 1.03
upper_delta = 5000
assert len(rows) == len(expected)
print(rows)
for i, (r, e) in enumerate(zip(rows, expected)):
assert r[0] == e[0]
assert r[3] == e[3]
if i == 0:
assert int(r[1]) > e[1]
else:
assert e[1] * lower <= int(r[1]) <= e[1] * upper + upper_delta
assert e[2] * lower <= int(r[2]) <= e[2] * upper + upper_delta
