aboutsummaryrefslogtreecommitdiff
path: root/examples/05-python
diff options
context:
space:
mode:
Diffstat (limited to 'examples/05-python')
-rw-r--r--examples/05-python/sha3.py28
1 files changed, 14 insertions, 14 deletions
diff --git a/examples/05-python/sha3.py b/examples/05-python/sha3.py
index bd0e11b..1e66c59 100644
--- a/examples/05-python/sha3.py
+++ b/examples/05-python/sha3.py
@@ -17,10 +17,10 @@ sha3 = ctypes.CDLL('../../libsha3.so')
# set function argument and return types
# (note: this is not required, but it helps to prevent argument and type
# oopsies)
-sha3.shake128_xof_once.argtypes = XOF_ARGS
-sha3.shake128_xof_once.rettype = None
-sha3.shake256_xof_once.argtypes = XOF_ARGS
-sha3.shake256_xof_once.rettype = None
+sha3.shake128.argtypes = XOF_ARGS
+sha3.shake128.rettype = None
+sha3.shake256.argtypes = XOF_ARGS
+sha3.shake256.rettype = None
sha3.sha3_224.argtypes = HASH_ARGS
sha3.sha3_224.rettype = None
sha3.sha3_256.argtypes = HASH_ARGS
@@ -66,9 +66,9 @@ class Results:
for r in self.rs:
cw.writerow(r.to_row())
-def test_shake128_xof(rs: Results, data: bytes) -> None:
- """append SHAKE128-XOF digest of data from hashlib and libsha3 to results"""
- algo = 'shake128-xof' # algorithm name
+def test_shake128(rs: Results, data: bytes) -> None:
+ """append SHAKE128 digest of data from hashlib and libsha3 to results"""
+ algo = 'shake128' # algorithm name
size = 32 # output size, in bytes
# append hashlib digest
@@ -76,12 +76,12 @@ def test_shake128_xof(rs: Results, data: bytes) -> None:
# append libsha3 digest
buf = ctypes.create_string_buffer(size)
- sha3.shake128_xof_once(data, len(data), buf, size)
+ sha3.shake128(data, len(data), buf, size)
rs.append(algo, 'libsha3', buf.raw)
-def test_shake256_xof(rs: Results, data: bytes) -> None:
- """append SHAKE256-XOF digest of data from hashlib and libsha3 to results"""
- algo = 'shake256-xof' # algorithm name
+def test_shake256(rs: Results, data: bytes) -> None:
+ """append SHAKE256 digest of data from hashlib and libsha3 to results"""
+ algo = 'shake256' # algorithm name
size = 32 # output size, in bytes
# append hashlib digest
@@ -89,7 +89,7 @@ def test_shake256_xof(rs: Results, data: bytes) -> None:
# append libsha3 digest
buf = ctypes.create_string_buffer(size)
- sha3.shake256_xof_once(data, len(data), buf, size)
+ sha3.shake256(data, len(data), buf, size)
rs.append(algo, 'libsha3', buf.raw)
def test_sha3_224(rs: Results, data: bytes) -> None:
@@ -145,8 +145,8 @@ DATA = b'foo bar'
rs = Results()
# run tests, add to results
-test_shake128_xof(rs, DATA)
-test_shake256_xof(rs, DATA)
+test_shake128(rs, DATA)
+test_shake256(rs, DATA)
test_sha3_224(rs, DATA)
test_sha3_256(rs, DATA)
test_sha3_384(rs, DATA)