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))