`unittest` is lacking compared to `pytest`. I'll take a pass at getting `pytest` setup and write some initial tests.