Switch from `unittest` to `pytest`