Add tests

[lukethehuman]

Add a test suite. Suggest using pytest: https://docs.pytest.org/en/8.2.x/