8c271d54da8f609c9b56e9d0ab98d453b3e4f81c,tensorly/metrics/tests/test_entropy.py,,test_vonNeumann_entropy_pure_state_CP,#,22

Before Change

def test_vonNeumann_entropy_pure_state_CP():
    mat = parafac(tl.tensor(mat_pure), rank=1, normalize_factors=True)
    tl_vne = np.array(vonNeumann_entropy(mat))
    assert np.allclose(tl_vne, np.array([0]), rtol=1e-03, atol=1e-03)

def test_vonNeumann_entropy_mixed_state():
    tl_vne = np.array(vonNeumann_entropy(mat_mixed))

After Change

    tl.testing.assert_array_almost_equal(tl_vne, 0, decimal=3)

def test_vonNeumann_entropy_pure_state_CP():
    state = tl.randn((8, 1))
    state = state/tl.norm(state)
    mat_pure = tl.dot(state, tl.transpose(state))
    mat = parafac(mat_pure, rank=1, normalize_factors=True)
    tl_vne = cp_vonneumann_entropy(mat)
    tl.testing.assert_array_almost_equal(tl_vne, 0, decimal=3)

Instances