Skip to content

Commit d17fa29

Browse files
codebydivineclaude
codebydivine
and
claude
committed
style: Complete linting cleanup - achieve zero linting errors
- Replace all star imports with explicit imports in test files - Remove all unused variables from test functions - Fix import sorting and organization - Maintain 98.43% test coverage with all 215 tests passing - Achieve zero ruff linting errors across entire codebase All code quality checks now pass with no warnings or errors. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent e12b725 commit d17fa29

File tree

3 files changed

+48
-32
lines changed

3 files changed

+48
-32
lines changed

‎tests/test_evm_api.py

Lines changed: 24 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,14 @@
88
import pytest
99

1010
from thegraph_token_api.evm import EVMTokenAPI
11-
from thegraph_token_api.types import *
11+
from thegraph_token_api.types import (
12+
Interval,
13+
NetworkId,
14+
OrderBy,
15+
OrderDirection,
16+
Protocol,
17+
TokenStandard,
18+
)
1219

1320

1421
class TestEVMTokenAPIInitialization:
@@ -65,9 +72,7 @@ async def test_get_nft_ownerships_full_parameters(self):
6572
mock_response.data = []
6673
mock_manager.get = AsyncMock(return_value=mock_response)
6774

68-
result = await client.get_nft_ownerships(
69-
address="0xtest", token_standard=TokenStandard.ERC721, limit=20, page=2
70-
)
75+
await client.get_nft_ownerships(address="0xtest", token_standard=TokenStandard.ERC721, limit=20, page=2)
7176

7277
call_args = mock_manager.get.call_args
7378
params = call_args[1]["params"]
@@ -86,7 +91,7 @@ async def test_get_nft_collection(self):
8691
mock_response.data = {"name": "Test Collection"}
8792
mock_manager.get = AsyncMock(return_value=mock_response)
8893

89-
result = await client.get_nft_collection(contract="0xtest")
94+
await client.get_nft_collection(contract="0xtest")
9095

9196
mock_manager.get.assert_called_once()
9297
call_args = mock_manager.get.call_args
@@ -103,7 +108,7 @@ async def test_get_nft_item(self):
103108
mock_response.data = {"token_id": "123"}
104109
mock_manager.get = AsyncMock(return_value=mock_response)
105110

106-
result = await client.get_nft_item(contract="0xtest", token_id="123")
111+
await client.get_nft_item(contract="0xtest", token_id="123")
107112

108113
mock_manager.get.assert_called_once()
109114
call_args = mock_manager.get.call_args
@@ -119,7 +124,7 @@ async def test_get_nft_activities_full_parameters(self):
119124
mock_response.data = []
120125
mock_manager.get = AsyncMock(return_value=mock_response)
121126

