Improve pytest folder structure

[kan-fu]

See #24 (comment)

Also there are some warnings on the GitHub Actions. Update actions/setup-python@v4.