Skip to content

Commit 83dcece

Browse files
committed
style: apply black formatting
1 parent a98d7c1 commit 83dcece

3 files changed

Lines changed: 16 additions & 19 deletions

File tree

diffyscan/diffyscan.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,9 @@ def run_bytecode_diff(
174174
deployment_call_data = _append_calldata(contract_creation_code, calldata)
175175
gas_limit = config.get("deployment_gas_limit")
176176
extra = {"gas_limit": gas_limit} if gas_limit else {}
177-
local_deployed_bytecode = simulate_deployment(deployment_call_data, remote_rpc_url, **extra)
177+
local_deployed_bytecode = simulate_deployment(
178+
deployment_call_data, remote_rpc_url, **extra
179+
)
178180

179181
is_fully_matched = local_deployed_bytecode == remote_deployed_bytecode
180182

@@ -358,7 +360,8 @@ def _load_explorer_token(config: dict) -> str:
358360
def _setup_binary_comparison(config: dict, use_local_rpc: bool = False) -> str:
359361
"""Load RPC URL and configure exception handling for bytecode comparison."""
360362
rpc_env_var = (
361-
"LOCAL_RPC_URL" if use_local_rpc
363+
"LOCAL_RPC_URL"
364+
if use_local_rpc
362365
else config.get("rpc_url_env_var", "REMOTE_RPC_URL")
363366
)
364367
remote_rpc_url = load_env(rpc_env_var, masked=True, required=True)
@@ -458,9 +461,7 @@ def process_config(
458461

459462
# Apply contract filter if specified
460463
filter_set = (
461-
set(addr.lower() for addr in contract_filter)
462-
if contract_filter
463-
else None
464+
set(addr.lower() for addr in contract_filter) if contract_filter else None
464465
)
465466

466467
for contract_address, contract_name in config["contracts"].items():

diffyscan/utils/node_handler.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,12 @@ def get_chain_id(rpc_url: str) -> int:
4444
return chain_id
4545

4646

47-
def simulate_deployment(data: str, rpc_url: str, caller: str = DEFAULT_CALLER, gas_limit: int = DEFAULT_DEPLOYMENT_GAS_LIMIT) -> str:
47+
def simulate_deployment(
48+
data: str,
49+
rpc_url: str,
50+
caller: str = DEFAULT_CALLER,
51+
gas_limit: int = DEFAULT_DEPLOYMENT_GAS_LIMIT,
52+
) -> str:
4853
"""Simulate contract deployment via eth_call and return deployed runtime bytecode."""
4954
logger.info(f'Simulating deployment via eth_call on "{mask_text(rpc_url)}" ...')
5055

tests/test_contract_filter.py

Lines changed: 4 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
import pytest
44

5-
65
CONTRACTS = {
76
"0xAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAa": "ContractA",
87
"0xBbBbBbBbBbBbBbBbBbBbBbBbBbBbBbBbBbBbBbBb": "ContractB",
@@ -29,12 +28,8 @@ def test_no_filter_returns_all():
2928

3029

3130
def test_filter_single_contract():
32-
filtered = apply_filter(
33-
CONTRACTS, ["0xAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAa"]
34-
)
35-
assert list(filtered.keys()) == [
36-
"0xAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAa"
37-
]
31+
filtered = apply_filter(CONTRACTS, ["0xAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAa"])
32+
assert list(filtered.keys()) == ["0xAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAa"]
3833
assert filtered["0xAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAaAa"] == "ContractA"
3934

4035

@@ -51,17 +46,13 @@ def test_filter_multiple_contracts():
5146

5247

5348
def test_filter_is_case_insensitive():
54-
filtered = apply_filter(
55-
CONTRACTS, ["0xaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"]
56-
)
49+
filtered = apply_filter(CONTRACTS, ["0xaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"])
5750
assert len(filtered) == 1
5851
assert "ContractA" in filtered.values()
5952

6053

6154
def test_filter_nonexistent_address_returns_empty():
62-
filtered = apply_filter(
63-
CONTRACTS, ["0x0000000000000000000000000000000000000000"]
64-
)
55+
filtered = apply_filter(CONTRACTS, ["0x0000000000000000000000000000000000000000"])
6556
assert filtered == {}
6657

6758

0 commit comments

Comments
 (0)