1+ # type: ignore
2+
13from enum import IntEnum
24from inspect import isclass , signature
35from typing import Callable , List
810from web3 .exceptions import ContractLogicError , ContractPanicError
911from web3 .contract .contract import ContractFunction
1012
11- from autonity import factories , networks
13+ import autonity
1214from autonity .constants import AUTONITY_CONTRACT_ADDRESS
1315from autonity .contracts import ierc20
1416
1517
16- BINDINGS = [
17- factories .Accountability ,
18- factories .ACU ,
19- factories .Autonity ,
20- factories .InflationController ,
21- factories .Liquid ,
22- factories .NonStakableVesting ,
23- factories .Oracle ,
24- factories .Stabilization ,
25- factories .SupplyControl ,
26- factories .UpgradeManager ,
27- ierc20 .IERC20 ,
18+ FACTORIES = [attr for attr in autonity .__dict__ .values () if isinstance (attr , Callable )]
19+ BINDINGS = FACTORIES + [
20+ ierc20 .IERC20 , # IERC20 is used internally by aut-cli, not part of the public API
2821]
2922
3023TEST_INPUTS = {
@@ -47,16 +40,16 @@ def pytest_generate_tests(metafunc):
4740 ids = []
4841
4942 for binding in BINDINGS :
50- w3 = Web3 (networks .piccadilly .http_provider )
43+ w3 = Web3 (autonity . networks .piccadilly .http_provider )
5144
5245 if binding .__name__ == "Liquid" :
53- autonity = factories .Autonity (w3 )
54- validator = autonity .get_validator (autonity .get_validators ()[0 ])
55- contract = binding (w3 , validator .liquid_contract ) # type: ignore
46+ aut = autonity .Autonity (w3 )
47+ validator = aut .get_validator (aut .get_validators ()[0 ])
48+ contract = binding (w3 , validator .liquid_contract )
5649 elif binding .__name__ == "IERC20" :
57- contract = binding (w3 , AUTONITY_CONTRACT_ADDRESS ) # type: ignore
50+ contract = binding (w3 , AUTONITY_CONTRACT_ADDRESS )
5851 else :
59- contract = binding (w3 ) # type: ignore
52+ contract = binding (w3 )
6053
6154 for attr_name in dir (contract ):
6255 if attr_name .startswith ("_" ):
0 commit comments