Function discrete_distribution

tests/python/test_docexamples_f32.py:712–725

()

            coder.decode(model, model_parameters[:, 0].copy(), model_parameters[:, 1].copy()) == symbols)

    def discrete_distribution():
        model = constriction.stream.model.CustomModel(
            lambda x, params: scipy.stats.binom.cdf(x, n=10, p=params),
            lambda x, params: scipy.stats.binom.ppf(x, n=10, p=params),
            0, 10)

        success_probabilities = np.array([0.3, 0.7, 0.2, 0.6])

        symbols = np.array([4, 8, 1, 5], dtype=np.int32)
        coder = constriction.stream.stack.AnsCoder()
        coder.encode_reverse(
            symbols, model, success_probabilities)
        assert np.all(
            coder.decode(model, success_probabilities) == symbols)

    fixed_model_params()
    variable_model_params()