revert import path changes

This commit is contained in:
Youngjoon Lee 2024-07-19 16:26:12 +09:00
parent 1b4a8609ba
commit 0e834e0d05
No known key found for this signature in database
GPG Key ID: B4253AFBA618BF4D
13 changed files with 33 additions and 37 deletions

View File

View File

@ -2,7 +2,7 @@ from typing import TypeAlias
import blspy import blspy
from .utils import random_bytes from mixnet.utils import random_bytes
BlsPrivateKey: TypeAlias = blspy.PrivateKey BlsPrivateKey: TypeAlias = blspy.PrivateKey
BlsPublicKey: TypeAlias = blspy.G1Element BlsPublicKey: TypeAlias = blspy.G1Element

View File

@ -4,10 +4,10 @@ import asyncio
from contextlib import suppress from contextlib import suppress
from typing import Self from typing import Self
from .config import MixClientConfig, MixnetTopology from mixnet.config import MixClientConfig, MixnetTopology
from .node import PacketQueue from mixnet.node import PacketQueue
from .packet import PacketBuilder from mixnet.packet import PacketBuilder
from .poisson import poisson_interval_sec from mixnet.poisson import poisson_interval_sec
class MixClient: class MixClient:

View File

@ -10,8 +10,8 @@ from cryptography.hazmat.primitives.asymmetric.x25519 import (
) )
from pysphinx.node import Node from pysphinx.node import Node
from .bls import BlsPrivateKey, BlsPublicKey from mixnet.bls import BlsPrivateKey, BlsPublicKey
from .fisheryates import FisherYates from mixnet.fisheryates import FisherYates
@dataclass @dataclass

View File

@ -4,13 +4,9 @@ import asyncio
from contextlib import suppress from contextlib import suppress
from typing import Self, TypeAlias from typing import Self, TypeAlias
from .client import MixClient from mixnet.client import MixClient
from .config import ( from mixnet.config import MixnetConfig, MixnetTopology, MixnetTopologyConfig
MixnetConfig, from mixnet.node import MixNode
MixnetTopology,
MixnetTopologyConfig,
)
from .node import MixNode
EntropyQueue: TypeAlias = "asyncio.Queue[bytes]" EntropyQueue: TypeAlias = "asyncio.Queue[bytes]"

View File

@ -15,8 +15,8 @@ from pysphinx.sphinx import (
UnknownHeaderTypeError, UnknownHeaderTypeError,
) )
from .config import MixNodeConfig, NodeAddress from mixnet.config import MixNodeConfig, NodeAddress
from .poisson import poisson_interval_sec from mixnet.poisson import poisson_interval_sec
PacketQueue: TypeAlias = "asyncio.Queue[Tuple[NodeAddress, SphinxPacket]]" PacketQueue: TypeAlias = "asyncio.Queue[Tuple[NodeAddress, SphinxPacket]]"
PacketPayloadQueue: TypeAlias = ( PacketPayloadQueue: TypeAlias = (

View File

@ -9,7 +9,7 @@ from typing import Dict, List, Self, Tuple, TypeAlias
from pysphinx.payload import Payload from pysphinx.payload import Payload
from pysphinx.sphinx import SphinxPacket from pysphinx.sphinx import SphinxPacket
from .config import MixnetTopology, MixNodeInfo from mixnet.config import MixnetTopology, MixNodeInfo
class MessageFlag(Enum): class MessageFlag(Enum):

View File

@ -3,13 +3,13 @@ from unittest import IsolatedAsyncioTestCase
import numpy import numpy
from .client import MixClient from mixnet.client import MixClient
from .poisson import poisson_mean_interval_sec from mixnet.poisson import poisson_mean_interval_sec
from .test_utils import ( from mixnet.test_utils import (
init_mixnet_config, init_mixnet_config,
with_test_timeout, with_test_timeout,
) )
from .utils import random_bytes from mixnet.utils import random_bytes
class TestMixClient(IsolatedAsyncioTestCase): class TestMixClient(IsolatedAsyncioTestCase):

View File

@ -1,6 +1,6 @@
from unittest import TestCase from unittest import TestCase
from .fisheryates import FisherYates from mixnet.fisheryates import FisherYates
class TestFisherYates(TestCase): class TestFisherYates(TestCase):

View File

@ -1,8 +1,8 @@
import asyncio import asyncio
from unittest import IsolatedAsyncioTestCase from unittest import IsolatedAsyncioTestCase
from .mixnet import Mixnet from mixnet.mixnet import Mixnet
from .test_utils import init_mixnet_config from mixnet.test_utils import init_mixnet_config
class TestMixnet(IsolatedAsyncioTestCase): class TestMixnet(IsolatedAsyncioTestCase):

View File

@ -5,10 +5,10 @@ from unittest import IsolatedAsyncioTestCase
import numpy import numpy
from pysphinx.sphinx import SphinxPacket from pysphinx.sphinx import SphinxPacket
from .node import MixNode, NodeAddress, PacketQueue from mixnet.node import MixNode, NodeAddress, PacketQueue
from .packet import PacketBuilder from mixnet.packet import PacketBuilder
from .poisson import poisson_interval_sec, poisson_mean_interval_sec from mixnet.poisson import poisson_interval_sec, poisson_mean_interval_sec
from .test_utils import ( from mixnet.test_utils import (
init_mixnet_config, init_mixnet_config,
with_test_timeout, with_test_timeout,
) )

View File

@ -3,15 +3,15 @@ from unittest import TestCase
from pysphinx.sphinx import ProcessedFinalHopPacket, SphinxPacket from pysphinx.sphinx import ProcessedFinalHopPacket, SphinxPacket
from .config import MixNodeInfo from mixnet.config import MixNodeInfo
from .packet import ( from mixnet.packet import (
Fragment, Fragment,
MessageFlag, MessageFlag,
MessageReconstructor, MessageReconstructor,
PacketBuilder, PacketBuilder,
) )
from .test_utils import init_mixnet_config from mixnet.test_utils import init_mixnet_config
from .utils import random_bytes from mixnet.utils import random_bytes
class TestPacket(TestCase): class TestPacket(TestCase):

View File

@ -2,17 +2,17 @@ import asyncio
from cryptography.hazmat.primitives.asymmetric.x25519 import X25519PrivateKey from cryptography.hazmat.primitives.asymmetric.x25519 import X25519PrivateKey
from .bls import generate_bls from mixnet.bls import generate_bls
from .config import ( from mixnet.config import (
MixClientConfig, MixClientConfig,
MixNodeConfig,
MixnetConfig, MixnetConfig,
MixNodeInfo,
MixnetTopology, MixnetTopology,
MixnetTopologyConfig, MixnetTopologyConfig,
MixnetTopologySize, MixnetTopologySize,
MixNodeConfig,
MixNodeInfo,
) )
from .utils import random_bytes from mixnet.utils import random_bytes
def with_test_timeout(t): def with_test_timeout(t):