aboutsummaryrefslogtreecommitdiff
path: root/test/test_graphic.py
blob: 73d8b82df517eecd4c8d54ef90fbade13348a906 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
from pathlib import Path
import sys
sys.path.insert(0, str(Path.cwd().parent/'schroedinger'))

import pytest
import numpy as np

import matplotlib.pyplot as plt

from schroedinger import (
    Config, potential_interp, build_potential, solve_schroedinger
)

potential = {}
e = {}
v = {}

FORMS = ['asymmetric', 'double_linear', 'double_cubic']

@pytest.mark.parametrize('form', FORMS)
def test_potential(form):
    potential_prec = np.loadtxt(f'test/{form}/potential.dat')
    assert np.allclose(potential[form], potential_prec, rtol=1e-2, atol=1e-2)


@pytest.mark.parametrize('form', FORMS)
def test_e(form):
    e_prec = np.loadtxt(f'test/{form}/energies.dat')
    assert np.allclose(e[form], e_prec, rtol=1e-2, atol=1e-2)


@pytest.mark.parametrize('form', FORMS)
def test_v(form):
    v_prec = np.loadtxt(f'test/{form}/wavefuncs.dat')[:, 1:]
    assert np.allclose(v[form], v_prec, rtol=1e-2, atol=1e-2)


def setup_module():
    global conf
    global potential
    global e
    global v

    for form in FORMS:
        conf = Config(f'test/{form}.inp')
        potential[form], delta = build_potential(conf)
        e[form], v[form] = solve_schroedinger(conf.mass, potential[form][:, 1],
                                              delta, conf.eig_interval)