def test_constructor(data_filename, canonical_filename, verbose=False): _make_loader() _make_canonical_loader() native1 = None native2 = None try: native1 = list(yaml.load_all(open(data_filename, 'rb'), Loader=MyLoader)) native2 = list(yaml.load_all(open(canonical_filename, 'rb'), Loader=MyCanonicalLoader)) assert native1 == native2, (native1, native2) finally: if verbose: print "NATIVE1:" pprint.pprint(native1) print "NATIVE2:" pprint.pprint(native2)
def test_constructor_types(data_filename, code_filename, verbose=False): _make_objects() native1 = None native2 = None try: native1 = list(yaml.load_all(open(data_filename, 'rb'), Loader=MyLoader)) if len(native1) == 1: native1 = native1[0] native2 = _load_code(open(code_filename, 'rb').read()) try: if native1 == native2: return except TypeError: pass 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 canonical_load_all(stream): return yaml.load_all(stream, Loader=CanonicalLoader)
def test_loader_error(error_filename, verbose=False): try: list(yaml.load_all(open(error_filename, "rb"))) except yaml.YAMLError, exc: if verbose: print "%s:" % exc.__class__.__name__, exc