From 87f31945500b13b2399621fe0787e866501bb599 Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Thu, 30 May 2019 16:34:24 -0700 Subject: fix parse_merge_metadata.py merge_spans() --- parse_merge_metadata.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'parse_merge_metadata.py') diff --git a/parse_merge_metadata.py b/parse_merge_metadata.py index 7e75cf8..b1d038b 100755 --- a/parse_merge_metadata.py +++ b/parse_merge_metadata.py @@ -129,7 +129,7 @@ def merge_spans(old, new): old.extend(new) years = set() for span in old: - for y in range(span[0], span[1]): + for y in range(span[0], span[1]+1): years.add(y) if not years: return [] @@ -153,19 +153,23 @@ def merge_spans(old, new): spans.append([start, last]) start = y last = y - todo = False + todo = True if todo: spans.append([start, last]) return spans def test_merge_spans(): - assert merge_spans([[5, 10]], [10, 20]) == \ + assert merge_spans([[5, 10]], [[10, 20]]) == \ + [[5, 20]] + assert merge_spans([[5, 9]], [[10, 20]]) == \ + [[5, 20]] + assert merge_spans([[5, 11]], [[10, 20]]) == \ [[5, 20]] assert merge_spans([], []) == \ [] assert merge_spans([[9, 11]], []) == \ [[9,11]] - assert merge_spans([[2000, 2000]], [1450, 1900]) == \ + assert merge_spans([[2000, 2000]], [[1450, 1900]]) == \ [[1450, 1900], [2000, 2000]] class Munger(): -- cgit v1.2.3