mparray_test.py 文件源码

def test_norm(nr_sites, local_dim, rank, dtype, rgen):
    mp_psi = factory.random_mpa(nr_sites, local_dim, rank, randstate=rgen,
                                dtype=dtype)
    psi = mp_psi.to_array()

    assert_almost_equal(mp.inner(mp_psi, mp_psi), mp.norm(mp_psi)**2)
    assert_almost_equal(np.sum(psi.conj() * psi), mp.norm(mp_psi)**2)