From 9776cea9f0802f04ed280df71299257a27a9124d Mon Sep 17 00:00:00 2001 From: Luke Murphy Date: Sat, 16 May 2020 18:37:43 +0200 Subject: [PATCH] Appease black --- merkle_tree_stream/__init__.py | 4 ++-- merkle_tree_stream/node.py | 2 +- merkle_tree_stream/tree.py | 4 ++-- test/test_tree.py | 28 ++++++++++++++-------------- test/test_version.py | 4 ++-- 5 files changed, 21 insertions(+), 21 deletions(-) diff --git a/merkle_tree_stream/__init__.py b/merkle_tree_stream/__init__.py index 9465804..71345c2 100644 --- a/merkle_tree_stream/__init__.py +++ b/merkle_tree_stream/__init__.py @@ -10,6 +10,6 @@ except ImportError: try: - __version__ = pkg_resources.get_distribution('merkle_tree_stream').version + __version__ = pkg_resources.get_distribution("merkle_tree_stream").version except Exception: - __version__ = 'unknown' + __version__ = "unknown" diff --git a/merkle_tree_stream/node.py b/merkle_tree_stream/node.py index 3fb1689..656565a 100644 --- a/merkle_tree_stream/node.py +++ b/merkle_tree_stream/node.py @@ -5,7 +5,7 @@ from typing import Any, Optional import attr from flat_tree import FlatTreeAccessor -__all__ = ['MerkleTreeNode'] +__all__ = ["MerkleTreeNode"] @attr.s(auto_attribs=True) diff --git a/merkle_tree_stream/tree.py b/merkle_tree_stream/tree.py index fda0198..7c45733 100644 --- a/merkle_tree_stream/tree.py +++ b/merkle_tree_stream/tree.py @@ -7,10 +7,10 @@ from flat_tree import FlatTreeAccessor from merkle_tree_stream.node import MerkleTreeNode -EMPTY_DATA = b'' +EMPTY_DATA = b"" EMPTY_HASH = None -__all__ = ['MerkleTreeGenerator'] +__all__ = ["MerkleTreeGenerator"] flat_tree = FlatTreeAccessor() diff --git a/test/test_tree.py b/test/test_tree.py index 23c0923..7bb143e 100644 --- a/test/test_tree.py +++ b/test/test_tree.py @@ -10,25 +10,25 @@ from merkle_tree_stream import MerkleTreeGenerator, MerkleTreeNode def test_hashes(leaf, parent): merkle = MerkleTreeGenerator(leaf=leaf, parent=parent) - merkle.write(b'a') - merkle.write(b'b') + merkle.write(b"a") + merkle.write(b"b") expected_count = 2 + 1 # nodes plus parent assert len(merkle) == expected_count assert next(merkle) == MerkleTreeNode( - index=0, parent=1, hash=hashlib.sha256(b'a').digest(), size=1, data=b'a' + index=0, parent=1, hash=hashlib.sha256(b"a").digest(), size=1, data=b"a" ) assert next(merkle) == MerkleTreeNode( - index=2, parent=1, hash=hashlib.sha256(b'b').digest(), size=1, data=b'b' + index=2, parent=1, hash=hashlib.sha256(b"b").digest(), size=1, data=b"b" ) - hashed = hashlib.sha256(b'a') - hashed.update(b'b') + hashed = hashlib.sha256(b"a") + hashed.update(b"b") assert next(merkle) == MerkleTreeNode( - index=1, parent=3, hash=hashed.digest(), size=2, data=b'' + index=1, parent=3, hash=hashed.digest(), size=2, data=b"" ) with pytest.raises(StopIteration): @@ -38,10 +38,10 @@ def test_hashes(leaf, parent): def test_single_root(leaf, parent): merkle = MerkleTreeGenerator(leaf=leaf, parent=parent) - merkle.write(b'a') - merkle.write(b'b') - merkle.write(b'c') - merkle.write(b'd') + merkle.write(b"a") + merkle.write(b"b") + merkle.write(b"c") + merkle.write(b"d") assert len(merkle.roots) == 1 @@ -49,8 +49,8 @@ def test_single_root(leaf, parent): def multiple_roots(leaf, parent): merkle = MerkleTreeGenerator(leaf=leaf, parent=parent) - merkle.write(b'a') - merkle.write(b'b') - merkle.write(b'c') + merkle.write(b"a") + merkle.write(b"b") + merkle.write(b"c") assert len(merkle.roots) > 1 diff --git a/test/test_version.py b/test/test_version.py index 8cab290..561699a 100644 --- a/test/test_version.py +++ b/test/test_version.py @@ -2,9 +2,9 @@ def test_version_fails_gracefully(mocker): - target = 'pkg_resources.get_distribution' + target = "pkg_resources.get_distribution" mocker.patch(target, side_effect=Exception()) from merkle_tree_stream.__init__ import __version__ - assert __version__ == 'unknown' + assert __version__ == "unknown"