예제 #1
0
def test_dist():
    """
    Test that the construct dist is accurate.
    """
    mimi = MinimalIntrinsicTotalCorrelation(intrinsic_1, [[0], [1]], [2], bound=3)
    mimi.optimize()
    d = mimi.construct_distribution()
    assert entropy(d, [3]) == pytest.approx(1.5, abs=1e-2)
예제 #2
0
def test_failure():
    """
    Test that things fail with no CRV specified.
    """
    with pytest.raises(ditException):
        MinimalIntrinsicTotalCorrelation(intrinsic_1, [[0], [1]])