Set up testing framework

Decide on and set up the framework for testing, e.g. via pytest, and potentially tools such as tox/nox for more robust testing, e.g. to allow local testing of multiple Python versions.

Note, there will be a separate issue for running tests as part of the CI.