mirror of
https://github.com/yaml/pyyaml.git
synced 2025-10-19 19:13:19 +00:00
35 lines
957 B
Python
35 lines
957 B
Python
|
|
||
|
|
import yaml.reader
|
||
|
|
|
||
|
|
def _run_reader(data, verbose):
|
||
|
|
try:
|
||
|
|
stream = yaml.reader.Reader(data)
|
||
|
|
while stream.peek() != '\0':
|
||
|
|
stream.forward()
|
||
|
|
except yaml.reader.ReaderError as exc:
|
||
|
|
if verbose:
|
||
|
|
print(exc)
|
||
|
|
else:
|
||
|
|
raise AssertionError("expected an exception")
|
||
|
|
|
||
|
|
def test_stream_error(error_filename, verbose=False):
|
||
|
|
_run_reader(open(error_filename, 'rb'), verbose)
|
||
|
|
_run_reader(open(error_filename, 'rb').read(), verbose)
|
||
|
|
for encoding in ['utf-8', 'utf-16-le', 'utf-16-be']:
|
||
|
|
try:
|
||
|
|
data = open(error_filename, 'rb').read().decode(encoding)
|
||
|
|
break
|
||
|
|
except UnicodeDecodeError:
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
return
|
||
|
|
_run_reader(data, verbose)
|
||
|
|
_run_reader(open(error_filename, encoding=encoding), verbose)
|
||
|
|
|
||
|
|
test_stream_error.unittest = ['.stream-error']
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
import test_appliance
|
||
|
|
test_appliance.run(globals())
|
||
|
|
|