Updated example unittest to pytest.