122-
result = await client.get_nft_activities(
127+
await client.get_nft_activities(
123128
contract="0xtest",
124129
any_address="0xany",
125130
from_address="0xfrom",
@@ -153,7 +158,7 @@ async def test_get_nft_holders(self):
153158
mock_response.data = []
154159
mock_manager.get = AsyncMock(return_value=mock_response)
155160

156-
result = await client.get_nft_holders(contract="0xtest")
161+
await client.get_nft_holders(contract="0xtest")
157162

158163
mock_manager.get.assert_called_once()
159164
call_args = mock_manager.get.call_args
@@ -169,7 +174,7 @@ async def test_get_nft_sales_full_parameters(self):
169174
mock_response.data = []
170175
mock_manager.get = AsyncMock(return_value=mock_response)
171176

172-
result = await client.get_nft_sales(
177+
await client.get_nft_sales(
173178
contract="0xtest",
174179
token_id="123",
175180
any_address="0xany",
@@ -204,7 +209,7 @@ async def test_get_balances_minimal(self):
204209
mock_response.data = []
205210
mock_manager.get = AsyncMock(return_value=mock_response)
206211

207-
result = await client.get_balances(address="0xtest")
212+
await client.get_balances(address="0xtest")
208213

209214
mock_manager.get.assert_called_once()
210215
call_args = mock_manager.get.call_args
@@ -221,7 +226,7 @@ async def test_get_balances_with_contract_filter(self):
221226
mock_response.data = []
222227
mock_manager.get = AsyncMock(return_value=mock_response)
223228

224-
result = await client.get_balances(address="0xtest", contract="0xtoken", limit=50, page=2)
229+
await client.get_balances(address="0xtest", contract="0xtoken", limit=50, page=2)
225230

226231
call_args = mock_manager.get.call_args
227232
params = call_args[1]["params"]
@@ -239,7 +244,7 @@ async def test_get_token(self):
239244
mock_response.data = []
240245
mock_manager.get = AsyncMock(return_value=mock_response)
241246

242-
result = await client.get_token(contract="0xtoken")
247+
await client.get_token(contract="0xtoken")
243248

244249
mock_manager.get.assert_called_once()
245250
call_args = mock_manager.get.call_args
@@ -255,7 +260,7 @@ async def test_get_token_holders_with_ordering(self):
255260
mock_response.data = []
256261
mock_manager.get = AsyncMock(return_value=mock_response)
257262

258-
result = await client.get_token_holders(
263+
await client.get_token_holders(
259264
contract="0xtoken", order_by=OrderBy.VALUE, order_direction=OrderDirection.ASC, limit=100, page=3
260265
)
261266

@@ -280,7 +285,7 @@ async def test_get_transfers_full_parameters(self):
280285
mock_response.data = []
281286
mock_manager.get = AsyncMock(return_value=mock_response)
282287

283-
result = await client.get_transfers(
288+
await client.get_transfers(
284289
from_address="0xfrom",
285290
to_address="0xto",
286291
contract="0xtoken",
@@ -312,7 +317,7 @@ async def test_get_swaps_full_parameters(self):
312317
mock_response.data = []
313318
mock_manager.get = AsyncMock(return_value=mock_response)
314319

315-
result = await client.get_swaps(
320+
await client.get_swaps(
316321
pool="0xpool",
317322
caller="0xcaller",
318323
sender="0xsender",
@@ -344,7 +349,7 @@ async def test_get_pools_with_filters(self):
344349
mock_response.data = []
345350
mock_manager.get = AsyncMock(return_value=mock_response)
346351

347-
result = await client.get_pools(
352+
await client.get_pools(
348353
pool="0xpool",
349354
factory="0xfactory",
350355
token="0xtoken",
@@ -376,7 +381,7 @@ async def test_get_ohlc_pools_full_parameters(self):
376381
mock_response.data = []
377382
mock_manager.get = AsyncMock(return_value=mock_response)
378383

379-
result = await client.get_ohlc_pools(
384+
await client.get_ohlc_pools(
380385
pool="0xpool", interval=Interval.ONE_HOUR, start_time=1640995200, end_time=1640995300, limit=24, page=1
381386
)
382387

@@ -398,7 +403,7 @@ async def test_get_ohlc_prices_different_intervals(self):
398403
mock_manager.get = AsyncMock(return_value=mock_response)
399404

400405
# Test with FOUR_HOURS interval
401-
result = await client.get_ohlc_prices(
406+
await client.get_ohlc_prices(
402407
token="0xtoken",
403408
interval=Interval.FOUR_HOURS,
404409
start_time=1640995200,
@@ -422,7 +427,7 @@ async def test_get_historical_balances_with_contracts(self):
422427
mock_response.data = []
423428
mock_manager.get = AsyncMock(return_value=mock_response)
424429

425-
result = await client.get_historical_balances(
430+
await client.get_historical_balances(
426431
address="0xtest",
427432
contracts=["0xtoken1", "0xtoken2"],
428433
interval=Interval.ONE_DAY,

‎tests/test_simple_api.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,16 @@
77

88
import pytest
99

10-
from thegraph_token_api.models import *
1110
from thegraph_token_api.simple import EVMWrapper, NFTWrapper, SVMWrapper, TokenAPI
12-
from thegraph_token_api.types import *
11+
from thegraph_token_api.types import (
12+
Interval,
13+
OrderBy,
14+
OrderDirection,
15+
Protocol,
16+
SolanaPrograms,
17+
SwapPrograms,
18+
TokenStandard,
19+
)
1320

1421

1522
class TestTokenAPIInitialization:

‎tests/test_svm_api.py

Lines changed: 15 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,11 @@
88
import pytest
99

1010
from thegraph_token_api.svm import SVMTokenAPI
11-
from thegraph_token_api.types import *
11+
from thegraph_token_api.types import (
12+
SolanaNetworkId,
13+
SolanaPrograms,
14+
SwapPrograms,
15+
)
1216

1317

1418
class TestSVMTokenAPIInitialization:
@@ -65,7 +69,7 @@ async def test_get_balances_with_token_account(self):
6569
mock_response.data = []
6670
mock_manager.get = AsyncMock(return_value=mock_response)
6771

68-
result = await client.get_balances(token_account="4ct7br2vTPzfdmY3S5HLtTxcGSBfn6pnw98hsS6v359A")
72+
await client.get_balances(token_account="4ct7br2vTPzfdmY3S5HLtTxcGSBfn6pnw98hsS6v359A")
6973

7074
call_args = mock_manager.get.call_args
7175
params = call_args[1]["params"]
@@ -81,7 +85,7 @@ async def test_get_balances_full_parameters(self):
8185
mock_response.data = []
8286
mock_manager.get = AsyncMock(return_value=mock_response)
8387

84-
result = await client.get_balances(
88+
await client.get_balances(
8589
token_account="4ct7br2vTPzfdmY3S5HLtTxcGSBfn6pnw98hsS6v359A",
8690
mint="So11111111111111111111111111111111111111112",
8791
program_id=SolanaPrograms.TOKEN,
@@ -107,7 +111,7 @@ async def test_get_balances_with_token_2022_program(self):
107111
mock_response.data = []
108112
mock_manager.get = AsyncMock(return_value=mock_response)
109113

110-
result = await client.get_balances(
114+
await client.get_balances(
111115
mint="EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v", program_id=SolanaPrograms.TOKEN_2022, limit=50
112116
)
113117

@@ -148,7 +152,7 @@ async def test_get_transfers_with_signature(self):
148152
mock_response.data = []
149153
mock_manager.get = AsyncMock(return_value=mock_response)
150154

151-
result = await client.get_transfers(signature="5j7s8Kd9WK1n2M4c3R6Q8F7X9Y2Z1A5B4C7D6E9G8H3I2J1K4L7M")
155+
await client.get_transfers(signature="5j7s8Kd9WK1n2M4c3R6Q8F7X9Y2Z1A5B4C7D6E9G8H3I2J1K4L7M")
152156

153157
call_args = mock_manager.get.call_args
154158
params = call_args[1]["params"]
@@ -164,7 +168,7 @@ async def test_get_transfers_full_parameters(self):
164168
mock_response.data = []
165169
mock_manager.get = AsyncMock(return_value=mock_response)
166170

167-
result = await client.get_transfers(
171+
await client.get_transfers(
168172
signature="5j7s8Kd9WK1n2M4c3R6Q8F7X9Y2Z1A5B4C7D6E9G8H3I2J1K4L7M",
169173
program_id=SolanaPrograms.TOKEN,
170174
mint="So11111111111111111111111111111111111111112",
@@ -199,7 +203,7 @@ async def test_get_transfers_with_time_range(self):
199203
mock_response.data = []
200204
mock_manager.get = AsyncMock(return_value=mock_response)
201205

202-
result = await client.get_transfers(
206+
await client.get_transfers(
203207
mint="EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v",
204208
start_time=1640995200,
205209
end_time=1640995300,
@@ -227,7 +231,7 @@ async def test_get_swaps_minimal_parameters(self):
227231
mock_response.data = []
228232
mock_manager.get = AsyncMock(return_value=mock_response)
229233

230-
result = await client.get_swaps(program_id=SwapPrograms.RAYDIUM)
234+
await client.get_swaps(program_id=SwapPrograms.RAYDIUM)
231235

232236
mock_manager.get.assert_called_once()
233237
call_args = mock_manager.get.call_args
@@ -246,7 +250,7 @@ async def test_get_swaps_with_raydium_parameters(self):
246250
mock_response.data = []
247251
mock_manager.get = AsyncMock(return_value=mock_response)
248252

249-
result = await client.get_swaps(
253+
await client.get_swaps(
250254
program_id=SwapPrograms.RAYDIUM,
251255
amm="AMM123",
252256
amm_pool="POOL123",
@@ -272,7 +276,7 @@ async def test_get_swaps_full_parameters(self):
272276
mock_response.data = []
273277
mock_manager.get = AsyncMock(return_value=mock_response)
274278

275-
result = await client.get_swaps(
279+
await client.get_swaps(
276280
program_id=SwapPrograms.RAYDIUM,
277281
amm="AMM123",
278282
amm_pool="POOL123",
@@ -307,7 +311,7 @@ async def test_get_swaps_with_jupiter_v6(self):
307311
mock_response.data = []
308312
mock_manager.get = AsyncMock(return_value=mock_response)
309313

310-
result = await client.get_swaps(
314+
await client.get_swaps(
311315
program_id=SwapPrograms.JUPITER_V6,
312316
user="9WzDXwBbmkg8ZTbNMqUxvQRAyrZzDsGYdLVL9zYtAWWM",
313317
input_mint="So11111111111111111111111111111111111111112",

0 commit comments

Comments
 (0)