Added pytest for better testing with vim and python