Add pytests for Travis-CI