TrFED.py 文件源码

def test_fed():
    print 'test_fed'
    t1,U,m=1.0,4.0,4
    lattice=Square('S1')('%sP-1O'%m)
    config=IDFConfig(priority=DEFAULT_FERMIONIC_PRIORITY,pids=lattice.pids,map=lambda pid: Fermi(norbital=1,nspin=2,nnambu=1))

    basis=FBasis(m*2,m,0.0)
    ed=FED(name='OneD_%s_%s'%(lattice.name,basis.rep),sectors=[basis],lattice=lattice,config=config,terms=[Hopping('t1',t1),Hubbard('U',U)],dtype=np.complex128)
    eigvals0=eigh(ed.matrix(basis.rep).todense(),eigvals_only=True)

    basis=TRBasis(FBasis(m*2,m,0.0),dk=2,nk=m)
    ed=TrFED(name='OneD_%s_%s'%(lattice.name,basis.rep),basis=basis,lattice=lattice,config=config,terms=[Hopping('t1',t1),Hubbard('U',U)],dtype=np.complex128)
    eigvals1=[]
    for k in xrange(m):
        eigvals1.append(eigh(ed.matrix(k=k).todense(),eigvals_only=True))
    eigvals1=sorted(np.concatenate(eigvals1))
    print 'diff: %s'%norm(eigvals0-eigvals1)
    print