Improve Code Testing and Coverage

[JHartzer]

Consider adding a test framework such as pytest or unittest and coupling that with a tool such as coverage. This would help improve understanding of which lines of code are being tested and identify any gaps in testing coverage.