Skip to content

Commit 2522f32

Browse files
Copilot0xrinegade
andcommitted
Fix import formatting issues - sort imports correctly with isort
Co-authored-by: 0xrinegade <[email protected]>
1 parent 3c18274 commit 2522f32

File tree

6 files changed

+42
-38
lines changed

6 files changed

+42
-38
lines changed

python/tests/unit/test_additional_coverage.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,11 @@
33
"""
44

55
import pytest
6-
from solana_ai_registries.types import ServiceEndpoint
6+
77
from solana_ai_registries.client import SolanaAIRegistriesClient
8-
from solana_ai_registries.idl import IDLLoader
98
from solana_ai_registries.exceptions import IDLError
9+
from solana_ai_registries.idl import IDLLoader
10+
from solana_ai_registries.types import ServiceEndpoint
1011

1112

1213
class TestServiceEndpointProtocolDetection:

python/tests/unit/test_coverage_boost.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22
Additional coverage tests for payments and MCP modules.
33
"""
44

5-
from solana_ai_registries.payments import PaymentManager
6-
from solana_ai_registries.mcp import McpServerRegistry
75
from solana_ai_registries.agent import AgentRegistry
6+
from solana_ai_registries.mcp import McpServerRegistry
7+
from solana_ai_registries.payments import PaymentManager
88

99

1010
class TestPaymentManagerBasics:
@@ -79,9 +79,9 @@ class TestConstantsCoverage:
7979
def test_import_constants(self):
8080
"""Test importing constants."""
8181
from solana_ai_registries.constants import (
82+
A2AMPL_BASE_UNIT,
8283
MAX_AGENT_ID_LEN,
8384
MAX_AGENT_NAME_LEN,
84-
A2AMPL_BASE_UNIT,
8585
)
8686

8787
assert MAX_AGENT_ID_LEN > 0
@@ -103,9 +103,9 @@ class TestExceptionsCoverage:
103103
def test_exception_creation(self):
104104
"""Test creating exceptions."""
105105
from solana_ai_registries.exceptions import (
106-
SolanaAIRegistriesError,
107106
AgentNotFoundError,
108107
McpServerNotFoundError,
108+
SolanaAIRegistriesError,
109109
)
110110

111111
base_error = SolanaAIRegistriesError("Base error")

