Skip to content

Commit 359a991

Browse files
ysaualanbato
authored andcommitted
TST: Fix error message check in np.argsort comparision (pandas-dev#17051)
Closes pandas-devgh-17046.
1 parent 1626dba commit 359a991

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pandas/tests/indexes/test_base.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1846,7 +1846,7 @@ def create_index(self):
18461846
def test_argsort(self):
18471847
idx = self.create_index()
18481848
if PY36:
1849-
with tm.assert_raises_regex(TypeError, "'>' not supported"):
1849+
with tm.assert_raises_regex(TypeError, "'>|<' not supported"):
18501850
result = idx.argsort()
18511851
elif PY3:
18521852
with tm.assert_raises_regex(TypeError, "unorderable types"):
@@ -1859,7 +1859,7 @@ def test_argsort(self):
18591859
def test_numpy_argsort(self):
18601860
idx = self.create_index()
18611861
if PY36:
1862-
with tm.assert_raises_regex(TypeError, "'>' not supported"):
1862+
with tm.assert_raises_regex(TypeError, "'>|<' not supported"):
18631863
result = np.argsort(idx)
18641864
elif PY3:
18651865
with tm.assert_raises_regex(TypeError, "unorderable types"):

0 commit comments

Comments
 (0)