def test_iadd1(self):
g = Grid([10, 10, 10])
g.fill(1)
old = g.copy()
g += g
g -= old
assert np.allclose(g.grid, 1)
g -= g
assert np.allclose(g.grid, 0)
class t():
def __init__(self):
alat = 1.42
sq3h = 3.**.5 * 0.5
self.sc = SuperCell(np.array([[1.5, sq3h, 0.],
[1.5, -sq3h, 0.],
[0., 0., 10.]], np.float64) * alat, nsc=[3, 3, 1])
self.g = Grid([10, 10, 100], sc=self.sc)
self.g.fill(2.)
def test_wavefunction1():
N = 50
o1 = SphericalOrbital(0, (np.linspace(0, 2, N), np.exp(-np.linspace(0, 100, N))))
G = Geometry([[1] * 3, [2] * 3], Atom(6, o1), sc=[4, 4, 4])
H = Hamiltonian(G)
R, param = [0.1, 1.5], [1., 0.1]
H.construct([R, param])
ES = H.eigenstate(dtype=np.float64)
# Plot in the full thing
grid = Grid(0.1, geometry=H.geom)
grid.fill(0.)
ES.sub(0).wavefunction(grid)
def test_wavefunction_eta():
N = 50
o1 = SphericalOrbital(0, (np.linspace(0, 2, N), np.exp(-np.linspace(0, 100, N))))
G = Geometry([[1] * 3, [2] * 3], Atom(6, o1), sc=[4, 4, 4])
H = Hamiltonian(G, spin=Spin('nc'))
R, param = [0.1, 1.5], [[0., 0., 0.1, -0.1],
[1., 1., 0.1, -0.1]]
H.construct([R, param])
ES = H.eigenstate()
# Plot in the full thing
grid = Grid(0.1, dtype=np.complex128, sc=SuperCell([2, 2, 2], origo=[-1] * 3))
grid.fill(0.)
ES.sub(0).wavefunction(grid, eta=True)
def test_wavefunction2():
N = 50
o1 = SphericalOrbital(0, (np.linspace(0, 2, N), np.exp(-np.linspace(0, 100, N))))
G = Geometry([[1] * 3, [2] * 3], Atom(6, o1), sc=[4, 4, 4])
H = Hamiltonian(G)
R, param = [0.1, 1.5], [1., 0.1]
H.construct([R, param])
ES = H.eigenstate(dtype=np.float64)
# This is effectively plotting outside where no atoms exists
# (there could however still be psi weight).
grid = Grid(0.1, sc=SuperCell([2, 2, 2], origo=[2] * 3))
grid.fill(0.)
ES.sub(0).wavefunction(grid)