Skip to content

Commit

Permalink
Merkle func cleanups
Browse files Browse the repository at this point in the history
  • Loading branch information
lee195 committed Jun 17, 2019
1 parent 6a630bd commit c476409
Show file tree
Hide file tree
Showing 4 changed files with 11 additions and 12 deletions.
9 changes: 3 additions & 6 deletions Block.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,13 +38,10 @@ def hash_data(data_a, data_b):
self.root = layer[0]

def __repr__(self):
out = ""
for layer in self.data[::-1]:
out += str(layer) + "\n"
return out[:-1]
return ''.join(str(layer) for layer in self.data[::-1])[:-1]

def verify(tx):
return tx in data[0]
def verify(self, tx):
return tx in self.data[0]


class Header:
Expand Down
Binary file modified __pycache__/Block.cpython-37.pyc
Binary file not shown.
Binary file added __pycache__/test_block.cpython-37.pyc
Binary file not shown.
14 changes: 8 additions & 6 deletions test_block.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,18 +8,20 @@ class TestBlock(unittest.TestCase):
def setUpClass(cls):
random.seed(0)

def test_genesis(self):
expected_hash = '810043e3e9dbfe3616a8b038f2886379bc9a04585c968f1ff9d291497766e980'
expected_tx = ['Genesis tx']
gen = Block.generate_genesis()
self.assertEqual(expected_hash, gen.hashed)
self.assertEqual(expected_tx, gen.get_tx())
# def test_genesis(self):
# expected_hash = '810043e3e9dbfe3616a8b038f2886379bc9a04585c968f1ff9d291497766e980'
# expected_tx = ['Genesis tx']
# gen = Block.generate_genesis()
# self.assertEqual(expected_hash, gen.hashed)
# self.assertEqual(expected_tx, gen.get_tx())

def test_merkle(self):
expected_root = '88cd668c2056e926cf9f6dad3acbeebf0c1e093da5ab7aceb244e65661d7e35e'

merkle_tree = Block.MerkleTree([1, 2, 3, 4])
self.assertEqual(expected_root, merkle_tree.root)
for tx in [1, 2, 3, 4]:
self.assertTrue(merkle_tree.verify(tx))


if __name__ == "__main__":
Expand Down

0 comments on commit c476409

Please sign in to comment.