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

Implement sample.{get,put}_bytes

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-07-27 10:01:32 -05:00
parent 962707e8b7
commit 284ba15db6
2 changed files with 39 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/features/test_sample.py
View File

@@ -35,6 +35,17 @@ def _test_sample(sample: Sample) -> None:
_assert_roundtrip_vector_list(sample, [[1], None, [2, 2], [3, 3, 3]])
_assert_roundtrip_vector_list(sample, [[1.0], None, [2.0, 2.0], [3.0, 3.0, 3.0]])
# Bytes
_assert_roundtrip_bytes(sample, b"\x00\x01\x02\x03\x04\x05")
def _assert_roundtrip_bytes(sample: Sample, expected: Any) -> None:
sample.put_bytes("key", expected)
actual = sample.get_bytes("key")
assert actual == expected
assert actual is not None
_assert_same_type(actual, expected)
def _assert_roundtrip_scalar(sample: Sample, expected: Any) -> None:
sample.put_scalar("key", expected)