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