chia 0.43.0

A meta-crate that exports all of the Chia crates in the workspace.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161

from typing import Optional, Any, Callable
from pytest import raises
import sys
import time
from chia_rs import BlockRecord, ClassgroupElement
from chia_rs.sized_bytes import bytes32, bytes100
from chia_rs.sized_ints import uint32, uint64, uint8, uint128
from random import Random
from run_gen import DEFAULT_CONSTANTS


def get_classgroup_element(rng: Random) -> ClassgroupElement:
    return ClassgroupElement(bytes100.random(rng))


def get_u4(rng: Random) -> uint8:
    return uint8(rng.randint(0, 0xF))


def get_u8(rng: Random) -> uint8:
    return uint8(rng.randint(0, 0xFF))


def get_u32(rng: Random) -> uint32:
    return uint32(rng.randint(0, 0xFFFFFFFF))


def get_ssi(rng: Random) -> uint64:
    return uint64(
        DEFAULT_CONSTANTS.NUM_SPS_SUB_SLOT * rng.randint(0, 0xFFFF) + rng.randint(0, 1)
    )


def get_u64(rng: Random) -> uint64:
    return uint64(rng.randint(0, 0xFFFFFFFFFFFFFFFF))


def get_u128(rng: Random) -> uint128:
    return uint128(rng.randint(0, 0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF))


def get_optional(rng: Random, gen: Callable[[Random], Any]) -> Optional[Any]:
    if rng.randint(0, 1) == 0:
        return None
    else:
        return gen(rng)


def get_list(rng: Random, gen: Callable[[Random], Any]) -> list[Any]:
    length = rng.sample([0, 1, 5, 32, 500], 1)[0]
    ret: list[Any] = []
    for i in range(length):
        ret.append(gen(rng))
    return ret


def get_bool(rng: Random) -> bool:
    return rng.randint(0, 1) == 1


def get_hash(rng: Random) -> bytes32:
    return bytes32.random(rng)


def get_block_record(
    rng: Random,
    ssi: Optional[uint64] = None,
    ri: Optional[uint64] = None,
    spi: Optional[uint8] = None,
) -> BlockRecord:
    height = get_u32(rng)
    weight = get_u128(rng)
    iters = get_u128(rng)
    sp_index = spi if spi is not None else get_u4(rng)
    vdf_out = get_classgroup_element(rng)
    infused_challenge = get_optional(rng, get_classgroup_element)
    sub_slot_iters = ssi if ssi is not None else get_ssi(rng)
    required_iters = ri if ri is not None else get_u64(rng)
    deficit = get_u8(rng)
    overflow = get_bool(rng)
    prev_tx_height = get_u32(rng)
    timestamp = uint64(123456789)
    prev_tx_hash = get_optional(rng, get_hash)
    fees = get_optional(rng, get_u64)

    return BlockRecord(
        bytes32.random(rng),
        bytes32.random(rng),
        height,
        weight,
        iters,
        sp_index,
        vdf_out,
        infused_challenge,
        bytes32.random(rng),
        bytes32.random(rng),
        sub_slot_iters,
        bytes32.random(rng),
        bytes32.random(rng),
        required_iters,
        deficit,
        overflow,
        prev_tx_height,
        timestamp,
        prev_tx_hash,
        fees,
        [],
        None,
        None,
        None,
        None,
    )


def test_bytes32() -> None:
    rng = Random()
    rng.seed(1337)
    br = get_block_record(rng)
    # the following line is commented until chia-blockchain uses the moved sized bytes class
    # assert isinstance(br.header_hash, bytes32)
    assert (
        f"{br.header_hash}"
        == "e433713dd932b2314eab219aa5504f71b9fe9f2d8e2f5cadfa892d8dc6a7ba53"
    )
    assert (
        br.header_hash.__str__()
        == "e433713dd932b2314eab219aa5504f71b9fe9f2d8e2f5cadfa892d8dc6a7ba53"
    )


def wrap_call(expr: str, br: Any) -> str:
    try:
        ret = eval(expr, None, {"br": br})
        return f"V:{ret}"
    except Exception as e:
        return f"E:{e}"


# TODO: more thoroughly check these new functions which use self
def test_calculate_sp_iters() -> None:
    ssi: uint64 = uint64(100001 * 64 * 4)
    rng = Random()
    rng.seed(1337)
    br = get_block_record(rng, ssi=ssi, spi=uint8(31))
    res = br.sp_iters(DEFAULT_CONSTANTS)
    assert res is not None


def test_calculate_ip_iters() -> None:
    ssi: uint64 = uint64(100001 * 64 * 4)
    sp_interval_iters = ssi // 32
    ri = uint64(sp_interval_iters - 1)
    rng = Random()
    rng.seed(1337)
    br = get_block_record(rng, ssi=ssi, spi=uint8(31), ri=ri)
    with raises(ValueError):
        res = br.ip_iters(DEFAULT_CONSTANTS)

    br = get_block_record(rng, ssi=ssi, spi=uint8(13), ri=uint64(1))
    res = br.ip_iters(DEFAULT_CONSTANTS)
    assert res == 6400065