Skip to content

Commit 445afd6

Browse files
committed
test that df.loc[:,0] returns a view
1 parent 86a2b05 commit 445afd6

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

pandas/tests/indexing/test_loc.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -641,10 +641,15 @@ def test_identity_slice_returns_new_object(self):
641641
original_df['a'] = [4, 4, 4]
642642
assert (sliced_df['a'] == 4).all()
643643

644+
# These should not return copies
645+
assert original_df is original_df.loc[:, :]
646+
df = DataFrame(np.random.randn(10, 4))
647+
assert df[0] is df.loc[:,0]
648+
649+
# Same tests for Series
644650
original_series = Series([1, 2, 3, 4, 5, 6])
645651
sliced_series = original_series.loc[:]
646652
assert sliced_series is not original_series
647653

648-
# should also be a shallow copy
649654
original_series[:3] = [7, 8, 9]
650655
assert all(sliced_series[:3] == [7, 8, 9])

0 commit comments

Comments
 (0)