def test_compress_obc(): dps = [1,3,2,2] mps1 = dMPX.rand(dps, D=7, bc="obc") print [m.shape for m in mps1] mps_diff1 = dMPX.add(MPX.mul(-1,mps1), mps1) assert not(np.isnan(MPX.norm(mps1))), MPX.norm(mps1) assert MPX.norm(mps_diff1)<1.0e-12, MPX.norm(mps_diff1) # check dimension preserving mps11 = dMPX.add(mps1,mps1) mps11,dwt = MPX.compress(mps11,0,preserve_dim="true") mps11_,dwt = MPX.compress(mps11,0,preserve_dim="false") print [m.shape for m in mps11] print [m.shape for m in mps11_] print [m.shape for m in mps1] print MPX.norm(mps11) assert(dwt == 0), dwt mps_diff = dMPX.add(MPX.mul(-2,mps1),mps11) print abs(MPX.norm(mps_diff)/MPX.norm(mps11))<1.0e-7, MPX.norm(mps_diff) mps_diff = dMPX.add(MPX.mul(-2,mps1),mps11_) print abs(MPX.norm(mps_diff)/MPX.norm(mps11_))<1.0e-7, MPX.norm(mps_diff)
def test_add(): dps = (2,2,2,2) #np.random.seed(417) mps_obc = dMPX.rand(dps, D=2, bc="obc") mps_obc2 = dMPX.rand(dps, D=2, bc="obc") mps_pbc = dMPX.rand(dps, D=2, bc="pbc") mps_pbc2 = dMPX.rand(dps, D=2, bc="pbc") assert np.allclose(np.linalg.norm(MPX.asfull(mps_obc)+MPX.asfull(mps_obc2)), MPX.norm(dMPX.add(mps_obc,mps_obc2))) assert np.allclose(np.linalg.norm(MPX.asfull(mps_pbc)+MPX.asfull(mps_pbc2)), MPX.norm(dMPX.add(mps_pbc,mps_pbc2))) assert np.allclose(np.linalg.norm(MPX.asfull(mps_obc)+MPX.asfull(mps_pbc2)), MPX.norm(dMPX.add(mps_obc,mps_pbc2)))
def test_product_state(): dp = (4, 5, 3, 2) occ_idx = (0, 1, 2, 1) mps1 = dMPX.product_state(dp, occ_idx, D=4, bc="obc") mps2 = dMPX.add(mps1, mps1) mps3 = dMPX.add(mps2, mps1) smps1 = sMPX.from_dense(mps1) smps3 = sMPX.from_dense(mps3) for i, m in enumerate(smps1): print i, m.coords.shape, m.shape, len(m.data), m.nnz assert m.nnz == 1 for i, m in enumerate(smps3): print i, m.coords.shape, m.shape, len(m.data), m.nnz assert m.nnz == 3
def test_dotcompress(): dps = [1,2,3,4,2] dpo = [(d,d) for d in dps] mps = dMPX.rand(dps) mpo = dMPX.rand(dpo) mps2 = MPX.dot(mpo,mps) mpsc, errc = MPX.compress(mps2,4) mpsdc, errdc = MPX.dot_compress(mpo,mps,4) print errc, errdc, errdc-errc diff_mps = dMPX.add(MPX.mul(-1,mpsc),mpsdc) assert(MPX.norm(diff_mps) < 1.0e-8), MPX.norm(diff_mps)
def test_compress_pbc(): np.random.seed(417) dps = [1,3,2,2] mps_obc = dMPX.rand(dps, D=7, bc="obc") print "full dim", MPX.obc_dim(dps) mps1 = dMPX.zeros(dps, D=7, bc="pbc") dMPX.overwrite(mps_obc, out=mps1) print MPX.norm(mps1), MPX.norm(mps_obc) mps2 = dMPX.add(mps1,mps1) mps11_0,dwt = MPX.compress(mps2,0,preserve_dim="true") mps11_1,dwt = MPX.compress(mps2,1,preserve_dim="true") mps11_2,dwt = MPX.compress(mps2,2,preserve_dim="true") mps11_2_,dwt = MPX.compress(mps2,2,preserve_dim="false") mps11_4,dwt = MPX.compress(mps2,4,preserve_dim="true") mps11_8,dwt = MPX.compress(mps2,8,preserve_dim="false") mps_diff0 = dMPX.add(MPX.mul(-1,mps11_0),mps2) mps_diff1 = dMPX.add(MPX.mul(-1,mps11_1),mps2) mps_diff2 = dMPX.add(MPX.mul(-1,mps11_2),mps2) mps_diff2_ = dMPX.add(MPX.mul(-1,mps11_2_),mps2) mps_diff4 = dMPX.add(MPX.mul(-1,mps11_4),mps2) mps_diff8 = dMPX.add(MPX.mul(-1,mps11_8),mps2) mps4o,dwt = MPX.compress(dMPX.add(mps1,mps1), 4) mps_diff4o = dMPX.add(MPX.mul(-1,mps4o),mps2) print "D full", abs(MPX.norm(mps_diff0)) print "D=1", abs(MPX.norm(mps_diff1)) print "D=2", abs(MPX.norm(mps_diff2)) print "D=2 (trunc)", abs(MPX.norm(mps_diff2_)) print "D=4", abs(MPX.norm(mps_diff4)) print "D=4 obc", abs(MPX.norm(mps_diff4o)) print "D=8", abs(MPX.norm(mps_diff8)) assert(abs(MPX.norm(mps_diff0))<1.0e-7)
def test_compress(): #np.random.seed(417) dps = [2] * 10 mps0 = dMPX.rand(dps, D=7, bc="obc") smps1 = sMPX.from_dense(mps0) print "Check sparse", [m.__class__.__name__ for m in smps1] # check dimension preserving mps00 = dMPX.add(mps0, mps0) smps11 = sMPX.add(smps1, smps1) print MPX.norm(mps00), MPX.norm(smps11) print "Initial dimensions", [m.shape for m in mps00] # compress for D in (2, 3, 7): mps00c = MPX.compress(mps00, D, preserve_dim="false") print "After DENSE compress", D, MPX.norm(mps00c) #print "Compressed dimensions", [m.shape for m in mps00c] print "IN SPARSE part" for D in (2, 3, 7): smps11c = MPX.compress(smps11, D, preserve_dim="false") print "After SPARSE compress", D, MPX.norm(smps11c)