As we get closer to moving to code cleanup, I'd like to start adding some unit tests. I would prefer to use pytest rather than unittest.