def test_constructor_types(data_filename, code_filename, verbose=False):
_make_objects()
native1 = None
native2 = None
try:
with open(data_filename, 'rb') as fp0:
native1 = list(ruyaml.load_all(fp0, Loader=MyLoader))
if len(native1) == 1:
native1 = native1[0]
with open(code_filename, 'rb') as fp0:
native2 = _load_code(fp0.read())
try:
if native1 == native2:
return
except TypeError:
pass
# print('native1', native1)
if verbose:
print('SERIALIZED NATIVE1:')
print(_serialize_value(native1))
print('SERIALIZED NATIVE2:')
print(_serialize_value(native2))
assert _serialize_value(native1) == _serialize_value(native2), (
native1,
native2,
)
finally:
if verbose:
print('NATIVE1:')
pprint.pprint(native1)
print('NATIVE2:')
pprint.pprint(native2)
def test_roundtrip_data(code_filename, roundtrip_filename, verbose=False):
_make_objects()
with open(code_filename, 'rb') as fp0:
value1 = fp0.read()
native2 = list(ruyaml.load_all(value1, Loader=MyLoader))
if len(native2) == 1:
native2 = native2[0]
try:
value2 = ruyaml.dump(
native2,
Dumper=MyDumper,
default_flow_style=False,
allow_unicode=True,
encoding='utf-8',
)
# value2 += x
if verbose:
print('SERIALIZED NATIVE1:')
print(value1)
print('SERIALIZED NATIVE2:')
print(value2)
assert value1 == value2, (value1, value2)
finally:
if verbose:
print('NATIVE2:')
pprint.pprint(native2)
def test_constructor(data_filename, canonical_filename, verbose=False):
_make_loader()
_make_canonical_loader()
native1 = None
native2 = None
try:
with open(data_filename, 'rb') as fp0:
native1 = list(yaml.load_all(fp0, Loader=MyLoader))
with open(canonical_filename, 'rb') as fp0:
native2 = list(yaml.load_all(fp0, Loader=MyCanonicalLoader))
assert native1 == native2, (native1, native2)
finally:
if verbose:
print('NATIVE1:')
pprint.pprint(native1)
print('NATIVE2:')
pprint.pprint(native2)
def test_loader_error_string(error_filename, verbose=False):
try:
with open(error_filename, 'rb') as fp0:
list(yaml.load_all(fp0.read()))
except yaml.YAMLError as exc:
if verbose:
print('%s:' % exc.__class__.__name__, exc)
else:
raise AssertionError('expected an exception')
def test_loader_error(error_filename, verbose=False):
yaml = YAML(typ='safe', pure=True)
try:
with open(error_filename, 'rb') as fp0:
list(yaml.load_all(fp0))
except yaml.YAMLError as exc:
if verbose:
print('%s:' % exc.__class__.__name__, exc)
else:
raise AssertionError('expected an exception')
def round_trip_load_all(inp, preserve_quotes=None, version=None):
import ruyaml # NOQA
dinp = dedent(inp)
return ruyaml.load_all(
dinp,
Loader=ruyaml.RoundTripLoader,
preserve_quotes=preserve_quotes,
version=version,
)
def canonical_load_all(stream):
return ruyaml.load_all(stream, Loader=CanonicalLoader)
