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

// 测试熵计算
test('calculateEntropy works correctly', () => {
    const probabilities = [0.5, 0.5];
    const result = calculateEntropy(probabilities);
    expect(result).toBeCloseTo(1, 3);
});

// 测试概率验证
test('validateProbabilities works correctly', () => {
    const probabilities = [0.5, 0.5];
    expect(() => validateProbabilities(probabilities)).not.toThrow();
});

test('validateProbabilities throws error for invalid probabilities', () => {
    const invalidProbabilities = [0.5, 0.6];
    expect(() => validateProbabilities(invalidProbabilities)).toThrow('概率总和必须为 1！');
});