@@ -205,25 +205,30 @@ def test_get_numeric_data_preserve_dtype(self):
205
205
206
206
def test_nonzero_single_element (self ):
207
207
208
- # single item to follow numpy
208
+ # allow single item via bool method
209
209
s = Series ([True ])
210
- self .assert_ (bool (s ) == True )
210
+ self .assert_ (s . bool () is True )
211
211
212
212
s = Series ([False ])
213
- self .assert_ (bool (s ) == False )
213
+ self .assert_ (s . bool () is False )
214
214
215
215
# single item nan to raise
216
- for s in [ Series ([np .nan ]), Series ([pd .NaT ]) ]:
216
+ for s in [ Series ([np .nan ]), Series ([pd .NaT ]), Series ([ True ]), Series ([ False ]) ]:
217
217
self .assertRaises (ValueError , lambda : bool (s ))
218
218
219
+ for s in [ Series ([np .nan ]), Series ([pd .NaT ])]:
220
+ self .assertRaises (ValueError , lambda : s .bool ())
221
+
219
222
# multiple bool are still an error
220
223
for s in [Series ([True ,True ]), Series ([False , False ])]:
221
224
self .assertRaises (ValueError , lambda : bool (s ))
225
+ self .assertRaises (ValueError , lambda : s .bool ())
222
226
223
227
# single non-bool are an error
224
228
for s in [Series ([1 ]), Series ([0 ]),
225
229
Series (['a' ]), Series ([0.0 ])]:
226
230
self .assertRaises (ValueError , lambda : bool (s ))
231
+ self .assertRaises (ValueError , lambda : s .bool ())
227
232
228
233
229
234
class TestDataFrame (unittest .TestCase , Generic ):
@@ -235,6 +240,19 @@ def test_rename_mi(self):
235
240
index = MultiIndex .from_tuples ([("A" ,x ) for x in ["a" ,"B" ,"c" ]]))
236
241
result = df .rename (str .lower )
237
242
243
+ def test_nonzero_single_element (self ):
244
+
245
+ # allow single item via bool method
246
+ df = DataFrame ([[True ]])
247
+ self .assert_ (df .bool () is True )
248
+
249
+ df = DataFrame ([[False ]])
250
+ self .assert_ (df .bool () is False )
251
+
252
+ df = DataFrame ([[False , False ]])
253
+ self .assertRaises (ValueError , lambda : df .bool ())
254
+ self .assertRaises (ValueError , lambda : bool (df ))
255
+
238
256
def test_get_numeric_data_preserve_dtype (self ):
239
257
240
258
# get the numeric data
0 commit comments