From e046af8ce83f64f661984c78307890e1719c4090 Mon Sep 17 00:00:00 2001 From: Jan Petykiewicz Date: Sat, 18 Apr 2020 02:54:33 -0700 Subject: [PATCH] Handle more error cases --- fatamorgana/basic.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/fatamorgana/basic.py b/fatamorgana/basic.py index 76a1d39..8a02a38 100644 --- a/fatamorgana/basic.py +++ b/fatamorgana/basic.py @@ -953,6 +953,8 @@ def read_repetition(stream: io.BufferedIOBase) -> repetition_t: return GridRepetition.read(stream, rtype) elif rtype in (4, 5, 6, 7, 10, 11): return ArbitraryRepetition.read(stream, rtype) + else: + raise InvalidDataError('Unexpected repetition type: {}'.format(rtype)) def write_repetition(stream: io.BufferedIOBase, repetition: repetition_t) -> int: @@ -1357,7 +1359,7 @@ def read_point_list(stream: io.BufferedIOBase) -> List[List[int]]: y += delta.y points.append([x, y]) else: - raise Exception('Invalid point list type') + raise InvalidDataError('Invalid point list type') return points