Add type hinting

1 files changed, 3 insertions, 7 deletions

diff --git a/test/test_graphic.py b/test/test_graphic.py
index 73d8b82..8433cae 100644
--- a/test/test_graphic.py
+++ b/test/test_graphic.py
@@ -1,7 +1,3 @@
-from pathlib import Path
-import sys
-sys.path.insert(0, str(Path.cwd().parent/'schroedinger'))
-
 import pytest
 import numpy as np
 
@@ -18,19 +14,19 @@ v = {}
 FORMS = ['asymmetric', 'double_linear', 'double_cubic']
 
 @pytest.mark.parametrize('form', FORMS)
-def test_potential(form):
+def test_potential(form: str) -> None:
     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):
+def test_e(form: str) -> None:
     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):
+def test_v(form: str) -> None:
     v_prec = np.loadtxt(f'test/{form}/wavefuncs.dat')[:, 1:]
     assert np.allclose(v[form], v_prec, rtol=1e-2, atol=1e-2)