Add unit tests to methods in codebase #48
Description
The codebase currently doesn't have any automated testing via unit tests. We can use the pytest library to write tests that test individual methods in the codebase, and have them be run by CI/CD (GitHub Actions) as well.