d1manson

{% test enum_distribution(model, column_name, expected, allow_other=False, fudge=0.1) %}

    SELECT
        COALESCE(expected.enum_val, actual.enum_val) AS enum_val,
        expected.fraction AS expected_fraction,
        actual.fraction AS actual_fraction
    FROM (
        {% for enum_val, expected_fraction in expected.items() -%}
            SELECT '{{enum_val}}' AS enum_val, {{expected_fraction}} AS fraction
            {%- if not loop.last %} UNION ALL {% endif %}