update: move tests

olivmath · Dec 31, 2023 · fcd24fe · fcd24fe
1 parent 9788ce4
commit fcd24fe
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/test/test_accelerator.py → test/accelerator/test_accelerator.py b/test/test_accelerator.py → test/accelerator/test_accelerator.py
@@ -24,7 +24,10 @@ def test_merkle_root_rust():
         print(x.hex())
 
     result = tree.make_root(leafs)
-    assert result.hex() == "6b403b6dbdd79d6bb882036292bbe89a57e10defabd5c6718e66321c79b96abd", "Rust merkle root"
+    assert (
+        result.hex()
+        == "6b403b6dbdd79d6bb882036292bbe89a57e10defabd5c6718e66321c79b96abd"
+    ), "Rust merkle root"
 
 
 def test_merkle_root_python():
@@ -36,6 +39,5 @@ def test_merkle_root_python():
 
     result = tree.make_root(leafs)
     assert (
-        result[0]
-        == "6b403b6dbdd79d6bb882036292bbe89a57e10defabd5c6718e66321c79b96abd"
+        result[0] == "6b403b6dbdd79d6bb882036292bbe89a57e10defabd5c6718e66321c79b96abd"
     ), "Rust merkle root"
diff --git a/test/merkletreejs/index.py b/test/merkletreejs/index.py
@@ -2,14 +2,17 @@
 
 import hashlib
 
+
 def hashing(data):
     return hashlib.sha256(data.encode()).hexdigest()
 
+
 def hashing2(data: str):
     keccak_256 = keccak.new(digest_bits=256)
     keccak_256.update(data.encode())
     return keccak_256.hexdigest()
 
+
 class MerkleTree:
     def __init__(self, leaves):
         self.leaves = [hashing2(leaf) for leaf in leaves]
@@ -31,12 +34,12 @@ def get_merkle_root(self):
         return self.tree[-1][0] if self.tree else None
 
 
-
-leaves = ['a', 'b', 'c', 'd']
+leaves = ["a", "b", "c", "d"]
 tree = MerkleTree(leaves)
 
 root = tree.get_merkle_root()
 leafs = tree.leaves
 
-for i in leafs: print(i)
+for i in leafs:
+    print(i)
 print(root)