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