Skip to content

Commit a6b3173

Browse files
committed
Use faster assertion for test_qcut_binning_issues
1 parent e52fe69 commit a6b3173

File tree

1 file changed

+4
-17
lines changed

1 file changed

+4
-17
lines changed

pandas/tests/reshape/test_qcut.py

Lines changed: 4 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -101,23 +101,10 @@ def test_qcut_binning_issues(datapath):
101101
arr = np.loadtxt(cut_file)
102102
result = qcut(arr, 20)
103103

104-
starts = []
105-
ends = []
106-
107-
for lev in np.unique(result):
108-
s = lev.left
109-
e = lev.right
110-
assert s != e
111-
112-
starts.append(float(s))
113-
ends.append(float(e))
114-
115-
for (sp, sn), (ep, en) in zip(
116-
zip(starts[:-1], starts[1:]), zip(ends[:-1], ends[1:])
117-
):
118-
assert sp < sn
119-
assert ep < en
120-
assert ep <= sn
104+
starts = result.categories.left
105+
ends = result.categories.right
106+
assert (starts < ends).all()
107+
assert (starts[1:] <= ends[:-1]).all()
121108

122109

123110
def test_qcut_return_intervals():

0 commit comments

Comments
 (0)