diff options
Diffstat (limited to 'tests/test_errors.py')
-rw-r--r-- | tests/test_errors.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tests/test_errors.py b/tests/test_errors.py index 2b6e9a2..431258b 100644 --- a/tests/test_errors.py +++ b/tests/test_errors.py @@ -18,14 +18,12 @@ class TestErrors(test_appliance.TestAppliance): def _load(self, filename): reader = Reader(file(filename, 'rb')) scanner = Scanner(reader) - while scanner.peek_token(): - scanner.get_token() + return list(scanner) def _load_string(self, filename): reader = Reader(file(filename, 'rb').read()) scanner = Scanner(reader) - while scanner.peek_token(): - scanner.get_token() + return list(scanner) TestErrors.add_tests('testErrors', '.error-message') TestErrors.add_tests('testStringErrors', '.error-message') |