Added pytest for travis