@@ -285,6 +285,36 @@ def test_qcut_duplicates_bin(self):
285
285
# invalid
286
286
self .assertRaises (ValueError , qcut , values , 3 , duplicates = 'foo' )
287
287
288
+ def test_single_quantile (self ):
289
+ # issue 15431
290
+ expected = Series ([0 , 0 ])
291
+
292
+ s = Series ([9. , 9. ])
293
+ result = qcut (s , 1 , labels = False )
294
+ tm .assert_series_equal (result , expected )
295
+
296
+ s = Series ([- 9. , - 9. ])
297
+ result = qcut (s , 1 , labels = False )
298
+ tm .assert_series_equal (result , expected )
299
+
300
+ s = Series ([0. , 0. ])
301
+ result = qcut (s , 1 , labels = False )
302
+ tm .assert_series_equal (result , expected )
303
+
304
+ expected = Series ([0 ])
305
+
306
+ s = Series ([9 ])
307
+ result = qcut (s , 1 , labels = False )
308
+ tm .assert_series_equal (result , expected )
309
+
310
+ s = Series ([- 9 ])
311
+ result = qcut (s , 1 , labels = False )
312
+ tm .assert_series_equal (result , expected )
313
+
314
+ s = Series ([0 ])
315
+ result = qcut (s , 1 , labels = False )
316
+ tm .assert_series_equal (result , expected )
317
+
288
318
def test_single_bin (self ):
289
319
# issue 14652
290
320
expected = Series ([0 , 0 ])
@@ -297,11 +327,29 @@ def test_single_bin(self):
297
327
result = cut (s , 1 , labels = False )
298
328
tm .assert_series_equal (result , expected )
299
329
330
+ expected = Series ([0 ])
331
+
332
+ s = Series ([9 ])
333
+ result = cut (s , 1 , labels = False )
334
+ tm .assert_series_equal (result , expected )
335
+
336
+ s = Series ([- 9 ])
337
+ result = cut (s , 1 , labels = False )
338
+ tm .assert_series_equal (result , expected )
339
+
300
340
# issue 15428
341
+ expected = Series ([0 , 0 ])
342
+
301
343
s = Series ([0. , 0. ])
302
344
result = cut (s , 1 , labels = False )
303
345
tm .assert_series_equal (result , expected )
304
346
347
+ expected = Series ([0 ])
348
+
349
+ s = Series ([0 ])
350
+ result = cut (s , 1 , labels = False )
351
+ tm .assert_series_equal (result , expected )
352
+
305
353
def test_datetime_cut (self ):
306
354
# GH 14714
307
355
# testing for time data to be present as series
0 commit comments