def runHubbardAtom(self):
h = Hubbard([[-0.5]], 1)
h.solve()
self.assertEqual(-0.5, h.getGroundStateEnergy())
self.assertEqual(h.getEnergies(), [0, 0.5])
self.assertTrue((h.eigenEnergies == numpy.array([0.5, 0, 0, 0.5])).all())
for state in h.getGroundStatesAlgebraically():
self.assertTrue(state in ["+1.0 c^('up', 0)\n", "+1.0 c^('dn', 0)\n"])
for states_e, e in zip(h.getStatesEnergySortedAlgebraically(), h.getEnergies()):
for state in states_e:
if e == 0:
self.assertTrue(state in ["+1.0 c^('up', 0)\n", "+1.0 c^('dn', 0)\n"])
elif e == 0.5:
self.assertTrue(state in ["+1.0 \n", "+1.0 c^('up', 0) c^('dn', 0)\n"])
else:
self.assertTrue(False)
c = AnnihilationOperator(h.getSingleParticleBasis())
n_tot_hat = numpy.sum([c[s, 0].H.dot(c[s, 0]) for s in ["up", "dn"]], axis=0)
self.assertEqual(h.getGroundStates()[0].getQuantumNumber(n_tot_hat), 1)
self.assertEqual(h.getGroundStates()[1].getQuantumNumber(n_tot_hat), 1)
from EasyED.hamiltonians import Hubbard
from EasyED.operators import AnnihilationOperator
from EasyED.util import report
from itertools import product
from matplotlib import pyplot as plt
from numpy import array, sqrt, sort, arange
import numpy
numpy.set_printoptions(suppress=True)
structure = Hubbard([[.5, 0], [0, .5]], 1)
c = AnnihilationOperator(structure.getSingleParticleBasis())
report(structure.blocksizes)
structure.solve()
print structure.eigenEnergies
print 'groundstateenergy: ', structure.getGroundStateEnergy()
print 'groundstates:'
for state in structure.getGroundStatesAlgebraically():
print state
print
print 'states:'
for energyGroup, energy in zip(structure.getStatesEnergySortedAlgebraically(), structure.getEnergies()):
print 'E = ', energy
print
for state in energyGroup:
print state
