From 069e9b57f203d9e71b78246b03e30e3928ceaa4d Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Mon, 10 Jul 2017 23:57:18 +0200 Subject: [PATCH] Fix an issue for python 2. --- test/test_prefix.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/test/test_prefix.py b/test/test_prefix.py index 369e6c8..0c79958 100644 --- a/test/test_prefix.py +++ b/test/test_prefix.py @@ -1,4 +1,9 @@ -from itertools import zip_longest +try: + from itertools import zip_longest +except ImportError: + # Python 2 + from itertools import izip_longest as zip_longest + from codecs import BOM_UTF8 import pytest