Add unit tests

[ericpan64]

Will give a good baseline understanding of the project and ensure changes aren't breaking. Try pytest