diff --git a/memory_profiler.py b/memory_profiler.py index 9de2ccd..6573e58 100644 --- a/memory_profiler.py +++ b/memory_profiler.py @@ -597,9 +597,7 @@ def trace(self, code, lineno, prev_lineno): prev_line_value = self[code].get(prev_lineno, None) if prev_lineno else None prev_line_memory = prev_line_value[1] if prev_line_value else 0 - #inc = (memory-prev_line_memory) - #print('trace lineno=%(lineno)s prev_lineno=%(prev_lineno)s mem=%(memory)s prev_inc=%(previous_inc)s inc=%(inc)s' % locals()) - self[code][lineno] = (previous_inc + (memory-prev_line_memory), max(memory, previous_memory)) + self[code][lineno] = (max(previous_inc, memory-prev_line_memory), max(memory, previous_memory)) def items(self): """Iterate on the toplevel code blocks.""" diff --git a/test/test_tracemalloc.py b/test/test_tracemalloc.py index 5bedc50..4e8006c 100644 --- a/test/test_tracemalloc.py +++ b/test/test_tracemalloc.py @@ -18,8 +18,6 @@ def test_memory_profiler(test_input, expected): mem_prof(test_input) inc, dec = parse_mem_prof() - assert abs(inc - dec) <= EPSILON, \ - 'inc = {}, dec = {}, err = {}'.format(inc, dec, abs(inc - dec)) assert abs(inc - expected) <= EPSILON, \ 'inc = {}, size = {}, err = {}'.format( inc, expected, abs(inc - expected)