diff --git a/cryptarchia/cryptarchia.py b/cryptarchia/cryptarchia.py index 1981e69..c59b6d0 100644 --- a/cryptarchia/cryptarchia.py +++ b/cryptarchia/cryptarchia.py @@ -1,4 +1,4 @@ -from typing import TypeAlias, List, Optional, Dict +from typing import TypeAlias, List, Dict from hashlib import sha256, blake2b from math import floor from copy import deepcopy diff --git a/cryptarchia/test_common.py b/cryptarchia/test_common.py index b032b6f..78cbd20 100644 --- a/cryptarchia/test_common.py +++ b/cryptarchia/test_common.py @@ -1,7 +1,5 @@ from .cryptarchia import ( Config, - TimeConfig, - Id, Slot, Coin, BlockHeader, diff --git a/cryptarchia/test_fork_choice.py b/cryptarchia/test_fork_choice.py index 5eaae15..2acaae7 100644 --- a/cryptarchia/test_fork_choice.py +++ b/cryptarchia/test_fork_choice.py @@ -1,7 +1,4 @@ from unittest import TestCase -from itertools import repeat -import numpy as np -import hashlib from copy import deepcopy from cryptarchia.cryptarchia import ( @@ -9,8 +6,6 @@ from cryptarchia.cryptarchia import ( ghost_maxvalid_bg, BlockHeader, Slot, - Id, - MockLeaderProof, Coin, Follower, common_prefix_depth, diff --git a/cryptarchia/test_leader.py b/cryptarchia/test_leader.py index 136f964..08d662a 100644 --- a/cryptarchia/test_leader.py +++ b/cryptarchia/test_leader.py @@ -2,16 +2,7 @@ from unittest import TestCase import numpy as np -from .cryptarchia import ( - Leader, - Config, - EpochState, - LedgerState, - Coin, - phi, - TimeConfig, - Slot, -) +from .cryptarchia import Leader, EpochState, LedgerState, Coin, phi, Slot from .test_common import mk_config diff --git a/cryptarchia/test_ledger_state_update.py b/cryptarchia/test_ledger_state_update.py index d71c7aa..92e2e83 100644 --- a/cryptarchia/test_ledger_state_update.py +++ b/cryptarchia/test_ledger_state_update.py @@ -2,17 +2,7 @@ from unittest import TestCase import numpy as np -from .cryptarchia import ( - Follower, - TimeConfig, - BlockHeader, - Config, - Coin, - LedgerState, - MockLeaderProof, - Slot, - Id, -) +from .cryptarchia import Follower, Coin from .test_common import mk_config, mk_block, mk_genesis_state diff --git a/cryptarchia/test_orphaned_proofs.py b/cryptarchia/test_orphaned_proofs.py index 5cce7ff..0114464 100644 --- a/cryptarchia/test_orphaned_proofs.py +++ b/cryptarchia/test_orphaned_proofs.py @@ -1,19 +1,8 @@ from unittest import TestCase -from itertools import repeat -import numpy as np -import hashlib -from copy import deepcopy -from cryptarchia.cryptarchia import ( - BlockHeader, - Slot, - Id, - MockLeaderProof, - Coin, - Follower, -) +from cryptarchia.cryptarchia import Coin, Follower -from .test_common import mk_chain, mk_config, mk_genesis_state, mk_block +from .test_common import mk_config, mk_genesis_state, mk_block class TestOrphanedProofs(TestCase): diff --git a/cryptarchia/test_stake_relativization.py b/cryptarchia/test_stake_relativization.py index 91a0e3a..cfa9473 100644 --- a/cryptarchia/test_stake_relativization.py +++ b/cryptarchia/test_stake_relativization.py @@ -1,5 +1,4 @@ from unittest import TestCase -from dataclasses import dataclass import itertools import numpy as np