This whole file needs to be refactored. Look into other tests file how to do tests. They need to be written as:
```
def test1():
    assert wavefunction(1,x) == ...
    assert energy(2.2,2,5) == ...