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)])