8c271d54da8f609c9b56e9d0ab98d453b3e4f81c,tensorly/metrics/tests/test_entropy.py,,test_vonNeumann_entropy_pure_state,#,18

Before Change

def test_vonNeumann_entropy_pure_state():
    tl_vne = np.array(vonNeumann_entropy(mat_pure))
    assert np.allclose(tl_vne, np.array([0]), rtol=1e-03, atol=1e-03)

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))

After Change

actual_vne = 0.5546

def test_vonNeumann_entropy_pure_state():
    state = tl.randn((8, 1))
    state = state/tl.norm(state)
    mat_pure = tl.dot(state, tl.transpose(state))
    tl_vne = vonneumann_entropy(mat_pure)
    tl.testing.assert_array_almost_equal(tl_vne, 0, decimal=3)

Instances