import numpy as np
import pytest

from sarus_statistics.ops.max_multiplicity.local import max_multiplicity

np.random.seed(0)
NOISE = 1e-9
MAX_MAX_MULTIPLICITY = 100


def test_max_multiplicity(ops_data, admin_cols):
    public, user_col, weights = admin_cols
    max_multiplicity_value = max_multiplicity(
        ops_data,
        user_col=user_col,
        private_col=public,
        weight_col=weights,
        epsilon_queries=1 / NOISE,
        noise_user_count=NOISE,
        noise_multiplicity=NOISE,
        max_max_multiplicity=MAX_MAX_MULTIPLICITY,
    )

    assert pytest.approx(max_multiplicity_value, 1e-2) == 10