Skip to content

Commit 0e622de

Browse files
committed
Include removed modules in table calulation
1 parent 814dbdb commit 0e622de

File tree

1 file changed

+10
-11
lines changed

1 file changed

+10
-11
lines changed

tools/memap.py

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -514,17 +514,16 @@ def reduce_depth(self, depth):
514514
self.short_modules.setdefault(new_name, defaultdict(int))
515515
for section_idx, value in v.items():
516516
self.short_modules[new_name][section_idx] += self.modules[module_name][section_idx]
517-
try:
518-
new_size = self.modules[module_name][section_idx]
519-
try:
520-
old_size = self.old_modules[module_name][section_idx]
521-
except KeyError:
522-
old_size = 0
523-
self.short_modules[new_name][section_idx + '-delta'] += (
524-
new_size - old_size
525-
)
526-
except TypeError:
527-
self.short_modules[new_name][section_idx + '-delta'] += 0
517+
self.short_modules[new_name][section_idx + '-delta'] += self.modules[module_name][section_idx]
518+
if self.old_modules:
519+
for module_name, v in self.old_modules.items():
520+
split_name = module_name.split(sep)
521+
if split_name[0] == '':
522+
split_name = split_name[1:]
523+
new_name = join(*split_name[:depth])
524+
self.short_modules.setdefault(new_name, defaultdict(int))
525+
for section_idx, value in v.items():
526+
self.short_modules[new_name][section_idx + '-delta'] -= self.old_modules[module_name][section_idx]
528527

529528
export_formats = ["json", "csv-ci", "html", "table"]
530529

0 commit comments

Comments
 (0)