def test_distribution_collection_serialization():
dists = DistributionCollection([('norm', Normal(5, 2), 2), ('bern', Bernoulli(), 5)])
json_str = dists.to_json()
dists_loaded = load_from_json(json_str)
assert dists_loaded.dtype == dists.dtype
assert dists_loaded.norm_dtype == dists.norm_dtype
n = 50
arr = dists.sample(n)
arr_norm = dists.normalize(arr)
arr_norm_loaded = dists_loaded.normalize(arr)
for name in dists.names:
np.testing.assert_allclose(arr_norm[name], arr_norm_loaded[name])
n = 10000
arr = dists.sample(n)
arr_loaded = dists_loaded.sample(n)
for name in dists.names:
assert abs(arr[name].mean() - arr_loaded[name].mean()) <= 0.10
assert abs(arr[name].std() - arr_loaded[name].std()) <= 0.10
assert dists_loaded.names == dists.names
assert dists_loaded.dtype == dists.dtype
assert dists_loaded.norm_dtype == dists.norm_dtype
assert dists_loaded.distributions == dists.distributions
assert dists_loaded.normalizations == dists.normalizations
assert dists_loaded.nb_elems == dists.nb_elems