#!/usr/bin/env node

const { calculateEntropy, validateProbabilities } = require('./entropy');

test('计算信源熵', () => {
    expect(calculateEntropy([0.5, 0.5])).toBe(1);
    expect(calculateEntropy([0.25, 0.25, 0.25, 0.25])).toBe(2);
});

test('验证概率', () => {
    expect(() => validateProbabilities([0.5, 0.5])).not.toThrow();
    expect(() => validateProbabilities([0.5, 0.3])).toThrow("概率之和必须等于 1");
    expect(() => validateProbabilities([1.5])).toThrow("每个概率必须在 0 和 1 之间");
});