Add some unit tests w/ pytest.