diff --git a/libp2p/stream_muxer/mplex/mplex.py b/libp2p/stream_muxer/mplex/mplex.py index 00dc25c3..95336d94 100644 --- a/libp2p/stream_muxer/mplex/mplex.py +++ b/libp2p/stream_muxer/mplex/mplex.py @@ -1,12 +1,13 @@ import asyncio from typing import Dict, Tuple +from multiaddr import Multiaddr + from libp2p.network.connection.raw_connection_interface import IRawConnection from libp2p.network.typing import GenericProtocolHandlerFn from libp2p.peer.id import ID from libp2p.security.secure_conn_interface import ISecureConn from libp2p.stream_muxer.abc import IMuxedConn, IMuxedStream -from multiaddr import Multiaddr from .constants import HeaderTags from .mplex_stream import MplexStream diff --git a/tests/libp2p/test_notify.py b/tests/libp2p/test_notify.py index 31029499..f1262a1f 100644 --- a/tests/libp2p/test_notify.py +++ b/tests/libp2p/test_notify.py @@ -18,8 +18,8 @@ from libp2p.network.notifee_interface import INotifee from tests.utils import ( cleanup, echo_stream_handler, - perform_two_host_set_up_custom_handler, generate_new_private_key, + perform_two_host_set_up_custom_handler, ) diff --git a/tests/pubsub/factories.py b/tests/pubsub/factories.py index 2b3aefff..e35839dc 100644 --- a/tests/pubsub/factories.py +++ b/tests/pubsub/factories.py @@ -1,4 +1,5 @@ import factory + from libp2p import initialize_default_swarm from libp2p.host.basic_host import BasicHost from libp2p.pubsub.floodsub import FloodSub diff --git a/tests/security/test_security_multistream.py b/tests/security/test_security_multistream.py index d2e9ea9f..bf4929ba 100644 --- a/tests/security/test_security_multistream.py +++ b/tests/security/test_security_multistream.py @@ -1,12 +1,13 @@ import asyncio +import multiaddr +import pytest + from libp2p import new_node from libp2p.peer.peerinfo import info_from_p2p_addr from libp2p.protocol_muxer.multiselect_client import MultiselectClientError from libp2p.security.insecure.transport import InsecureSession, InsecureTransport from libp2p.security.simple.transport import SimpleSecurityTransport -import multiaddr -import pytest from tests.utils import cleanup, connect, generate_new_private_key # TODO: Add tests for multiple streams being opened on different diff --git a/tests/utils.py b/tests/utils.py index 6a6336d2..39368694 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,9 +1,8 @@ import asyncio from contextlib import suppress -import multiaddr - from Crypto.PublicKey import RSA +import multiaddr from libp2p import new_node from libp2p.peer.peerinfo import info_from_p2p_addr