test: Add basic Python/C++ Muhash implementation parity unit test

This commit is contained in:
Fabian Jahr 2020-05-31 20:31:17 +02:00
parent b85543cb73
commit 0e2b400fea
No known key found for this signature in database
GPG Key ID: F13D1E9D890798CD
2 changed files with 12 additions and 0 deletions

View File

@ -4,6 +4,7 @@
"""Native Python MuHash3072 implementation."""
import hashlib
import unittest
from .util import modinv
@ -88,3 +89,13 @@ class MuHash3072:
val = (self.numerator * modinv(self.denominator, self.MODULUS)) % self.MODULUS
bytes384 = val.to_bytes(384, 'little')
return hashlib.sha256(bytes384).digest()
class TestFrameworkMuhash(unittest.TestCase):
def test_muhash(self):
muhash = MuHash3072()
muhash.insert([0]*32)
muhash.insert([1] + [0]*31)
muhash.remove([2] + [0]*31)
finalized = muhash.digest()
# This mirrors the result in the C++ MuHash3072 unit test
self.assertEqual(finalized[::-1].hex(), "a44e16d5e34d259b349af21c06e65d653915d2e208e4e03f389af750dc0bfdc3")

View File

@ -69,6 +69,7 @@ TEST_EXIT_SKIPPED = 77
TEST_FRAMEWORK_MODULES = [
"address",
"blocktools",
"muhash",
"script",
"util",
]