Allow py3 and py2 message of import error in test