ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 01:18:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Make Hdf5Sample work with bytearray

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-07-28 09:06:15 -05:00
parent a69cbed7b7
commit 7d5ec1344a
2 changed files with 23 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/features/test_sample.py
View File

@@ -41,6 +41,11 @@ def _test_sample(sample: Sample) -> None:
# Bytes
_assert_roundtrip_bytes(sample, b"\x00\x01\x02\x03\x04\x05")
_assert_roundtrip_bytes(
sample,
bytearray(b"\x00\x01\x02\x03\x04\x05"),
check_type=False,
)
# Querying unknown keys should return None
assert sample.get_scalar("unknown-key") is None
@@ -53,12 +58,15 @@ def _test_sample(sample: Sample) -> None:
sample.put_vector("key", None)
def _assert_roundtrip_bytes(sample: Sample, expected: Any) -> None:
def _assert_roundtrip_bytes(
sample: Sample, expected: Any, check_type: bool = False
) -> None:
sample.put_bytes("key", expected)
actual = sample.get_bytes("key")
assert actual == expected
assert actual is not None
_assert_same_type(actual, expected)
if check_type:
_assert_same_type(actual, expected)
def _assert_roundtrip_scalar(sample: Sample, expected: Any) -> None: