Skip to content

Commit 23f3d35

Browse files
author
shirzady1934
committed
issue 1162 added test_loc_list_str function for test loc issue with list of str's
1 parent 08ad871 commit 23f3d35

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/test_frame.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3585,6 +3585,16 @@ def test_in_columns() -> None:
35853585
check(assert_type(df[cols], pd.DataFrame), pd.DataFrame)
35863586
check(assert_type(df.groupby(by=cols).sum(), pd.DataFrame), pd.DataFrame)
35873587

3588+
def test_loc_list_str() -> None:
3589+
# GH 1162 (PR)
3590+
df = pd.DataFrame([[1, 2], [4, 5], [7, 8]],
3591+
index=['cobra', 'viper', 'sidewinder'],
3592+
columns=['max_speed', 'shield'])
3593+
3594+
result = df.loc[['viper', 'sidewinder']]
3595+
check(assert_type(result, pd.DataFrame), pd.DataFrame)
3596+
3597+
35883598

35893599
def test_insert_newvalues() -> None:
35903600
df = pd.DataFrame({"a": [1, 2]})

0 commit comments

Comments
 (0)