diff options
author | Tim Suhling <tim@suhling.com> | 2024-07-19 15:12:35 +0200 |
---|---|---|
committer | Tim Suhling <tim@suhling.com> | 2024-07-19 15:12:35 +0200 |
commit | c02bb4ac9df63c3bed3a0771e02e0f6900e3074d (patch) | |
tree | 763abf36441a1518fd5a825824df9c9b2bf329f9 /test/test_infinite.py | |
parent | 1c0aee1f87f9f35da422f51601e3baffda071d19 (diff) | |
parent | cf5c404add0c69d25effbc8e6cc10545918ca4b8 (diff) |
Merge branch 'master' into plot
Diffstat (limited to 'test/test_infinite.py')
-rw-r--r-- | test/test_infinite.py | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/test/test_infinite.py b/test/test_infinite.py new file mode 100644 index 0000000..aa85ae5 --- /dev/null +++ b/test/test_infinite.py @@ -0,0 +1,45 @@ +import pytest +import numpy as np + +import matplotlib.pyplot as plt + +from schroedinger import ( + Config, potential_interp, build_potential, solve_schroedinger +) + + +def psi(x, n, a): + n += 1 # Index starting from 0 + x = -x + a / 2 # Reflect x and move to the left + return np.sqrt(2 / a) * np.sin(n * np.pi * x / a) + + +def energy(n, a, m): + return (n + 1)**2 * np.pi**2 / m / a**2 / 2.0 + + +def test_infinite(): + conf = Config('test/infinite.inp') + potential, delta = build_potential(conf) + + e, v = solve_schroedinger(conf.mass, potential[:, 1], delta, conf.eig_interval) + # Account for -v also being an eigenvector if v is one + v = np.abs(v) + + a = np.abs(conf.points[1][0] - conf.points[0][0]) + e_theory = np.zeros(e.shape) + v_theory = np.zeros(v.shape) + + + for n in range(e.shape[0]): + e_theory[n] = energy(n, a, conf.mass) + v_theory[:, n] = np.abs(psi(potential[:, 0], n, a)) + + # for n in range(e.shape[0]): + # plt.plot(potential[:, 0], np.abs(v[:, n]), label='Num{}'.format(n)) + # plt.plot(potential[:, 0], np.abs(v_theory[:, n]), ls='--', label='Theory{}'.format(n)) + # plt.legend() + # plt.savefig('test.pdf') + + assert (np.allclose(e, e_theory, rtol=1e-2, atol=1e-2) + and np.allclose(v, v_theory, rtol=1e-2, atol=1e-2)) |