Pytest Main Function In Python