Skip to content

Commit 1d45182

Browse files
committed
TST: unit test on indexing single column
1 parent 440e41a commit 1d45182

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-1
lines changed

pandas/core/generic.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,14 @@ def drop(self, labels, axis=0):
169169

170170

171171
class NDFrame(PandasObject):
172-
172+
"""
173+
174+
Parameters
175+
----------
176+
data : BlockManager
177+
axes : list
178+
copy : boolean, default False
179+
"""
173180
# kludge
174181
_default_stat_axis = 0
175182

pandas/tests/test_frame.py

+9
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,15 @@ def test_setitem_boolean(self):
124124
expected.values[mask.values] = nan
125125
assert_frame_equal(df, expected)
126126

127+
def test_setitem_boolean_column(self):
128+
expected = self.frame.copy()
129+
mask = self.frame['A'] > 0
130+
131+
self.frame.ix[mask, 'B'] = 0
132+
expected.values[mask, 1] = 0
133+
134+
assert_frame_equal(self.frame, expected)
135+
127136
def test_setitem_corner(self):
128137
# corner case
129138
df = DataFrame({'B' : [1., 2., 3.],

0 commit comments

Comments
 (0)