cchecksum 0.4.0__cp312-cp312-musllinux_1_2_i686.whl

benchmarks/data.py

@@ -0,0 +1,70 @@
+from binascii import unhexlify
+from typing import Final
+
+BASE_ADDRESSES: Final = [
+    "0x52908400098527886e0f7030069857d2e4169ee7",
+    "0xde709f2102306220921060314715629080e2fb77",
+    "0x27b1fdb04752bbc536007a920d24acb045561c26",
+    "0x5aeda56215b167893e80b4fe645ba6d5bab767de",
+    "0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2",
+]
+
+MIXED_ADDRESSES: Final = [
+    "0x52908400098527886E0F7030069857D2E4169EE7",
+    "0x8617E340B3D01FA5F11F306F4090FD50E238070D",
+    "0x5aAeb6053F3E94C9b9A09f33669435E7Ef1BeAed",
+    "0xfb6916095ca1df60bb79ce92ce3ea74c37c5d359",
+    "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2",
+]
+
+STR_CASES: Final = []
+STR_CASE_IDS: Final = []
+for index, address in enumerate(BASE_ADDRESSES):
+    STR_CASES.append(address)
+    STR_CASE_IDS.append(f"lower-0x-{index}")
+
+    STR_CASES.append(address.upper())
+    STR_CASE_IDS.append(f"upper-0x-{index}")
+
+    STR_CASES.append(address[2:])
+    STR_CASE_IDS.append(f"lower-no-0x-{index}")
+
+    STR_CASES.append(address[2:].upper())
+    STR_CASE_IDS.append(f"upper-no-0x-{index}")
+
+for index, address in enumerate(MIXED_ADDRESSES):
+    STR_CASES.append(address)
+    STR_CASE_IDS.append(f"mixed-{index}")
+
+BYTES_CASES: Final = [unhexlify(address[2:]) for address in BASE_ADDRESSES]
+BYTES_CASE_IDS: Final = [f"bytes-{index}" for index in range(len(BASE_ADDRESSES))]
+
+BYTEARRAY_CASES: Final = [bytearray(address) for address in BYTES_CASES]
+BYTEARRAY_CASE_IDS: Final = [f"bytearray-{index}" for index in range(len(BYTES_CASES))]
+
+SINGLE_CASES: Final = STR_CASES + BYTES_CASES + BYTEARRAY_CASES
+SINGLE_CASE_IDS: Final = STR_CASE_IDS + BYTES_CASE_IDS + BYTEARRAY_CASE_IDS
+
+
+CONTAINER_SIZES: Final = 100, 1_000, 10_000
+
+
+def _build_container(cases: list, size: int) -> list:
+    return [cases[index % len(cases)] for index in range(size)]
+
+
+STR_CONTAINER_CASES: Final = [_build_container(STR_CASES, size) for size in CONTAINER_SIZES]
+STR_CONTAINER_CASE_IDS: Final = [f"str-container-{size}" for size in CONTAINER_SIZES]
+
+BYTES_CONTAINER_CASES: Final = [_build_container(BYTES_CASES, size) for size in CONTAINER_SIZES]
+BYTES_CONTAINER_CASE_IDS: Final = [f"bytes-container-{size}" for size in CONTAINER_SIZES]
+
+BYTEARRAY_CONTAINER_CASES: Final = [
+    _build_container(BYTEARRAY_CASES, size) for size in CONTAINER_SIZES
+]
+BYTEARRAY_CONTAINER_CASE_IDS: Final = [f"bytearray-container-{size}" for size in CONTAINER_SIZES]
+
+CONTAINER_CASES: Final = STR_CONTAINER_CASES + BYTES_CONTAINER_CASES + BYTEARRAY_CONTAINER_CASES
+CONTAINER_CASE_IDS: Final = (
+    STR_CONTAINER_CASE_IDS + BYTES_CONTAINER_CASE_IDS + BYTEARRAY_CONTAINER_CASE_IDS
+)

benchmarks/test_checksum_benchmarks.py

@@ -0,0 +1,23 @@
+import pytest
+from pytest_codspeed import BenchmarkFixture
+
+from benchmarks.data import CONTAINER_CASES, CONTAINER_CASE_IDS, SINGLE_CASES, SINGLE_CASE_IDS
+from cchecksum import to_checksum_address
+
+
+@pytest.mark.benchmark(group="to_checksum_address_str")
+@pytest.mark.parametrize("value", SINGLE_CASES, ids=SINGLE_CASE_IDS)
+def test_to_checksum_address(benchmark: BenchmarkFixture, value: str) -> None:
+    @benchmark
+    def run_10k():
+        for _ in range(10_000):
+            to_checksum_address(value)
+
+
+@pytest.mark.benchmark(group="to_checksum_address_bytes_container")
+@pytest.mark.parametrize("values", CONTAINER_CASES, ids=CONTAINER_CASE_IDS)
+def test_to_checksum_address_multi(benchmark: BenchmarkFixture, values: list[bytes]) -> None:
+    @benchmark
+    def run_container():
+        for value in values:
+            to_checksum_address(value)

cchecksum/__init__.py

@@ -0,0 +1,15 @@
+"""CChecksum is a ~8x faster drop-in replacement for eth_utils.to_checksum_address,
+with the most cpu-intensive part implemented in c.
+
+It keeps the exact same API as the existing implementation, exceptions and all.
+"""
+
+from cchecksum._checksum import to_checksum_address, to_checksum_address_many
+from cchecksum.monkey_patch import monkey_patch_eth_utils, monkey_patch_web3py
+
+__all__ = [
+    "to_checksum_address",
+    "to_checksum_address_many",
+    "monkey_patch_eth_utils",
+    "monkey_patch_web3py",
+]