kris/pqc
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

use yaml.safe_load to silence warning.

Browse Source
This commit is contained in:
Thom Wiggers 2019-04-18 09:26:40 +02:00
parent ee680f218b
commit 98780c6e8e
No known key found for this signature in database
GPG Key ID: 001BB0A7CE26E363
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/pqclean.py
View File

@ -55,7 +55,7 @@ class Scheme:
metafile = os.path.join(self.path(), 'META.yml') metafile = os.path.join(self.path(), 'META.yml')
try: try:
with open(metafile, encoding='utf-8') as f: with open(metafile, encoding='utf-8') as f:
metadata = yaml.load(f.read()) metadata = yaml.safe_load(f.read())
return metadata return metadata
except Exception as e: except Exception as e:
print("Can't open {}: {}".format(metafile, e)) print("Can't open {}: {}".format(metafile, e))

2
test/test_duplicate_consistency.py
View File

@ -16,7 +16,7 @@ def test_duplicate_consistency():
if os.path.isfile(os.path.join('duplicate_consistency', '{}_{}.yml'.format(scheme.name, implementation.name))): if os.path.isfile(os.path.join('duplicate_consistency', '{}_{}.yml'.format(scheme.name, implementation.name))):
metafile = os.path.join('duplicate_consistency', '{}_{}.yml'.format(implementation.scheme.name, implementation.name)) metafile = os.path.join('duplicate_consistency', '{}_{}.yml'.format(implementation.scheme.name, implementation.name))
with open(metafile, encoding='utf-8') as f: with open(metafile, encoding='utf-8') as f:
metadata = yaml.load(f.read()) metadata = yaml.safe_load(f.read())
for group in metadata['consistency_checks']: for group in metadata['consistency_checks']:
source = pqclean.Implementation.by_name(group['source']['scheme'], group['source']['implementation']) source = pqclean.Implementation.by_name(group['source']['scheme'], group['source']['implementation'])
for file in group['files']: for file in group['files']: