Chia-Network/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2024-11-28 11:38:38 +03:00
Code Issues Packages Projects Releases Wiki Activity

checkpoint: into main from release/2.1.3 @ a6730326e2 (#17085)

Browse Source
This commit is contained in:
Chris Marslender 2023-12-19 07:44:09 -06:00 committed by GitHub
commit 2a2d0a59ff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 26 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
chia/util/struct_stream.py
View File

@ -87,6 +87,5 @@ class StructStream(int):
def stream_to_bytes(self) -> bytes:
return super().to_bytes(length=self.SIZE, byteorder="big", signed=self.SIGNED)
# this is meant to avoid mixing up construcing a bytes object of a specific
# size (i.e. bytes(int)) vs. serializing the integer to bytes (i.e. bytes(uint32))
__bytes__ = None
def __bytes__(self) -> bytes:
return self.stream_to_bytes()

24
tests/util/test_struct_stream.py
View File

@ -1,5 +1,6 @@
from __future__ import annotations
import enum
import io
import struct
from dataclasses import dataclass
@ -27,6 +28,13 @@ def dataclass_parameters(instances: Iterable[object]) -> List[ParameterSet]:
return [dataclass_parameter(instance) for instance in instances]
class StreamAndBytesMatchMode(enum.Enum):
minimum = "minimum"
middle_low = "middle low"
middle_high = "middle high"
maximum = "maximum"
@dataclass(frozen=True)
class BadName:
name: str
@ -288,3 +296,19 @@ class TestStructStream:
def test_parse_metadata_from_name_correct_minimum(self, good: Good) -> None:
assert good.cls.MINIMUM == good.minimum
@pytest.mark.parametrize("mode", list(StreamAndBytesMatchMode), ids=lambda mode: mode.value)
def test_stream_to_bytes_and_bytes_match_minimum(self, good: Good, mode: StreamAndBytesMatchMode) -> None:
if mode == StreamAndBytesMatchMode.minimum:
value = good.minimum
elif mode == StreamAndBytesMatchMode.middle_low:
value = int(good.minimum + ((good.maximum - good.minimum) * 0.3))
elif mode == StreamAndBytesMatchMode.middle_high:
value = int(good.minimum + ((good.maximum - good.minimum) * 0.7))
elif mode == StreamAndBytesMatchMode.maximum:
value = good.maximum
else:
raise Exception(f"unhandled parametrization: {mode!r}")
instance = good.cls(value)
assert bytes(instance) == instance.stream_to_bytes()