Switch to pytest
Add travis CI and pytest files