diff options
Diffstat (limited to 'tests/test_bip340.py')
| -rw-r--r-- | tests/test_bip340.py | 97 |
1 files changed, 18 insertions, 79 deletions
diff --git a/tests/test_bip340.py b/tests/test_bip340.py index a951492..cc9e416 100644 --- a/tests/test_bip340.py +++ b/tests/test_bip340.py @@ -1,16 +1,18 @@ -# -# The following code is only used to verify the test vectors. -# import csv -import os -import sys +from pathlib import Path +from typing import Any -def test_vectors() -> bool: - all_passed = True - with open(os.path.join(sys.path[0], "test-vectors.csv"), newline="") as csvfile: +from nostr.bip340 import pubkey_gen, schnorr_sign, schnorr_verify + +TEST_VECTORS = Path(__file__).parent / "test-vectors.csv" + + +def test_vectors(): + with open(TEST_VECTORS, newline="") as csvfile: reader = csv.reader(csvfile) - reader.__next__() + next(reader) # skip column titles + for row in reader: ( index, @@ -22,82 +24,19 @@ def test_vectors() -> bool: result_str, comment, ) = row + pubkey = bytes.fromhex(pubkey_hex) msg = bytes.fromhex(msg_hex) sig = bytes.fromhex(sig_hex) result = result_str == "TRUE" - print("\nTest vector", ("#" + index).rjust(3, " ") + ":") + if seckey_hex != "": seckey = bytes.fromhex(seckey_hex) pubkey_actual = pubkey_gen(seckey) - if pubkey != pubkey_actual: - print(" * Failed key generation.") - print(" Expected key:", pubkey.hex().upper()) - print(" Actual key:", pubkey_actual.hex().upper()) + assert pubkey == pubkey_actual aux_rand = bytes.fromhex(aux_rand_hex) - try: - sig_actual = schnorr_sign(msg, seckey, aux_rand) - if sig == sig_actual: - print(" * Passed signing test.") - else: - print(" * Failed signing test.") - print(" Expected signature:", sig.hex().upper()) - print(" Actual signature:", sig_actual.hex().upper()) - all_passed = False - except RuntimeError as e: - print(" * Signing test raised exception:", e) - all_passed = False - result_actual = schnorr_verify(msg, pubkey, sig) - if result == result_actual: - print(" * Passed verification test.") - else: - print(" * Failed verification test.") - print(" Expected verification result:", result) - print(" Actual verification result:", result_actual) - if comment: - print(" Comment:", comment) - all_passed = False - print() - if all_passed: - print("All test vectors passed.") - else: - print("Some test vectors failed.") - return all_passed - + sig_actual = schnorr_sign(msg, seckey, aux_rand) + assert sig == sig_actual -# -# The following code is only used for debugging -# -import inspect - - -def pretty(v: Any) -> Any: - if isinstance(v, bytes): - return "0x" + v.hex() - if isinstance(v, int): - return pretty(bytes_from_int(v)) - if isinstance(v, tuple): - return tuple(map(pretty, v)) - return v - - -def debug_print_vars() -> None: - if DEBUG: - current_frame = inspect.currentframe() - assert current_frame is not None - frame = current_frame.f_back - assert frame is not None - print( - " Variables in function ", - frame.f_code.co_name, - " at line ", - frame.f_lineno, - ":", - sep="", - ) - for var_name, var_val in frame.f_locals.items(): - print(" " + var_name.rjust(11, " "), "==", pretty(var_val)) - - -if __name__ == "__main__": - test_vectors() + result_actual = schnorr_verify(msg, pubkey, sig) + assert result == result_actual |