Use pytest instead of nose