from CalculateEntropy import calculate_shannon_entropy


def test_calculate_shannon_entropy():
    _, context_result, _ = calculate_shannon_entropy('1234', 1)
    assert context_result == {'EntropyResult': {'checked_value': '1234', 'entropy': 2.0}}

    _, context_result, _ = calculate_shannon_entropy('1234', 3)
    assert context_result == {}

from CalculateEntropy import calculate_shannon_entropy


def test_calculate_shannon_entropy():
    _, context_result, _ = calculate_shannon_entropy('1234', 1)
    assert context_result == {'EntropyResult': {'checked_value': '1234', 'entropy': 2.0}}

    _, context_result, _ = calculate_shannon_entropy('1234', 3)
    assert context_result == {}
