def test_potential1(): """The funtion tests the potential """ refpot1 = di.readcalcpotential('reference_files/potential1.dat') pot1 = di.readcalcpotential('testfiles/potentialtest1.dat') assert np.all(pot1-refpot1 < 0.01)
""" ii = 0 plt.figure() while ii <= numev: vec = evec[:, ii] nvec = slv.normalize(vec) nvec = float(scalingfactor) * nvec + ev[ii] plt.subplot(1, 2, 1) #first plot plt.plot(xx, pot) plt.title('Potential, eigenstates, Erw(x)') plt.plot(np.array(erw), ev, 'x', color='blue') plt.plot(xx, nvec) plt.xlabel('x[Bohr]') plt.ylabel('Energy[Hartree]') plt.plot(xx, np.zeros(nn + 1) + ev[ii], color='grey') plt.xlim(di.getxmin(), di.getxmax()) plt.ylim(float(llimit), float(ulimit)) ii += 1 plt.subplot(1, 2, 2) #second plot plt.plot(unschaerfe, ev, 'x', color='blue') plt.title('$\sigma_x$') plt.xlabel('x[Bohr]') plt.savefig("schroedinger.pdf") plotwavefunc(NUMEV, EIVE, di.readcalcpotential("potential.dat"), ERW, di.readcalceigenval("energies.dat"), di.readcalcxvalue("potential.dat"), N, UNSCHAERFE, ARGS.scale, ARGS.ulimit, ARGS.llimit)
def test_potential(): """The function tests the potentials """ refpot2 = di.readcalcpotential('reference_files/potential2.dat') pot2 = di.readcalcpotential('testfiles/potentialtest2.dat') assert np.all(pot2-refpot2 < 0.01)
def test_potential(): """The funtion tests the potential """ refpot4 = di.readcalcpotential('reference_files/potential4.dat') pot4 = di.readcalcpotential('testfiles/potentialtest4.dat') assert np.all(pot4 - refpot4 < 0.01)