def encode_mx_ndarray(v: mx.nd.NDArray) -> Any:
return {
"__kind__": kind_inst,
"class": "mxnet.nd.array",
"args": encode([v.asnumpy().tolist()]),
"kwargs": {
"dtype": encode(v.dtype)
},
}
def test_serde_fq():
add_ = serde.decode(serde.encode(add))
assert add_(1, 2) == 3
def foo():
pass
with pytest.raises(Exception):
serde.encode(foo)
def test_stateless():
m = StatelessClass(1, 2)
m2 = serde.decode(serde.encode(m))
assert m.x == m2.x
assert m.y == m2.y
assert m.z == m2.z
def encode_mx_context(v: mx.Context) -> Any:
"""
Specializes :func:`encode` for invocations where ``v`` is an instance of
the :class:`~mxnet.Context` class.
"""
return {
"__kind__": kind_inst,
"class": fqname_for(v.__class__),
"args": encode([v.device_type, v.device_id]),
}
def test_serde_stateful():
o = MyClass()
o2 = serde.decode(serde.encode(o))
assert o.n == o2.n
def test_timestamp_encode_decode() -> None:
now = pd.Timestamp.now()
assert now == serde.decode(serde.encode(now))
def test_stateless_typed():
m = StatelessClassTyped(a="42")
assert m.a == 42
assert serde.encode(m)["kwargs"]["a"] == 42
def test_prophet_serialization():
predictor = ProphetPredictor(freq="1D", prediction_length=3)
assert predictor == serde.decode(serde.encode(predictor))
def test_r_predictor_serialization():
predictor = RForecastPredictor(freq="1D", prediction_length=3)
assert predictor == serde.decode(serde.encode(predictor))
def test_distribution_output_serde(distr_output: DistributionOutput):
distr_output_copy = decode(encode(distr_output))
assert isinstance(distr_output_copy, type(distr_output))
assert dump_json(distr_output_copy) == dump_json(distr_output)
def test_serde_method():
x = X()
m = serde.decode(serde.encode(x.m))
assert m() == 42
def test_serde_partial():
add_1 = partial(add, 1)
add_1_ = serde.decode(serde.encode(add_1))
assert add_1_(2) == 3