python/tests/unit/test_final_coverage_push.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,10 @@
33
"""
44

55
import pytest
6+
67
from solana_ai_registries.client import SolanaAIRegistriesClient
7-
from solana_ai_registries.idl import IDLLoader
88
from solana_ai_registries.exceptions import IDLError
9+
from solana_ai_registries.idl import IDLLoader
910

1011

1112
class TestClientPropertiesAndMethods:
@@ -56,7 +57,7 @@ class TestModuleInitialization:
5657

5758
def test_import_all_modules(self):
5859
"""Test importing all main modules."""
59-
from solana_ai_registries import agent, client, mcp, payments, idl
60+
from solana_ai_registries import agent, client, idl, mcp, payments
6061

6162
assert agent is not None
6263
assert client is not None
@@ -124,8 +125,8 @@ class TestConstantsAndUtilities:
124125
def test_token_mint_constants(self):
125126
"""Test token mint constants."""
126127
from solana_ai_registries.constants import (
127-
A2AMPL_TOKEN_MINT_MAINNET,
128128
A2AMPL_TOKEN_MINT_DEVNET,
129+
A2AMPL_TOKEN_MINT_MAINNET,
129130
)
130131

131132
assert len(A2AMPL_TOKEN_MINT_MAINNET) > 0
@@ -146,9 +147,9 @@ def test_tier_constants(self):
146147
"""Test tier stake constants."""
147148
from solana_ai_registries.constants import (
148149
BRONZE_TIER_STAKE,
149-
SILVER_TIER_STAKE,
150150
GOLD_TIER_STAKE,
151151
PLATINUM_TIER_STAKE,
152+
SILVER_TIER_STAKE,
152153
)
153154

154155
assert BRONZE_TIER_STAKE < SILVER_TIER_STAKE
@@ -170,8 +171,8 @@ def test_specific_exception_types(self):
170171
"""Test specific exception types."""
171172
from solana_ai_registries.exceptions import (
172173
AgentNotFoundError,
173-
McpServerNotFoundError,
174174
ConnectionError,
175+
McpServerNotFoundError,
175176
)
176177

177178
agent_error = AgentNotFoundError("test_agent", "test_owner")
@@ -243,8 +244,8 @@ class TestMcpRegistryMethods:
243244

244245
def test_mcp_registry_initialization(self):
245246
"""Test McpServerRegistry initialization."""
246-
from solana_ai_registries.mcp import McpServerRegistry
247247
from solana_ai_registries.client import SolanaAIRegistriesClient
248+
from solana_ai_registries.mcp import McpServerRegistry
248249

249250
client = SolanaAIRegistriesClient()
250251
registry = McpServerRegistry(client)

python/tests/unit/test_final_push.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ def test_final_missing_lines(self):
1919

2020
def test_idl_loader_one_more_path(self):
2121
"""Test one more IDL loader path."""
22-
from solana_ai_registries.idl import IDLLoader
2322
from solana_ai_registries.exceptions import IDLError
23+
from solana_ai_registries.idl import IDLLoader
2424

2525
loader = IDLLoader()
2626

@@ -50,7 +50,7 @@ def test_client_with_different_commitment(self):
5050

5151
def test_exception_str_methods(self):
5252
"""Test exception __str__ methods."""
53-
from solana_ai_registries.exceptions import ValidationError, AgentExistsError
53+
from solana_ai_registries.exceptions import AgentExistsError, ValidationError
5454

5555
# Test ValidationError string representation
5656
val_error = ValidationError("test_field", "test_constraint", "Test message")
@@ -66,12 +66,12 @@ def test_exception_str_methods(self):
6666
def test_more_constants_usage(self):
6767
"""Test more constants usage."""
6868
from solana_ai_registries.constants import (
69+
GOLD_TIER_STAKE,
70+
MAX_AGENT_DESCRIPTION_LEN,
6971
MAX_AGENT_ID_LEN,
7072
MAX_AGENT_NAME_LEN,
71-
MAX_AGENT_DESCRIPTION_LEN,
72-
SILVER_TIER_STAKE,
73-
GOLD_TIER_STAKE,
7473
PLATINUM_TIER_STAKE,
74+
SILVER_TIER_STAKE,
7575
)
7676

7777
# Just test they exist and are reasonable values

python/tests/unit/test_minimal_coverage.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ class TestSimpleCoverage:
1010

1111
def test_import_and_use_types(self):
1212
"""Test importing and using types."""
13-
from solana_ai_registries.types import ServiceEndpoint, McpCapabilities
13+
from solana_ai_registries.types import McpCapabilities, ServiceEndpoint
1414

1515
# Simple ServiceEndpoint creation
1616
endpoint = ServiceEndpoint(url="https://example.com")
@@ -56,8 +56,8 @@ def test_payment_manager_creation(self):
5656

5757
def test_registries_creation(self):
5858
"""Test registry creation."""
59-
from solana_ai_registries.client import SolanaAIRegistriesClient
6059
from solana_ai_registries.agent import AgentRegistry
60+
from solana_ai_registries.client import SolanaAIRegistriesClient
6161
from solana_ai_registries.mcp import McpServerRegistry
6262

6363
client = SolanaAIRegistriesClient()
@@ -70,15 +70,16 @@ def test_registries_creation(self):
7070

7171
def test_constants_access(self):
7272
"""Test constants access."""
73+
from decimal import Decimal
74+
7375
from solana_ai_registries.constants import (
7476
A2AMPL_BASE_UNIT,
7577
AGENT_REGISTRATION_FEE,
76-
MCP_REGISTRATION_FEE,
7778
BRONZE_TIER_STAKE,
79+
MCP_REGISTRATION_FEE,
7880
a2ampl_to_base_units,
7981
base_units_to_a2ampl,
8082
)
81-
from decimal import Decimal
8283

8384
assert A2AMPL_BASE_UNIT == 1_000_000_000
8485
assert AGENT_REGISTRATION_FEE > 0
@@ -96,14 +97,14 @@ def test_constants_access(self):
9697
def test_exception_creation(self):
9798
"""Test exception creation."""
9899
from solana_ai_registries.exceptions import (
99-
SolanaAIRegistriesError,
100100
AgentNotFoundError,
101-
McpServerNotFoundError,
102-
IDLError,
103101
ConnectionError,
104-
TransactionError,
102+
IDLError,
103+
McpServerNotFoundError,
105104
PaymentError,
106105
RegistrationError,
106+
SolanaAIRegistriesError,
107+
TransactionError,
107108
)
108109

109110
base_err = SolanaAIRegistriesError("base")

python/tests/unit/test_simple_coverage_push.py

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@ def test_import_main_modules(self):
1212
"""Test importing main modules."""
1313
import solana_ai_registries.agent
1414
import solana_ai_registries.client
15+
import solana_ai_registries.idl
1516
import solana_ai_registries.mcp
1617
import solana_ai_registries.payments
17-
import solana_ai_registries.idl
1818

1919
assert solana_ai_registries.agent is not None
2020
assert solana_ai_registries.client is not None
@@ -24,9 +24,9 @@ def test_import_main_modules(self):
2424

2525
def test_basic_class_instantiation(self):
2626
"""Test basic class instantiation."""
27+
from solana_ai_registries.agent import AgentRegistry
2728
from solana_ai_registries.client import SolanaAIRegistriesClient
2829
from solana_ai_registries.idl import IDLLoader
29-
from solana_ai_registries.agent import AgentRegistry
3030
from solana_ai_registries.mcp import McpServerRegistry
3131
from solana_ai_registries.payments import PaymentManager
3232

@@ -59,8 +59,8 @@ def test_client_init_variations(self):
5959

6060
def test_idl_loader_methods(self):
6161
"""Test IDL loader methods."""
62-
from solana_ai_registries.idl import IDLLoader
6362
from solana_ai_registries.exceptions import IDLError
63+
from solana_ai_registries.idl import IDLLoader
6464

6565
loader = IDLLoader()
6666

@@ -113,9 +113,9 @@ def test_constants_validation(self):
113113
def test_type_creations(self):
114114
"""Test creating type instances."""
115115
from solana_ai_registries.types import (
116-
ServiceEndpoint,
117116
AgentSkill,
118117
McpCapabilities,
118+
ServiceEndpoint,
119119
)
120120

121121
# ServiceEndpoint variations
@@ -145,13 +145,13 @@ def test_type_creations(self):
145145
def test_exception_instantiation(self):
146146
"""Test exception instantiation."""
147147
from solana_ai_registries.exceptions import (
148-
SolanaAIRegistriesError,
149148
AgentNotFoundError,
149+
ConnectionError,
150+
IDLError,
150151
McpServerNotFoundError,
151-
RegistrationError,
152152
PaymentError,
153-
IDLError,
154-
ConnectionError,
153+
RegistrationError,
154+
SolanaAIRegistriesError,
155155
TransactionError,
156156
ValidationError,
157157
)
@@ -179,14 +179,14 @@ def test_exception_instantiation(self):
179179
def test_more_constants(self):
180180
"""Test more constants."""
181181
from solana_ai_registries.constants import (
182-
AGENT_REGISTRY_PROGRAM_ID,
183-
MCP_SERVER_REGISTRY_PROGRAM_ID,
184-
A2AMPL_DECIMALS,
185182
A2AMPL_BASE_UNIT,
183+
A2AMPL_DECIMALS,
184+
AGENT_REGISTRY_PROGRAM_ID,
186185
BRONZE_TIER_STAKE,
187-
SILVER_TIER_STAKE,
188186
GOLD_TIER_STAKE,
187+
MCP_SERVER_REGISTRY_PROGRAM_ID,
189188
PLATINUM_TIER_STAKE,
189+
SILVER_TIER_STAKE,
190190
)
191191

192192
assert len(AGENT_REGISTRY_PROGRAM_ID) > 0
@@ -199,11 +199,12 @@ def test_more_constants(self):
199199

200200
def test_conversion_functions(self):
201201
"""Test conversion utility functions."""
202+
from decimal import Decimal
203+
202204
from solana_ai_registries.constants import (
203205
a2ampl_to_base_units,
204206
base_units_to_a2ampl,
205207
)
206-
from decimal import Decimal
207208

208209
# Test conversions
209210
base_units = a2ampl_to_base_units(Decimal("1.5"))

0 commit comments

Comments
 (0)