Browse Source

refactor testvector check

master
Matthias J. Kannwischer 5 years ago
parent
commit
8c68841417
1 changed files with 8 additions and 8 deletions
  1. +8
    -8
      test/check_testvectors.py

+ 8
- 8
test/check_testvectors.py View File

@@ -18,9 +18,9 @@ SCHEMEFULL = SCHEME.replace('/', '_') # e.g. crypto_kem_kyber768
SCHEMESHORT = SCHEME.split('/')[1].upper()

def get_hash(scheme):
with open(f"{scheme}/META.yml", 'r') as stream:
meta = yaml.load(stream)
return hex(meta['testvectors-sha3-256']).replace('0x', '')
with open(f"{scheme}/META.yml", 'r') as stream:
meta = yaml.load(stream)
return hex(meta['testvectors-sha3-256']).replace('0x', '')


expectedTestvectorsHash = get_hash(SCHEME)
@@ -29,10 +29,10 @@ subprocess.run(["make", "testvectors", f"SCHEME={SCHEME}"])
implementations = [x for x in os.listdir('bin') if 'testvectors' in x and SCHEMEFULL in x]

for impl in implementations:
testvectors = subprocess.run([f"bin/{impl}"],stdout=subprocess.PIPE)
testvectorsHash = hashlib.sha3_256(testvectors.stdout).hexdigest()
if testvectorsHash.lower() != expectedTestvectorsHash.lower():
print(f"testvectors of {SCHEME} should be {expectedTestvectorsHash}, but is {testvectorsHash}")
sys.exit(1)
testvectors = subprocess.run([f"bin/{impl}"],stdout=subprocess.PIPE)
testvectorsHash = hashlib.sha3_256(testvectors.stdout).hexdigest()
if testvectorsHash.lower() != expectedTestvectorsHash.lower():
print(f"testvectors of {SCHEME} should be {expectedTestvectorsHash}, but is {testvectorsHash}")
sys.exit(1)



Loading…
Cancel
Save