Update pytest.py