fixed file modes for core/frame.py, test/test_frame.py

pandas-dev · Nov 11, 2012 · 030bc66 · 030bc66
1 parent 540fafd
commit 030bc66
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pandas/core/frame.py b/pandas/core/frame.py
@@ -1878,8 +1878,7 @@ def __setitem__(self, key, value):
             self._set_item(key, value)
 
     def _boolean_set(self, key, value):
-        mask = key.values
-        if mask.dtype != np.bool_:
+        if key.values.dtype != np.bool_:
             raise ValueError('Must pass DataFrame with boolean values only')
 
         if self._is_mixed_type:

diff --git a/pandas/tests/test_frame.py b/pandas/tests/test_frame.py
@@ -142,6 +142,7 @@ def test_getitem_boolean(self):
         self.assertRaises(ValueError, self.tsframe.__getitem__, self.tsframe)
 
         # test df[df >0] works
+
         bif = self.tsframe[self.tsframe > 0]
         bifw = DataFrame(np.where(self.tsframe>0,self.tsframe,np.nan),index=self.tsframe.index,columns=self.tsframe.columns)
         self.assert_(isinstance(bif,DataFrame))