Ensure tests do not break without schemes

This commit is contained in:
Joost Rijneveld 2019-04-08 11:14:45 +02:00
parent e7602479f1
commit 493d77d89b
No known key found for this signature in database
GPG Key ID: A4FE39CF49CBC553

View File

@ -40,14 +40,15 @@ class Scheme:
def all_schemes_of_type(type: str) -> list: def all_schemes_of_type(type: str) -> list:
schemes = [] schemes = []
p = os.path.join('..', 'crypto_' + type) p = os.path.join('..', 'crypto_' + type)
for d in os.listdir(p): if os.path.isdir(p):
if os.path.isdir(os.path.join(p, d)): for d in os.listdir(p):
if type == 'kem': if os.path.isdir(os.path.join(p, d)):
schemes.append(KEM(d)) if type == 'kem':
elif type == 'sign': schemes.append(KEM(d))
schemes.append(Signature(d)) elif type == 'sign':
else: schemes.append(Signature(d))
assert('Unknown type') else:
assert('Unknown type')
return schemes return schemes
def metadata(self): def metadata(self):