summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_mixins.py
diff options
context:
space:
mode:
authorSeth Troisi <sethtroisi@google.com>2020-01-08 23:49:17 -0800
committerSeth Troisi <sethtroisi@google.com>2020-01-15 13:19:56 -0800
commit1427484e9015e73b7017ee9336ce914a6f15187b (patch)
tree2db89078addee6a01697ab8bd4160c25d59fffc4 /numpy/lib/tests/test_mixins.py
parentb6bc0941d4f07310456079ab2497c3d1bde4a5e7 (diff)
downloadnumpy-1427484e9015e73b7017ee9336ce914a6f15187b.tar.gz
MAINT: Remove sys.version checks in tests
Diffstat (limited to 'numpy/lib/tests/test_mixins.py')
-rw-r--r--numpy/lib/tests/test_mixins.py12
1 files changed, 4 insertions, 8 deletions
diff --git a/numpy/lib/tests/test_mixins.py b/numpy/lib/tests/test_mixins.py
index 7c22dae94..e184ffe19 100644
--- a/numpy/lib/tests/test_mixins.py
+++ b/numpy/lib/tests/test_mixins.py
@@ -6,9 +6,6 @@ import numpy as np
from numpy.testing import assert_, assert_equal, assert_raises
-PY2 = sys.version_info.major < 3
-
-
# NOTE: This class should be kept as an exact copy of the example from the
# docstring for NDArrayOperatorsMixin.
@@ -202,11 +199,10 @@ class TestNDArrayOperatorsMixin:
array_like = ArrayLike(array)
expected = ArrayLike(np.float64(5))
_assert_equal_type_and_value(expected, np.matmul(array_like, array))
- if not PY2:
- _assert_equal_type_and_value(
- expected, operator.matmul(array_like, array))
- _assert_equal_type_and_value(
- expected, operator.matmul(array, array_like))
+ _assert_equal_type_and_value(
+ expected, operator.matmul(array_like, array))
+ _assert_equal_type_and_value(
+ expected, operator.matmul(array, array_like))
def test_ufunc_at(self):
array = ArrayLike(np.array([1, 2, 3, 4]))