Convert tests from `UnitTest` to `pytest`. For discussion, see #41