main.py

from next_num import RandomGen

from collections import Counter


def main():
    random_nums = [-1, 0, 1, 2, 3]
    probabilities = [0.01, 0.3, 0.58, 0.1, 0.01]

    rg = RandomGen(random_nums, probabilities)
    
    total_runs = 10000
    frequencies = Counter(rg.next_num() for _ in range(total_runs))
    # frequencies = Counter(rg.next_num_o_of_n() for _ in range(total_runs))

    expected = dict(zip(random_nums, probabilities))
    for num, freq in frequencies.items():
        print(f"{num} actual freq: {freq / total_runs}, expected: {expected[num]}")


main()