Under the TODOs you mention a unit test suite. Are you looking for unit tests beyond just assert statements? So using something like Python unittest module? Currently, there are only a few unit tests so before I add more I wanted to know which direction you'd like to go.