diff options
author | Bryan Newbold <bnewbold@robocracy.org> | 2019-05-30 16:34:24 -0700 |
---|---|---|
committer | Bryan Newbold <bnewbold@robocracy.org> | 2019-05-30 16:34:24 -0700 |
commit | 87f31945500b13b2399621fe0787e866501bb599 (patch) | |
tree | ae3933bb220835526c18d08e1cef80d74096b5f0 /parse_merge_metadata.py | |
parent | a4d8566cfdddb213b6c5ba3547ff7d336b097681 (diff) | |
download | chocula-87f31945500b13b2399621fe0787e866501bb599.tar.gz chocula-87f31945500b13b2399621fe0787e866501bb599.zip |
fix parse_merge_metadata.py merge_spans()
Diffstat (limited to 'parse_merge_metadata.py')
-rwxr-xr-x | parse_merge_metadata.py | 12 |
1 files changed, 8 insertions, 4 deletions
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(): |