summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_arrayterator.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/lib/tests/test_arrayterator.py')
-rw-r--r--numpy/lib/tests/test_arrayterator.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/numpy/lib/tests/test_arrayterator.py b/numpy/lib/tests/test_arrayterator.py
index 28feb4824..3fa15fef1 100644
--- a/numpy/lib/tests/test_arrayterator.py
+++ b/numpy/lib/tests/test_arrayterator.py
@@ -1,15 +1,13 @@
from __future__ import division, absolute_import
from operator import mul
+from functools import reduce
import numpy as np
from numpy.random import randint
from numpy.lib import Arrayterator
from numpy.testing import assert_
-import sys
-if sys.version_info[0] >= 3:
- from functools import reduce
def test():
np.random.seed(np.arange(10))