def test_memory_usage():
# Valid measures
for mem in (500, 200, 100):
assert_memory_less_than(mem, 0.1, create_object, mem * 1024**2)
# Ensure an exception is raised with too small objects as
# memory_profiler can return non trustable memory measure in this case.
assert_raises_regex(ValueError,
"Memory profiler measured an untrustable memory",
assert_memory_less_than, 50, 0.1, create_object,
25 * 1024**2)
# Ensure ValueError is raised if memory used is above expected memory
# limit.
assert_raises_regex(ValueError, "Memory consumption measured",
assert_memory_less_than, 50, 0.1, create_object,
100 * 1024**2)
def test_memory_usage():
# Valid measures (larger objects)
for mem in (500, 200):
assert_memory_less_than(mem, 0.1, create_object, mem * 1024 ** 2)
# Ensure an exception is raised with too small objects as
# memory_profiler can return non trustable memory measure in this case.
assert_raises_regex(ValueError,
"Memory profiler measured an untrustable memory",
assert_memory_less_than, 50, 0.1,
create_object, 25 * 1024 ** 2)
# Ensure ValueError is raised if memory used is above expected memory
# limit.
assert_raises_regex(ValueError,
"Memory consumption measured",
assert_memory_less_than, 100, 0.1,
create_object, 200 * 1024 ** 2)
def test_iter_check_niimgs_memory():
# Verify that iterating over a list of images doesn't consume extra
# memory.
assert_memory_less_than(
100, 0.1, _check_memory,
[Nifti1Image(np.ones((100, 100, 200)), np.eye(4)) for i in range(10)])
def test_iter_check_niimgs_memory():
# Verify that iterating over a list of images doesn't consume extra
# memory.
assert_memory_less_than(100, 0.1, _check_memory,
[Nifti1Image(np.ones((100, 100, 200)), np.eye(4))
for i in range(10)])
