import pytest

from pbcabi.data import BlockchainAddress

EXAMPLE_SHARDS = [
    ('Shard1', 2, '000000000000000000000000000000000000000001'),
    ('Shard0', 2, '000000000000000000000000000000000000000002'),
    ('Shard0', 3, '025FA781D389D7C7CAAF836E5E47ABED6CEFD2D928'),
    ('Shard1', 3, '04FE17D1009372C8ED3AC5B790B32E349359C2C7E9'),
    ('Shard0', 3, '01A2020BB33EF9E0323C7A3210D5CB7FD492AA0D65'),
]


@pytest.mark.parametrize('shard_id,num_shards,address', EXAMPLE_SHARDS)
def test_parse_abi(shard_id: int, num_shards: int, address: str):
    address = BlockchainAddress.from_hex_hash(address)
    assert address.shard_id(num_shards) == shard_id