summaryrefslogtreecommitdiff
path: root/numpy/lib/tests
diff options
context:
space:
mode:
authorCharles Harris <charlesr.harris@gmail.com>2013-04-13 15:34:26 -0600
committerCharles Harris <charlesr.harris@gmail.com>2013-04-13 17:04:13 -0600
commit0dfe67afd1ee9e4c905bf119673f6e634221f21b (patch)
treea45a1e39de66d5d462a69040ea7e996cf234a011 /numpy/lib/tests
parente589c6ed1dac7755bb7bd9e181a43ebeff62dcec (diff)
downloadnumpy-0dfe67afd1ee9e4c905bf119673f6e634221f21b.tar.gz
2to3: Apply zip fixer.
In Python 3 zip returns an iterator instead of a list. Consequently, in places where an iterator won't do it must be enclosed in list(...). Lists instead of iterators are also used in array constructors as using iterators there usually results in an object array containing an iterator object. Closes #3094
Diffstat (limited to 'numpy/lib/tests')
-rw-r--r--numpy/lib/tests/test_arraysetops.py4
-rw-r--r--numpy/lib/tests/test_io.py4
-rw-r--r--numpy/lib/tests/test_recfunctions.py24
3 files changed, 16 insertions, 16 deletions
diff --git a/numpy/lib/tests/test_arraysetops.py b/numpy/lib/tests/test_arraysetops.py
index f7f75922d..4ba6529e4 100644
--- a/numpy/lib/tests/test_arraysetops.py
+++ b/numpy/lib/tests/test_arraysetops.py
@@ -61,8 +61,8 @@ class TestSetOps(TestCase):
# test for structured arrays
dt = [('', 'i'), ('', 'i')]
- aa = np.array(zip(a,a), dt)
- bb = np.array(zip(b,b), dt)
+ aa = np.array(list(zip(a,a)), dt)
+ bb = np.array(list(zip(b,b)), dt)
check_all(aa, bb, i1, i2, dt)
diff --git a/numpy/lib/tests/test_io.py b/numpy/lib/tests/test_io.py
index af4eed05d..5987a15b0 100644
--- a/numpy/lib/tests/test_io.py
+++ b/numpy/lib/tests/test_io.py
@@ -517,7 +517,7 @@ class TestLoadTxt(TestCase):
c = TextIO(data)
names = ['stid', 'temp']
dtypes = ['S4', 'f8']
- arr = np.loadtxt(c, usecols=(0, 2), dtype=zip(names, dtypes))
+ arr = np.loadtxt(c, usecols=(0, 2), dtype=list(zip(names, dtypes)))
assert_equal(arr['stid'], [b"JOE", b"BOB"])
assert_equal(arr['temp'], [25.3, 27.9])
@@ -1119,7 +1119,7 @@ M 33 21.99
data = TextIO("JOE 70.1 25.3\nBOB 60.5 27.9")
names = ['stid', 'temp']
dtypes = ['S4', 'f8']
- test = np.ndfromtxt(data, usecols=(0, 2), dtype=zip(names, dtypes))
+ test = np.ndfromtxt(data, usecols=(0, 2), dtype=list(zip(names, dtypes)))
assert_equal(test['stid'], [b"JOE", b"BOB"])
assert_equal(test['temp'], [25.3, 27.9])
diff --git a/numpy/lib/tests/test_recfunctions.py b/numpy/lib/tests/test_recfunctions.py
index 899031451..ef22ca413 100644
--- a/numpy/lib/tests/test_recfunctions.py
+++ b/numpy/lib/tests/test_recfunctions.py
@@ -543,11 +543,11 @@ class TestStackArrays(TestCase):
class TestJoinBy(TestCase):
def setUp(self):
- self.a = np.array(zip(np.arange(10), np.arange(50, 60),
- np.arange(100, 110)),
+ self.a = np.array(list(zip(np.arange(10), np.arange(50, 60),
+ np.arange(100, 110))),
dtype=[('a', int), ('b', int), ('c', int)])
- self.b = np.array(zip(np.arange(5, 15), np.arange(65, 75),
- np.arange(100, 110)),
+ self.b = np.array(list(zip(np.arange(5, 15), np.arange(65, 75),
+ np.arange(100, 110))),
dtype=[('a', int), ('b', int), ('d', int)])
#
def test_inner_join(self):
@@ -620,11 +620,11 @@ class TestJoinBy(TestCase):
class TestJoinBy2(TestCase):
@classmethod
def setUp(cls):
- cls.a = np.array(zip(np.arange(10), np.arange(50, 60),
- np.arange(100, 110)),
+ cls.a = np.array(list(zip(np.arange(10), np.arange(50, 60),
+ np.arange(100, 110))),
dtype=[('a', int), ('b', int), ('c', int)])
- cls.b = np.array(zip(np.arange(10), np.arange(65, 75),
- np.arange(100, 110)),
+ cls.b = np.array(list(zip(np.arange(10), np.arange(65, 75),
+ np.arange(100, 110))),
dtype=[('a', int), ('b', int), ('d', int)])
def test_no_r1postfix(self):
@@ -660,12 +660,12 @@ class TestJoinBy2(TestCase):
assert_equal(test, control)
def test_two_keys_two_vars(self):
- a = np.array(zip(np.tile([10,11],5),np.repeat(np.arange(5),2),
- np.arange(50, 60), np.arange(10,20)),
+ a = np.array(list(zip(np.tile([10,11],5),np.repeat(np.arange(5),2),
+ np.arange(50, 60), np.arange(10,20))),
dtype=[('k', int), ('a', int), ('b', int),('c',int)])
- b = np.array(zip(np.tile([10,11],5),np.repeat(np.arange(5),2),
- np.arange(65, 75), np.arange(0,10)),
+ b = np.array(list(zip(np.tile([10,11],5),np.repeat(np.arange(5),2),
+ np.arange(65, 75), np.arange(0,10))),
dtype=[('k', int), ('a', int), ('b', int), ('c',int)])
control = np.array([(10, 0, 50, 65, 10, 0), (11, 0, 51, 66, 11, 1),