def test_binary_serialization(e) -> None:
assert equals(e, serde.load_binary(serde.dump_binary(e)))
"a",
[
mx.nd.random.uniform(shape=(3, 5, 2), dtype="float16"),
mx.nd.random.uniform(shape=(3, 5, 2), dtype="float32"),
mx.nd.random.uniform(shape=(3, 5, 2), dtype="float64"),
mx.nd.array([[1, 2, 3], [-1, -2, 0]], dtype=np.uint8),
mx.nd.array([[1, 2, 3], [-1, -2, 0]], dtype=np.int32),
mx.nd.array([[1, 2, 3], [-1, -2, 0]], dtype=np.int64),
mx.nd.array([[1, 2, 3], [1, 2, 0]], dtype=np.uint8),
],
)
@pytest.mark.parametrize(
"serialize_fn",
[
lambda x: serde.load_json(serde.dump_json(x)),
lambda x: serde.load_binary(serde.dump_binary(x)),
lambda x: serde.load_code(serde.dump_code(x)),
],
)
def test_ndarray_serialization(a, serialize_fn) -> None:
b = serialize_fn(a)
assert type(a) == type(b)
assert a.dtype == b.dtype
assert a.shape == b.shape
assert np.all((a == b).asnumpy())
def test_timestamp_encode_decode() -> None:
now = pd.Timestamp.now()
assert now == serde.decode(serde.encode(now))