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