summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_shape_base.py
diff options
context:
space:
mode:
authorRoman Yurchak <rth.yurchak@pm.me>2018-11-24 20:59:42 +0100
committerRoman Yurchak <rth.yurchak@pm.me>2018-11-24 20:59:42 +0100
commit09992482c93f1b9e28b7958a792e6b3b709834fa (patch)
treeb807248742d02c48403d77b076f53b71f22df707 /numpy/lib/tests/test_shape_base.py
parent983bbb5ed2495e034b437a0b58a69371d4dfed74 (diff)
downloadnumpy-09992482c93f1b9e28b7958a792e6b3b709834fa.tar.gz
Use set litterals
Diffstat (limited to 'numpy/lib/tests/test_shape_base.py')
-rw-r--r--numpy/lib/tests/test_shape_base.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/lib/tests/test_shape_base.py b/numpy/lib/tests/test_shape_base.py
index e338467f9..01ea028bb 100644
--- a/numpy/lib/tests/test_shape_base.py
+++ b/numpy/lib/tests/test_shape_base.py
@@ -260,8 +260,8 @@ class TestApplyAlongAxis(object):
def test_with_iterable_object(self):
# from issue 5248
d = np.array([
- [set([1, 11]), set([2, 22]), set([3, 33])],
- [set([4, 44]), set([5, 55]), set([6, 66])]
+ [{1, 11}, {2, 22}, {3, 33}],
+ [{4, 44}, {5, 55}, {6, 66}]
])
actual = np.apply_along_axis(lambda a: set.union(*a), 0, d)
expected = np.array([{1, 11, 4, 44}, {2, 22, 5, 55}, {3, 33, 6, 66}])