Having a suite of pytest test cases would definitely be useful for this project, to make sure all of our changes aren't breaking anything