blob: 7d799cea22152bbded142b85366f717a0524f943 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
import sys
import yaml
def main():
# various smoke tests on an installed PyYAML with extension
if not getattr(yaml, '_yaml', None):
raise Exception('C extension is not available at `yaml._yaml`')
print('embedded libyaml version is {0}'.format(yaml._yaml.get_version_string()))
for loader, dumper in [(yaml.CLoader, yaml.CDumper), (yaml.Loader, yaml.Dumper)]:
testyaml = 'dude: mar'
loaded = yaml.load(testyaml, Loader=loader)
dumped = yaml.dump(loaded, Dumper=dumper)
if testyaml != dumped.strip():
raise Exception('roundtrip failed with {0}/{1}'.format(loader, dumper))
print('smoke test passed for {0}'.format(sys.executable))
if __name__ == '__main__':
main()
|
