Skip to content

Commit 86a2b05

Browse files
committed
add tests for Series
1 parent d80bab4 commit 86a2b05

File tree

2 files changed

+26
-10
lines changed

2 files changed

+26
-10
lines changed

pandas/tests/indexing/test_iloc.py

+13-5
Original file line numberDiff line numberDiff line change
@@ -594,10 +594,18 @@ def test_iloc_empty_list_indexer_is_ok(self):
594594

595595
def test_identity_slice_returns_new_object(self):
596596
# GH13873
597-
df = DataFrame({'a': [1, 2, 3]})
598-
result = df.iloc[:]
599-
assert result is not df
597+
original_df = DataFrame({'a': [1, 2, 3]})
598+
sliced_df = original_df.iloc[:]
599+
assert sliced_df is not original_df
600600

601601
# should be a shallow copy
602-
df['a'] = [4, 4, 4]
603-
assert (result['a'] == 4).all()
602+
original_df['a'] = [4, 4, 4]
603+
assert (sliced_df['a'] == 4).all()
604+
605+
original_series = Series([1, 2, 3, 4, 5, 6])
606+
sliced_series = original_series.iloc[:]
607+
assert sliced_series is not original_series
608+
609+
# should also be a shallow copy
610+
original_series[:3] = [7, 8, 9]
611+
assert all(sliced_series[:3] == [7, 8, 9])

pandas/tests/indexing/test_loc.py

+13-5
Original file line numberDiff line numberDiff line change
@@ -633,10 +633,18 @@ def test_loc_empty_list_indexer_is_ok(self):
633633

634634
def test_identity_slice_returns_new_object(self):
635635
# GH13873
636-
df = DataFrame({'a': [1, 2, 3]})
637-
result = df.loc[:]
638-
assert result is not df
636+
original_df = DataFrame({'a': [1, 2, 3]})
637+
sliced_df = original_df.loc[:]
638+
assert sliced_df is not original_df
639639

640640
# should be a shallow copy
641-
df['a'] = [4, 4, 4]
642-
assert (result['a'] == 4).all()
641+
original_df['a'] = [4, 4, 4]
642+
assert (sliced_df['a'] == 4).all()
643+
644+
original_series = Series([1, 2, 3, 4, 5, 6])
645+
sliced_series = original_series.loc[:]
646+
assert sliced_series is not original_series
647+
648+
# should also be a shallow copy
649+
original_series[:3] = [7, 8, 9]
650+
assert all(sliced_series[:3] == [7, 8, 9])

0 commit comments

Comments
 (0)