Skip to content

Commit

Permalink
tests, isort, black
Browse files Browse the repository at this point in the history
  • Loading branch information
dvmazur committed Nov 4, 2024
1 parent 7567d3d commit 227d628
Show file tree
Hide file tree
Showing 24 changed files with 823 additions and 31 deletions.
2 changes: 1 addition & 1 deletion hivemind/dht/dht.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,10 +8,10 @@
from typing import Awaitable, Callable, Iterable, List, Optional, Sequence, TypeVar, Union

from hivemind.dht.node import DEFAULT_NUM_WORKERS, DHTNode
from hivemind.p2p.multiaddr import Multiaddr
from hivemind.dht.routing import DHTKey, DHTValue, Subkey
from hivemind.dht.validation import CompositeValidator, RecordValidatorBase
from hivemind.p2p import P2P, PeerID
from hivemind.p2p.multiaddr import Multiaddr
from hivemind.utils import MPFuture, get_logger, switch_to_uvloop
from hivemind.utils.timed_storage import DHTExpiration, ValueWithExpiration

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import importlib


# These are special sizes
LENGTH_PREFIXED_VAR_SIZE = -1

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/cid.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@

from . import LENGTH_PREFIXED_VAR_SIZE


SIZE = LENGTH_PREFIXED_VAR_SIZE
IS_PATH = False

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/domain.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@

from . import LENGTH_PREFIXED_VAR_SIZE


SIZE = LENGTH_PREFIXED_VAR_SIZE
IS_PATH = False

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/fspath.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@

from . import LENGTH_PREFIXED_VAR_SIZE


SIZE = LENGTH_PREFIXED_VAR_SIZE
IS_PATH = True

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/ip4.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import netaddr


SIZE = 32
IS_PATH = False

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/ip6.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import netaddr


SIZE = 128
IS_PATH = False

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/onion.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import base64
import struct


SIZE = 96
IS_PATH = False

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/onion3.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import base64
import struct


SIZE = 296
IS_PATH = False

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/uint16be.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import struct


SIZE = 16
IS_PATH = False

Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/multiaddr/codecs/utf8.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@

from . import LENGTH_PREFIXED_VAR_SIZE


SIZE = LENGTH_PREFIXED_VAR_SIZE
IS_PATH = False

Expand Down
6 changes: 1 addition & 5 deletions hivemind/p2p/multiaddr/multiaddr.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,7 @@
import varint

from . import exceptions, protocols

from .transforms import bytes_iter
from .transforms import string_to_bytes
from .transforms import bytes_to_string

from .transforms import bytes_iter, bytes_to_string, string_to_bytes

__all__ = ("Multiaddr",)

Expand Down
9 changes: 3 additions & 6 deletions hivemind/p2p/multiaddr/transforms.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,10 @@
import io

import varint

from . import exceptions

from .codecs import LENGTH_PREFIXED_VAR_SIZE
from .codecs import codec_by_name

from .protocols import protocol_with_code
from .protocols import protocol_with_name
from .codecs import LENGTH_PREFIXED_VAR_SIZE, codec_by_name
from .protocols import protocol_with_code, protocol_with_name


def string_to_bytes(string):
Expand Down
2 changes: 1 addition & 1 deletion hivemind/p2p/p2p_daemon.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,10 @@

import hivemind.hivemind_cli as cli
import hivemind.p2p.p2p_daemon_bindings.p2pclient as p2pclient
from hivemind.p2p.multiaddr import Multiaddr
from hivemind.p2p.p2p_daemon_bindings.control import DEFAULT_MAX_MSG_SIZE, P2PDaemonError, P2PHandlerError
from hivemind.p2p.p2p_daemon_bindings.datastructures import PeerID, PeerInfo, StreamInfo
from hivemind.p2p.p2p_daemon_bindings.utils import ControlFailure
from hivemind.p2p.multiaddr import Multiaddr
from hivemind.proto import crypto_pb2
from hivemind.proto.p2pd_pb2 import RPCError
from hivemind.utils.asyncio import as_aiter, asingle
Expand Down
2 changes: 1 addition & 1 deletion hivemind/p2p/p2p_daemon_bindings/control.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,8 @@
from typing import AsyncIterator, Awaitable, Callable, Dict, Iterable, Optional, Sequence, Tuple
from uuid import UUID, uuid4

from hivemind.p2p.p2p_daemon_bindings.datastructures import PeerID, PeerInfo, StreamInfo
from hivemind.p2p.multiaddr import Multiaddr, protocols
from hivemind.p2p.p2p_daemon_bindings.datastructures import PeerID, PeerInfo, StreamInfo
from hivemind.p2p.p2p_daemon_bindings.utils import (
DispatchFailure,
P2PDaemonError,
Expand Down
2 changes: 1 addition & 1 deletion hivemind/p2p/p2p_daemon_bindings/datastructures.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@
import multihash
from cryptography.hazmat.primitives import serialization

from hivemind.proto import crypto_pb2, p2pd_pb2
from hivemind.p2p.multiaddr import Multiaddr, protocols
from hivemind.proto import crypto_pb2, p2pd_pb2


class PeerID:
Expand Down
1 change: 0 additions & 1 deletion hivemind/p2p/p2p_daemon_bindings/p2pclient.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
from contextlib import asynccontextmanager
from typing import AsyncIterator, Iterable, Sequence, Tuple


from hivemind.p2p.multiaddr import Multiaddr
from hivemind.p2p.p2p_daemon_bindings.control import (
DEFAULT_MAX_MSG_SIZE,
Expand Down
2 changes: 1 addition & 1 deletion hivemind/utils/networking.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
from ipaddress import ip_address
from typing import List, Sequence

from hivemind.utils.logging import TextStyle, get_logger
from hivemind.p2p.multiaddr import Multiaddr
from hivemind.utils.logging import TextStyle, get_logger

LOCALHOST = "127.0.0.1"

Expand Down
3 changes: 1 addition & 2 deletions tests/test_dht.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,11 @@
import pytest

import hivemind
from hivemind.p2p.multiaddr import Multiaddr

from test_utils.dht_swarms import launch_dht_instances
from test_utils.networking import get_free_port

from hivemind.p2p.multiaddr import Multiaddr


@pytest.mark.asyncio
async def test_startup_error():
Expand Down
Loading

0 comments on commit 227d628

Please sign in to comment.