Switching to pytest
Update travis of the changes.