removed more duplication of python test