@@ -272,18 +272,20 @@ def test_series_retbins(self):
272
272
np .array ([0 , 0 , 1 , 1 ], dtype = np .int8 ))
273
273
tm .assert_numpy_array_equal (bins , np .array ([0 , 1.5 , 3 ]))
274
274
275
- def test_qcut_duplicates_drop (self ):
275
+ def test_qcut_duplicates_bin (self ):
276
276
# GH 7751
277
277
values = [0 , 0 , 0 , 0 , 1 , 2 , 3 ]
278
+ result_levels = ['[0, 1]' , '(1, 3]' ]
279
+
278
280
cats = qcut (values , 3 , duplicates = 'drop' )
279
- ex_levels = ['[0, 1]' , '(1, 3]' ]
280
- self .assertTrue ((cats .categories == ex_levels ).all ())
281
+ self .assertTrue ((cats .categories == result_levels ).all ())
281
282
282
- def test_qcut_duplicates_raise (self ):
283
- # GH 7751
284
- values = [0 , 0 , 0 , 0 , 1 , 2 , 3 ]
283
+ self .assertRaises (ValueError , qcut , values , 3 )
285
284
self .assertRaises (ValueError , qcut , values , 3 , duplicates = 'raise' )
286
285
286
+ # invalid
287
+ self .assertRaises (ValueError , qcut , values , 3 , duplicates = 'foo' )
288
+
287
289
def test_single_bin (self ):
288
290
# issue 14652
289
291
expected = Series ([0 , 0 ])
0 commit comments